Fixing testcases for mongomock