Add unit tests with Mocha