def testAdd(self): firewalls = Firewalls() before_count = len(firewalls.all()) self.assertTrue(firewalls.add(firewalls.new(), persist=False)) after_count = len(firewalls.all()) self.assertGreater(after_count, before_count) self.assertEqual(before_count + 1, after_count)
def testFirst(self): firewalls = Firewalls() self.assertEqual(firewalls.first(), None) entity = firewalls.new() firewalls.add(entity, persist=False) self.assertEqual(firewalls.first(), entity)