migrated stage tests to Mocha