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())
