Research unit test writing tools and add some tests for the project

[andrejste7]

Description

We have now a code base for working project, but we have no unit tests.. So lets try to fix this

TODO

• [ ] Find package for adding unit tests
• [ ] Write unit tests for logic we have

[RJSkudra]

@mesahil321 Any updates?