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)
