def testAdd(self): vms = VirtualMachines() before_count = len(vms.all()) self.assertTrue(vms.add(vms.new(), persist=False)) after_count = len(vms.all()) self.assertGreater(after_count, before_count) self.assertEqual(before_count + 1, after_count)
def testInit(self): vms = VirtualMachines() self.assertIsInstance(vms, ModelCollection) self.assertGreater(vms._items, 0) self.assertIsInstance(vms.all(), list) for i in vms.all(): self.assertIsInstance(i, VirtualMachine)
def testNew(self): vms = VirtualMachines() self.assertIsInstance(vms.new(), VirtualMachine) e = vms.new(name="lol") self.assertEqual(e.name, 'lol') self.assertEqual(e.ip, None) e = vms.new() self.assertIsNotNone(e.name) self.assertNotEqual(vms.new().name, e.name)
def test_InsertQuery(self): vms = VirtualMachines() entity = vms.new(name=None) expected = OrderedDict([ ('id', None), ('name', None), ('ip', None), ('status', None), ('image_id', None), ('image_id', None), ('instance_type_id', None), ('instance_id', None),('user', None), ('config', None) ]) data = vms._generate_query_data(entity) self.assertEqual(expected, data) expected = 'INSERT INTO virtual_machines(id,name,ip,status,image_id,instance_type_id,instance_id,user,config) VALUES (?,?,?,?,?,?,?,?,?)' self.assertEqual(vms._generate_insert_query(data), expected)
def testDelete(self): vms = VirtualMachines() entity = vms.new() expected = 'DELETE FROM virtual_machines WHERE id = ?' self.assertEqual(vms._generate_delete_query(entity.name), expected)
def testTableDefinition(self): vms = VirtualMachines() expected = 'CREATE TABLE virtual_machines (id TEXT PRIMARY KEY,name TEXT,ip TEXT,status TEXT,image_id TEXT,instance_type_id TEXT,instance_id TEXT,user TEXT,config TEXT)' self.assertEqual(str(vms.table_definition()), expected)
def test_GenerateSelectQuery(self): vms = VirtualMachines() expected = 'SELECT * FROM virtual_machines' self.assertEqual(vms._generate_select_query(), expected)
def testFirst(self): vms = VirtualMachines() self.assertEqual(vms.first(), None) entity = vms.new() vms.add(entity, persist=False) self.assertEqual(vms.first(), entity)