Ejemplo n.º 1
0
    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])
Ejemplo n.º 2
0
 def test_BeforeAdd(self):
     mc = ModelCollection()
     self.assertTrue(mc._before_add(None))