Write tests

[shankarpriyank]

It would be really useful if we can have some tests in the project, at this moment the project is still in development phase, so end to end tests won't be useful, but we can still use some unit tests