def test_with_manager_name(self): m_fs = mock.Mock() m_fs.find.return_value = [{"name": "thomas FS"}, {"name": "nicolas FS"}] m_mongo = mock.Mock() m_mongo.find.return_value = [{"name": "thomas Mongo"}, {"name": "nicolas Mongo"}] manager = ChainManager([("fs", m_fs), ("mongo", m_mongo)]) result = manager.find_one(manager="fs") self.assertEquals("fs", result['manager']) self.assertEquals("thomas FS", result['name']) result = manager.find_one(manager="mongo") self.assertEquals("mongo", result['manager']) self.assertEquals("thomas Mongo", result['name'])
def test_with_managers_find_one(self): m = mock.Mock() m.find.return_value = [{"name": "thomas"}, {"name": "nicolas"}] manager = ChainManager([("fs", m)]) result = manager.find_one() self.assertEquals("fs", result['manager']) self.assertEquals("thomas", result['name'])
def test_with_no_manager_and_find(self): manager = ChainManager(None) self.assertEquals([], manager.find()) self.assertEquals(None, manager.find_one())