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):