Add test for vip transformer

Change-Id: I65124237604fe6718ad6c351825f192a98d470fb
(cherry picked from commit 55422ddce7)
This commit is contained in:
Nikita Zubkov 2016-08-29 17:56:09 +03:00 committed by Ilya Kharin
parent e4d4a0b4b4
commit d87125662f
1 changed files with 35 additions and 0 deletions

View File

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