Added in Mocha tests for integration