def case_resolve_ip_not_found(self, refresh_db_before):
""" Try to resolve Ip row which does not exists."""
exception_raised = False
try:
ServerOp.resolve_ip("11.11.11.11")
except ServerIpNotFoundError:
exception_raised = True
self.assertTrue(exception_raised)
def test_resolve_ip_not_found(self, mock_ip_op):
""" Assumptions:
- IpOp.get() returns no records
"""
mock_ip_op.get.return_value = list()
with self.assertRaisesRegex(ServerIpNotFoundError, "Not found"):
ServerOp.resolve_ip("11.11.11.11")
mock_ip_get_calls = [mock.call(address="11.11.11.11")]
mock_ip_op.get.assert_has_calls(mock_ip_get_calls)
def case_resolve_ip_positive(self, refresh_db_before):
""" Try to resolve Ip row."""
ips = ["11.11.11.11", "22.22.22.22", "33.33.33.33"]
for ip in ips:
IpOp.add(address=ip)
for id, ip in enumerate(ips):
ip_obj = ServerOp.resolve_ip(ip)
self.assertEqual(ip_obj.id, id + 1)
self.assertEqual(ip_obj.address, ip)
self.assertTrue(isinstance(ip_obj, Ip))
def test_resolve_ip_positive(self, mock_ip_op):
""" Assumptions:
- IpOp.get() returns one record
"""
mock_ip = mock.MagicMock()
mock_ip_op.get.return_value = [mock_ip]
resolved_ip_obj = ServerOp.resolve_ip("11.11.11.11")
self.assertEqual(resolved_ip_obj, mock_ip)
mock_ip_get_calls = [mock.call(address="11.11.11.11")]
mock_ip_op.get.assert_has_calls(mock_ip_get_calls)