Merge pull request #965 from seriva/fix/nsg-unittests

Commited changed unit tests for NSG disabling.
hitachienergy · Mar 4, 2020 · fdf0a49 · fdf0a49
2 parents 5038272 + c9fa7ed
commit fdf0a49
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/core/src/epicli/tests/cli/engine/providers/aws/test_AWSConfigBuilder.py b/core/src/epicli/tests/cli/engine/providers/aws/test_AWSConfigBuilder.py
@@ -111,7 +111,10 @@ def get_cluster_model(address_pool='10.22.0.0/22', cluster_name='EpiphanyTestClu
             'name': cluster_name,
             'prefix': 'prefix',
             'cloud': {
-                'vnet_address_pool': address_pool
+                'vnet_address_pool': address_pool,
+                'network': {
+                    'use_network_security_groups': True
+                }
             }
         }
     })

diff --git a/core/src/epicli/tests/cli/engine/providers/azure/test_AzureConfigBuilder.py b/core/src/epicli/tests/cli/engine/providers/azure/test_AzureConfigBuilder.py
@@ -125,7 +125,10 @@ def get_cluster_model(address_pool='10.22.0.0/22', cluster_name='EpiphanyTestClu
             'cloud': {
                 'region': 'West Europe',
                 'vnet_address_pool': address_pool,
-                'use_public_ips': True
+                'use_public_ips': True,
+                'network': {
+                    'use_network_security_groups': True
+                }
             }
         }
     })