Merge "Add IPv6 unit test for _get_route_source"

Jenkins · openstack-gerrit · commit d691a726b31f · 2017-01-19T22:34:21.000Z
diff --git a/ironic_python_agent/tests/unit/test_agent.py b/ironic_python_agent/tests/unit/test_agent.py
@@ -467,12 +467,20 @@ def test_get_node_uuid_invalid_node(self):
                           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):
