def testAdd(self): jails = Jails() before_count = len(jails.all()) self.assertTrue(jails.add(jails.new(), persist=False)) after_count = len(jails.all()) self.assertGreater(after_count, before_count) self.assertEqual(before_count + 1, after_count)
def testInit(self): jails = Jails() self.assertIsInstance(jails, ModelCollection) self.assertGreater(jails._items, 0) self.assertIsInstance(jails.all(), list) for i in jails.all(): self.assertIsInstance(i, Jail)