Add missing Mocha tests