Esempio n. 1
0
    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)
Esempio n. 2
0
    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])