Add test to the project

[nicolam1]

We should have test coverage for the entire functionality of the project
@cronicc might make sense to prioritize what we want at first and what's nice to have.