Expand test suit

[qequ]

Currently there are only integration tests. To ensure a more robust and reliable codebase, it's needed to expand our test suite to include unit tests.