Expand test suit

[Alvaro Frias]

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.