def testFind(self): mc = ModelCollection() self.assertEqual(mc.find(name="derp"), []) expected = [MockModel(name="test")] mc._items = expected self.assertEqual(mc.find(name="test"), expected) self.assertEqual(mc.find(name="civic"), []) items = [MockModel(name="ello"), MockModel(name="ello", ip="world")] mc._items = items expected = [items[1]] self.assertEqual(mc.find(ip="world"), expected) self.assertEqual(mc.find(name="ello", ip="world"), expected)
def testFirst(self): mc = ModelCollection() self.assertIsNone(mc.first()) self.assertIsNone(mc.first(name="test")) items = [MockModel(name="ello"), MockModel(name="ello")] mc._items = items self.assertEqual(mc.first(), items[0]) self.assertIsNone(mc.first(outlook="thunderbird")) self.assertEqual(mc.first(name="ello"), items[0])