def case_update_record(self, refresh_db_before):
""" Create test record in Ip table and then update it."""
original_address = "0.0.0.0"
update_address = "11.11.11.11"
original_ip = IpOp.add(original_address)
updated_ip_local = IpOp.update(original_ip, update_address)
self.assertEqual(updated_ip_local.address, update_address)
updated_ip_get = IpOp.get(address=update_address)
self.assertTrue(len(updated_ip_get) is 1)
self.assertEqual(updated_ip_get[0].address, update_address)
self.assertEqual(updated_ip_get[0].id, original_ip.id)
def test_update(self, mock_ip, mock_val_address, mock_db):
""" Assumptions:
- update method run
"""
new_name = "Test Name"
ip_obj = mock_ip()
updated_ip = IpOp.update(ip_obj, new_name)
self.assertTrue(mock_val_address.called)
db_exp_calls = [
mock.call.session.add(updated_ip),
mock.call.session.commit()
]
mock_db.assert_has_calls(db_exp_calls)
