def testAdd(self):
envs = Environments()
before_count = len(envs.all())
self.assertTrue(envs.add(envs.new(), persist=False))
after_count = len(envs.all())
self.assertGreater(after_count, before_count)
self.assertEqual(before_count + 1, after_count)
def testInit(self):
envs = Environments()
self.assertIsInstance(envs, ModelCollection)
self.assertGreater(envs._items, 0)
self.assertIsInstance(envs.all(), list)
for i in envs.all():
self.assertIsInstance(i, Environment)
def test_InsertQuery(self):
envs = Environments()
entity = envs.new()
expected = OrderedDict([('id', None), ('code', None), ('name', None)])
data = envs._generate_query_data(entity)
self.assertEqual(expected, data)
expected = 'INSERT INTO environments(id,code,name) VALUES (?,?,?)'
self.assertEqual(envs._generate_insert_query(data), expected)
def testFirst(self):
envs = Environments()
self.assertIsNone(envs.first())
envs.add(envs.new(), persist=False)
first = envs._items[0]
self.assertEqual(envs.first(), first)
def testDelete(self):
envs = Environments()
entity = envs.new()
expected = 'DELETE FROM environments WHERE id = ?'
self.assertEqual(envs._generate_delete_query(entity.id), expected)
def testTableDefinition(self):
envs = Environments()
expected = 'CREATE TABLE environments (id TEXT PRIMARY KEY,code TEXT,name TEXT)'
self.assertEqual(str(envs.table_definition()), expected)
def test_GenerateSelectQuery(self):
envs = Environments()
expected = 'SELECT * FROM environments'
self.assertEqual(envs._generate_select_query(), expected)
def testNew(self):
envs = Environments()
self.assertIsInstance(envs.new(), Environment)
e = envs.new(id="lol")
self.assertEqual(e.id, 'lol')
