From 256b233b4b5d91dae9f4103c50b2351aee74b80c Mon Sep 17 00:00:00 2001 From: Derek Higgins <derekh@redhat.com> Date: Thu, 12 Jan 2017 10:23:10 +0000 Subject: [PATCH] Add IPv6 unit test for _get_route_source Follow up from previous patch Change-Id: Ic0286c4eac57f3ae06b237b700d9dbe95ea8b2c0 --- ironic_python_agent/tests/unit/test_agent.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/ironic_python_agent/tests/unit/test_agent.py b/ironic_python_agent/tests/unit/test_agent.py index 71ee80cc9..0b8ddc0ee 100644 --- a/ironic_python_agent/tests/unit/test_agent.py +++ b/ironic_python_agent/tests/unit/test_agent.py @@ -467,12 +467,20 @@ class TestBaseAgent(test_base.BaseTestCase): self.agent.get_node_uuid) @mock.patch.object(utils, 'execute', autospec=True) - def test_get_route_source(self, mock_execute): + def test_get_route_source_ipv4(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(utils, 'execute', autospec=True) + def test_get_route_source_ipv6(self, mock_execute): + mock_execute.return_value = ('XXX src 1:2::3:4 metric 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):