Add mocha test