def test_replace_machine_right_data(self):
online_determinator.insert_machine('pony', pony_dict_1, 'id')
online_determinator.replace_machine('pony', pony_dict_2, pony_dict_1)
if select('id', 'pony', pony_dict_2) is not None:
contain = True
else:
contain = False
self.assertEqual(contain, True)
delete('pony', pony_dict_2)
def test_insert_machine_wrong_column(self):
with self.assertRaises(Exception) as context:
online_determinator.insert_machine('pony', pony_dict_wrong, 'id')
self.assertTrue("Unknown column 'second_name' in 'field list'" in str(context.exception))
def test_insert_machine_wrong_output_column(self):
with self.assertRaises(Exception) as context:
online_determinator.insert_machine('pony', pony_dict_1, 'id_number')
self.assertTrue("Unknown column 'id_number' in 'field list'" in str(context.exception))
delete('pony', pony_dict_1)
def test_insert_machine_wrong_table(self):
with self.assertRaises(Exception) as context:
online_determinator.insert_machine('not_pony', pony_dict_1, 'id')
self.assertTrue("Table 'pony_color_db.not_pony' doesn't exist" in str(context.exception))
def test_insert_machine_right_data(self):
self.assertEqual(online_determinator.insert_machine('pony', pony_dict_1, 'id'), select('id', 'pony', pony_dict_1))
delete('pony', pony_dict_1)
def test_replace_machine_wrong_key_value(self):
id = online_determinator.insert_machine('pony', pony_dict_1, 'id')
online_determinator.replace_machine('pony', pony_dict_2, pony_dict_wrong_value)
self.assertEqual(id, select('id', 'pony', pony_dict_1 ))
delete('pony', pony_dict_1)
def test_replace_machine_wrong_key_column(self):
online_determinator.insert_machine('pony', pony_dict_1, 'id')
with self.assertRaises(Exception) as context:
online_determinator.replace_machine('pony', pony_dict_2, pony_dict_wrong)
self.assertTrue("Unknown column 'second_name' in 'where clause'" in str(context.exception))
delete('pony', pony_dict_1)
def test_replace_machine_wrong_new_data_column(self):
online_determinator.insert_machine('pony', pony_dict_1, 'id')
with self.assertRaises(Exception) as context:
online_determinator.replace_machine('pony', pony_dict_wrong, pony_dict_1)
self.assertTrue("Unknown column 'second_name' in 'field list'" in str(context.exception))
delete('pony', pony_dict_1)