fix(test): Added unit tests

Added the missing unit tests.

fixes #2
Merge pull request #3 from esarbanis/tests

fix(test): Added unit tests