Add test for vip transformer

Change-Id: I65124237604fe6718ad6c351825f192a98d470fb

1 files changed, 35 insertions, 0 deletions

diff --git a/cluster_upgrade/tests/test_transformations.py b/cluster_upgrade/tests/test_transformations.py
index 880ee2d..7186e1b 100644
--- a/cluster_upgrade/tests/test_transformations.py
+++ b/cluster_upgrade/tests/test_transformations.py
@@ -18,6 +18,7 @@ import six
 
 from .. import transformations
 from ..transformations import cluster
+from ..transformations import vip
 
 
 class TestTransformations(nailgun_test_base.BaseUnitTest):
@@ -219,3 +220,37 @@ class TestClusterTransformers(nailgun_test_base.BaseUnitTest):
             cluster.transform_ntp_list,
             cluster.drop_generated_provision,
         ])])
+
+
+class TestVipTransformers(nailgun_test_base.BaseUnitTest):
+    def setUp(self):
+        ip = '0.0.0.0'
+
+        self.data = {
+            'management': {
+                'haproxy': ip,
+                'vrouter': ip,
+                'test': ip,
+            },
+            'public': {
+                'haproxy': ip,
+                'vrouter': ip,
+                'test': ip,
+            }
+        }
+
+    def test_vip_transform(self):
+        ip = '0.0.0.0'
+
+        data = vip.transform_vips(self.data)
+        self.assertEqual(
+            data, {
+                'management': {
+                    'management': ip,
+                    'vrouter': ip,
+                },
+                'public': {
+                    'public': ip,
+                    'vrouter_pub': ip,
+                }}
+        )