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 testFirst(self):
envs = Environments()
self.assertIsNone(envs.first())
envs.add(envs.new(), persist=False)
first = envs._items[0]
self.assertEqual(envs.first(), first)
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 testDelete(self):
envs = Environments()
entity = envs.new()
expected = 'DELETE FROM environments WHERE id = ?'
self.assertEqual(envs._generate_delete_query(entity.id), expected)
def testNew(self):
envs = Environments()
self.assertIsInstance(envs.new(), Environment)
e = envs.new(id="lol")
self.assertEqual(e.id, 'lol')