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)
def test_InsertQuery(self): jails = Jails() entity = jails.new() expected = OrderedDict( [ ("id", None), ("jail_type", None), ("name", None), ("ip", None), ("netmask", None), ("environment", None), ("url", None), ] ) data = jails._generate_query_data(entity) self.assertEqual(expected, data) expected = "INSERT INTO jails(id,jail_type,name,ip,netmask,environment,url) VALUES (?,?,?,?,?,?,?)" self.assertEqual(jails._generate_insert_query(data), expected)
def testDelete(self): jails = Jails() entity = jails.new() expected = "DELETE FROM jails WHERE id = ?" self.assertEqual(jails._generate_delete_query(entity.id), expected)
def testTableDefinition(self): jails = Jails() expected = "CREATE TABLE jails (id TEXT PRIMARY KEY,jail_type TEXT,name TEXT,ip TEXT,netmask TEXT,environment TEXT,url TEXT)" self.assertEqual(str(jails.table_definition()), expected)
def test_GenerateSelectQuery(self): jails = Jails() expected = "SELECT * FROM jails" self.assertEqual(jails._generate_select_query(), expected)
def testNew(self): jails = Jails() self.assertIsInstance(jails.new(), Jail) e = jails.new(id="lol") self.assertEqual(e.id, "lol")
def testFirst(self): jails = Jails() self.assertEqual(jails.first(), None) entity = jails.new() jails.add(entity, persist=False) self.assertEqual(jails.first(), entity)