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 find(self, **kwargs): self._items = self.all() return ModelCollection.find(self, **kwargs)