Add mocha tests