def testAdd(self):
envs = Images()
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 test_InsertQuery(self):
envs = Images()
entity = envs.new()
expected = OrderedDict([("id", None), ("name", None), ("backend_id", None), ("description", None)])
data = envs._generate_query_data(entity)
self.assertEqual(expected, data)
expected = "INSERT INTO images(id,name,backend_id,description) VALUES (?,?,?,?)"
self.assertEqual(envs._generate_insert_query(data), expected)
def testDelete(self):
envs = Images()
entity = envs.new()
expected = "DELETE FROM images WHERE id = ?"
self.assertEqual(envs._generate_delete_query(entity.id), expected)
def testNew(self):
envs = Images()
self.assertIsInstance(envs.new(), Image)
e = envs.new(id="lol")
self.assertEqual(e.id, "lol")
def testFirst(self):
envs = Images()
self.assertEqual(envs.first(), None)
entity = envs.new()
envs.add(entity, persist=False)
self.assertEqual(envs.first(), entity)
