From d87125662f2f55204244e43ec6522fe36c6bf21e Mon Sep 17 00:00:00 2001
From: Nikita Zubkov <nzubkov@mirantis.com>
Date: Mon, 29 Aug 2016 17:56:09 +0300
Subject: [PATCH] Add test for vip transformer

Change-Id: I65124237604fe6718ad6c351825f192a98d470fb
(cherry picked from commit 55422ddce7b0e9e1b9976504500de8a1688a99d7)
---
 cluster_upgrade/tests/test_transformations.py | 35 +++++++++++++++++++
 1 file changed, 35 insertions(+)

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,
+                }}
+        )