def case_update_record(self, refresh_db_before):
""" Create test record in ServerType table and then update it."""
original_type = "TypeOrig"
update_type = "TypeUpdate"
original_type_obj = ServerTypeOp.add(original_type)
updated_type_local = ServerTypeOp.update(original_type_obj, update_type)
self.assertEqual(updated_type_local.name, update_type)
updated_type_get = ServerTypeOp.get(name=update_type)
self.assertTrue(len(updated_type_get) is 1)
self.assertEqual(updated_type_get[0].name, update_type)
self.assertEqual(updated_type_get[0].id, original_type_obj.id)
def test_update(self, mock_servtype, mock_val_name, mock_db):
""" Assumptions:
- update method run
"""
new_name = "TestName"
ip_obj = mock_servtype()
updated_ip = ServerTypeOp.update(ip_obj, new_name)
self.assertTrue(mock_val_name.called)
db_exp_calls = [
mock.call.session.add(updated_ip),
mock.call.session.commit()
]
mock_db.assert_has_calls(db_exp_calls)