def setUp(self):
# manually register CUSTOM_MGMT_AKS_PREVIEW
register_aks_preview_resource_type()
self.cli_ctx = MockCLI()
self.cmd = MockCmd(self.cli_ctx)
self.resource_type = CUSTOM_MGMT_AKS_PREVIEW
self.agentpool_decorator_mode = AgentPoolDecoratorMode.MANAGED_CLUSTER
self.models = AKSPreviewAgentPoolModels(self.cmd, self.resource_type, self.agentpool_decorator_mode)
self.client = MockClient()
class AKSPreviewAgentPoolAddDecoratorManagedClusterModeTestCase(AKSPreviewAgentPoolAddDecoratorCommonTestCase):
def setUp(self):
# manually register CUSTOM_MGMT_AKS_PREVIEW
register_aks_preview_resource_type()
self.cli_ctx = MockCLI()
self.cmd = MockCmd(self.cli_ctx)
self.resource_type = CUSTOM_MGMT_AKS_PREVIEW
self.agentpool_decorator_mode = AgentPoolDecoratorMode.MANAGED_CLUSTER
self.models = AKSPreviewAgentPoolModels(self.cmd, self.resource_type, self.agentpool_decorator_mode)
self.client = MockClient()
def test_set_up_preview_vm_properties(self):
self.common_set_up_preview_vm_properties()
def test_set_up_motd(self):
self.common_set_up_motd()
def test_set_up_gpu_propertes(self):
self.common_set_up_gpu_propertes()
def test_set_up_custom_ca_trust(self):
self.common_set_up_custom_ca_trust()
def test_construct_agentpool_profile_preview(self):
import inspect
from azext_aks_preview.custom import aks_create
optional_params = {}
positional_params = []
for _, v in inspect.signature(aks_create).parameters.items():
if v.default != v.empty:
optional_params[v.name] = v.default
else:
positional_params.append(v.name)
ground_truth_positional_params = [
"cmd",
"client",
"resource_group_name",
"name",
"ssh_key_value",
]
self.assertEqual(positional_params, ground_truth_positional_params)
# prepare a dictionary of default parameters
raw_param_dict = {
"resource_group_name": "test_rg_name",
"name": "test_cluster_name",
"ssh_key_value": None,
}
raw_param_dict.update(optional_params)
# default value in `aks_create`
dec_1 = AKSPreviewAgentPoolAddDecorator(
self.cmd,
self.client,
raw_param_dict,
self.resource_type,
self.agentpool_decorator_mode,
)
with patch(
"azure.cli.command_modules.acs.agentpool_decorator.cf_agent_pools",
return_value=Mock(list=Mock(return_value=[])),
):
dec_agentpool_1 = dec_1.construct_agentpool_profile_preview()
upgrade_settings_1 = self.models.AgentPoolUpgradeSettings()
ground_truth_agentpool_1 = self.create_initialized_agentpool_instance(
nodepool_name="nodepool1",
orchestrator_version="",
vm_size=CONST_DEFAULT_NODE_VM_SIZE,
os_type=CONST_DEFAULT_NODE_OS_TYPE,
enable_node_public_ip=False,
enable_auto_scaling=False,
count=3,
node_taints=[],
os_disk_size_gb=0,
upgrade_settings=upgrade_settings_1,
type=CONST_VIRTUAL_MACHINE_SCALE_SETS,
enable_encryption_at_host=False,
enable_ultra_ssd=False,
enable_fips=False,
mode=CONST_NODEPOOL_MODE_SYSTEM,
workload_runtime=CONST_WORKLOAD_RUNTIME_OCI_CONTAINER,
enable_custom_ca_trust=False,
)
self.assertEqual(dec_agentpool_1, ground_truth_agentpool_1)
dec_1.context.raw_param.print_usage_statistics()