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)
