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)