def test_get_enable_managed_identity(self):
# custom value
ctx_1 = AKSPreviewContext(
self.cmd,
{
"enable_managed_identity": False,
"enable_pod_identity": True
},
self.models,
decorator_mode=DecoratorMode.CREATE,
)
with self.assertRaises(RequiredArgumentMissingError):
self.assertEqual(ctx_1.get_enable_managed_identity(), False)
def test_get_pod_subnet_id(self):
# default
ctx_1 = AKSPreviewContext(
self.cmd,
{"pod_subnet_id": None},
self.models,
decorator_mode=DecoratorMode.CREATE,
)
self.assertEqual(ctx_1.get_pod_subnet_id(), None)
agent_pool_profile = self.models.ManagedClusterAgentPoolProfile(
name="test_nodepool_name", pod_subnet_id="test_mc_pod_subnet_id")
mc = self.models.ManagedCluster(
location="test_location", agent_pool_profiles=[agent_pool_profile])
ctx_1.attach_mc(mc)
self.assertEqual(ctx_1.get_pod_subnet_id(), "test_mc_pod_subnet_id")
def test_get_enable_pod_security_policy(self):
# default
ctx_1 = AKSPreviewContext(
self.cmd,
{"enable_pod_security_policy": False},
self.models,
decorator_mode=DecoratorMode.CREATE,
)
self.assertEqual(ctx_1.get_enable_pod_security_policy(), False)
mc = self.models.ManagedCluster(
location="test_location",
enable_pod_security_policy=True,
)
ctx_1.attach_mc(mc)
self.assertEqual(ctx_1.get_enable_pod_security_policy(), True)
def test_get_node_resource_group(self):
# default
ctx_1 = AKSPreviewContext(
self.cmd,
{"node_resource_group": None},
self.models,
decorator_mode=DecoratorMode.CREATE,
)
self.assertEqual(ctx_1.get_node_resource_group(), None)
mc = self.models.ManagedCluster(
location="test_location",
node_resource_group="test_node_resource_group",
)
ctx_1.attach_mc(mc)
self.assertEqual(ctx_1.get_node_resource_group(),
"test_node_resource_group")
def test_get_enable_fips_image(self):
# default
ctx_1 = AKSPreviewContext(
self.cmd,
{"enable_fips_image": False},
self.models,
decorator_mode=DecoratorMode.CREATE,
)
self.assertEqual(ctx_1.get_enable_fips_image(), False)
agent_pool_profile = self.models.ManagedClusterAgentPoolProfile(
name="test_nodepool_name",
enable_fips=True,
)
mc = self.models.ManagedCluster(
location="test_location", agent_pool_profiles=[agent_pool_profile])
ctx_1.attach_mc(mc)
self.assertEqual(ctx_1.get_enable_fips_image(), True)
def test_get_nat_gateway_idle_timeout(self):
# default
ctx_1 = AKSPreviewContext(
self.cmd,
{"nat_gateway_idle_timeout": None},
self.models,
decorator_mode=DecoratorMode.CREATE,
)
self.assertEqual(ctx_1.get_nat_gateway_idle_timeout(), None)
nat_gateway_profile = self.models.nat_gateway_models.get(
"ManagedClusterNATGatewayProfile")(idle_timeout_in_minutes=20, )
network_profile = self.models.ContainerServiceNetworkProfile(
nat_gateway_profile=nat_gateway_profile)
mc = self.models.ManagedCluster(
location="test_location",
network_profile=network_profile,
)
ctx_1.attach_mc(mc)
self.assertEqual(ctx_1.get_nat_gateway_idle_timeout(), 20)
def test_get_nat_gateway_managed_outbound_ip_count(self):
# default
ctx_1 = AKSPreviewContext(
self.cmd,
{"nat_gateway_managed_outbound_ip_count": None},
self.models,
decorator_mode=DecoratorMode.CREATE,
)
self.assertEqual(ctx_1.get_nat_gateway_managed_outbound_ip_count(),
None)
nat_gateway_profile = self.models.nat_gateway_models.get(
"ManagedClusterNATGatewayProfile")(
managed_outbound_ip_profile=self.models.nat_gateway_models.get(
"ManagedClusterManagedOutboundIPProfile")(count=10))
network_profile = self.models.ContainerServiceNetworkProfile(
nat_gateway_profile=nat_gateway_profile)
mc = self.models.ManagedCluster(
location="test_location",
network_profile=network_profile,
)
ctx_1.attach_mc(mc)
self.assertEqual(ctx_1.get_nat_gateway_managed_outbound_ip_count(), 10)
def test_get_enable_pod_identity_with_kubenet(self):
# default
ctx_1 = AKSPreviewContext(
self.cmd,
{"enable_pod_identity_with_kubenet": False},
self.models,
decorator_mode=DecoratorMode.CREATE,
)
self.assertEqual(ctx_1.get_enable_pod_identity_with_kubenet(), False)
pod_identity_profile = self.models.ManagedClusterPodIdentityProfile(
enabled=True,
allow_network_plugin_kubenet=True,
)
mc = self.models.ManagedCluster(
location="test_location",
pod_identity_profile=pod_identity_profile,
)
ctx_1.attach_mc(mc)
# fail on enable_managed_identity not specified
# with self.assertRaises(RequiredArgumentMissingError):
self.assertEqual(ctx_1.get_enable_pod_identity_with_kubenet(), True)
# custom value
ctx_2 = AKSPreviewContext(
self.cmd,
{
"enable_managed_identity": True,
"enable_pod_identity": True,
"enable_pod_identity_with_kubenet": False,
},
self.models,
decorator_mode=DecoratorMode.CREATE,
)
network_profile_2 = self.models.ContainerServiceNetworkProfile(
network_plugin="kubenet")
mc_2 = self.models.ManagedCluster(
location="test_location",
network_profile=network_profile_2,
)
ctx_2.attach_mc(mc_2)
# fail on enable_pod_identity_with_kubenet not specified
with self.assertRaises(RequiredArgumentMissingError):
self.assertEqual(ctx_2.get_enable_pod_identity_with_kubenet(),
False)
def test_get_http_proxy_config(self):
# default
ctx_1 = AKSPreviewContext(
self.cmd,
{"http_proxy_config": None},
self.models,
decorator_mode=DecoratorMode.CREATE,
)
self.assertEqual(ctx_1.get_http_proxy_config(), None)
mc = self.models.ManagedCluster(
location="test_location",
http_proxy_config=self.models.ManagedClusterHTTPProxyConfig(
http_proxy="test_http_proxy"),
)
ctx_1.attach_mc(mc)
self.assertEqual(
ctx_1.get_http_proxy_config(),
self.models.ManagedClusterHTTPProxyConfig(
http_proxy="test_http_proxy"),
)
# custom value
ctx_2 = AKSPreviewContext(
self.cmd,
{"http_proxy_config": "fake-path"},
self.models,
decorator_mode=DecoratorMode.CREATE,
)
# fail on invalid file path
with self.assertRaises(InvalidArgumentValueError):
ctx_2.get_http_proxy_config()
# custom value
ctx_3 = AKSPreviewContext(
self.cmd,
{"http_proxy_config": _get_test_data_file("invalidconfig.json")},
self.models,
decorator_mode=DecoratorMode.CREATE,
)
# fail on invalid file path
with self.assertRaises(InvalidArgumentValueError):
ctx_3.get_http_proxy_config()
def test_get_linux_os_config(self):
# default
ctx_1 = AKSPreviewContext(
self.cmd,
{"linux_os_config": None},
self.models,
decorator_mode=DecoratorMode.CREATE,
)
self.assertEqual(ctx_1.get_linux_os_config(), None)
agent_pool_profile = self.models.ManagedClusterAgentPoolProfile(
name="test_nodepool_name",
linux_os_config=self.models.LinuxOSConfig(swap_file_size_mb=200),
)
mc = self.models.ManagedCluster(
location="test_location", agent_pool_profiles=[agent_pool_profile])
ctx_1.attach_mc(mc)
self.assertEqual(
ctx_1.get_linux_os_config(),
self.models.LinuxOSConfig(swap_file_size_mb=200),
)
# custom value
ctx_2 = AKSPreviewContext(
self.cmd,
{"linux_os_config": "fake-path"},
self.models,
decorator_mode=DecoratorMode.CREATE,
)
# fail on invalid file path
with self.assertRaises(InvalidArgumentValueError):
ctx_2.get_linux_os_config()
# custom value
ctx_3 = AKSPreviewContext(
self.cmd,
{"linux_os_config": _get_test_data_file("invalidconfig.json")},
self.models,
decorator_mode=DecoratorMode.CREATE,
)
# fail on invalid file path
with self.assertRaises(InvalidArgumentValueError):
ctx_3.get_linux_os_config()