Add unit tests

[Morgul]

We need to add unit tests for the current api. The tests should be done in Mocha.