示例#1
0
文件: test_model.py 项目: masom/Puck
    def testAdd(self):
        mc = ModelCollection()

        entity = MockModel()
        mc._before_add = lambda x: False
        self.assertFalse(mc.add(entity, persist=False))

        mc._before_add = lambda x: True
        self.assertTrue(mc.add(entity, persist=False))
        self.assertEqual(mc._items, [entity])
示例#2
0
文件: test_model.py 项目: masom/Puck
 def test_BeforeAdd(self):
     mc = ModelCollection()
     self.assertTrue(mc._before_add(None))