diff options
Diffstat (limited to 'ironic_python_agent/tests')
| -rw-r--r-- | ironic_python_agent/tests/unit/test_agent.py | 32 | ||||
| -rw-r--r-- | ironic_python_agent/tests/unit/test_ironic_api_client.py | 29 | ||||
| -rw-r--r-- | ironic_python_agent/tests/unit/test_netutils.py | 8 |
3 files changed, 69 insertions, 0 deletions
diff --git a/ironic_python_agent/tests/unit/test_agent.py b/ironic_python_agent/tests/unit/test_agent.py index 62bd509e..71ee80cc 100644 --- a/ironic_python_agent/tests/unit/test_agent.py +++ b/ironic_python_agent/tests/unit/test_agent.py @@ -466,6 +466,22 @@ class TestBaseAgent(test_base.BaseTestCase): self.assertRaises(errors.UnknownNodeError, self.agent.get_node_uuid) + @mock.patch.object(utils, 'execute', autospec=True) + def test_get_route_source(self, mock_execute): + mock_execute.return_value = ('XXX src 1.2.3.4 XXX\n cache', None) + + source = self.agent._get_route_source('XXX') + self.assertEqual('1.2.3.4', source) + + @mock.patch.object(agent, 'LOG', autospec=True) + @mock.patch.object(utils, 'execute', autospec=True) + def test_get_route_source_indexerror(self, mock_execute, mock_log): + mock_execute.return_value = ('XXX src \n cache', None) + + source = self.agent._get_route_source('XXX') + self.assertIsNone(source) + mock_log.warning.assert_called_once() + @mock.patch.object(hardware.GenericHardwareManager, '_wait_for_disks', lambda self: None) @@ -597,6 +613,22 @@ class TestAdvertiseAddress(test_base.BaseTestCase): mock_exec.assert_called_once_with('ip', 'route', 'get', '1.2.1.2') mock_gethostbyname.assert_called_once_with('1.2.1.2') + def test_route_with_ipv6(self, mock_exec, mock_gethostbyname): + self.agent.api_url = 'http://[fc00:1111::1]:8081/v1' + mock_gethostbyname.side_effect = socket.gaierror() + mock_exec.return_value = ( + """fc00:101::1 dev br-ctlplane src fc00:101::4 metric 0 + cache """, + "" + ) + + self.agent.set_agent_advertise_addr() + + self.assertEqual(('fc00:101::4', 9990), + self.agent.advertise_address) + mock_exec.assert_called_once_with('ip', 'route', 'get', 'fc00:1111::1') + mock_gethostbyname.assert_called_once_with('fc00:1111::1') + def test_route_with_host(self, mock_exec, mock_gethostbyname): mock_gethostbyname.return_value = '1.2.1.2' mock_exec.return_value = ( diff --git a/ironic_python_agent/tests/unit/test_ironic_api_client.py b/ironic_python_agent/tests/unit/test_ironic_api_client.py index 3fe1807c..7046f5f9 100644 --- a/ironic_python_agent/tests/unit/test_ironic_api_client.py +++ b/ironic_python_agent/tests/unit/test_ironic_api_client.py @@ -70,8 +70,29 @@ class TestBaseIronicPythonAgent(test_base.BaseTestCase): heartbeat_path = 'v1/heartbeat/deadbeef-dabb-ad00-b105-f00d00bab10c' request_args = self.api_client.session.request.call_args[0] + data = self.api_client.session.request.call_args[1]['data'] self.assertEqual('POST', request_args[0]) self.assertEqual(API_URL + heartbeat_path, request_args[1]) + self.assertEqual('{"callback_url": "http://192.0.2.1:9999"}', data) + + def test_successful_heartbeat_ip6(self): + response = FakeResponse(status_code=202) + + self.api_client.session.request = mock.Mock() + self.api_client.session.request.return_value = response + + self.api_client.heartbeat( + uuid='deadbeef-dabb-ad00-b105-f00d00bab10c', + advertise_address=('fc00:1111::4', '9999') + ) + + heartbeat_path = 'v1/heartbeat/deadbeef-dabb-ad00-b105-f00d00bab10c' + request_args = self.api_client.session.request.call_args[0] + data = self.api_client.session.request.call_args[1]['data'] + self.assertEqual('POST', request_args[0]) + self.assertEqual(API_URL + heartbeat_path, request_args[1]) + self.assertEqual('{"callback_url": "http://[fc00:1111::4]:9999"}', + data) def test_heartbeat_requests_exception(self): self.api_client.session.request = mock.Mock() @@ -240,3 +261,11 @@ class TestBaseIronicPythonAgent(test_base.BaseTestCase): node_uuid=None) self.assertFalse(error) + + def test_get_agent_url_ipv4(self): + url = self.api_client._get_agent_url(('1.2.3.4', '9999')) + self.assertEqual('http://1.2.3.4:9999', url) + + def test_get_agent_url_ipv6(self): + url = self.api_client._get_agent_url(('1:2::3:4', '9999')) + self.assertEqual('http://[1:2::3:4]:9999', url) diff --git a/ironic_python_agent/tests/unit/test_netutils.py b/ironic_python_agent/tests/unit/test_netutils.py index e3fa823f..14c8b82e 100644 --- a/ironic_python_agent/tests/unit/test_netutils.py +++ b/ironic_python_agent/tests/unit/test_netutils.py @@ -336,3 +336,11 @@ class TestNetutils(test_base.BaseTestCase): sock1.close.assert_called_once_with() sock2.close.assert_called_once_with() + + def test_wrap_ipv6(self): + res = netutils.wrap_ipv6('1:2::3:4') + self.assertEqual('[1:2::3:4]', res) + + def test_wrap_ipv6_with_ipv4(self): + res = netutils.wrap_ipv6('1.2.3.4') + self.assertEqual('1.2.3.4', res) |