def testFirst(self): jail_types = JailTypes() self.assertIsNone(jail_types.first()) jail_types.add(jail_types.new(), persist=False) first = jail_types._items[0] self.assertEqual(jail_types.first(), first)
def testAdd(self): jail_types = JailTypes() before_count = len(jail_types.all()) self.assertTrue(jail_types.add(jail_types.new(), persist=False)) after_count = len(jail_types.all()) self.assertGreater(after_count, before_count) self.assertEqual(before_count + 1, after_count)