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()