예제 #1
0
파일: test_model.py 프로젝트: masom/Puck
    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)
예제 #2
0
파일: test_model.py 프로젝트: masom/Puck
    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])