[KMS] Update tests

private keyvault + konnectivity is blocked now, so vnet integration should be used. Signed-off-by: Zhecheng Li <[email protected]>
Azure · Sep 25, 2024 · 4d8a7c1 · 4d8a7c1
1 parent 2a94098
commit 4d8a7c1
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/azure-cli/azure/cli/command_modules/acs/tests/latest/test_aks_commands.py b/src/azure-cli/azure/cli/command_modules/acs/tests/latest/test_aks_commands.py
@@ -9213,9 +9213,11 @@ def test_aks_create_with_azurekeyvaultkms_private_key_vault(self, resource_group
         ]).get_output_in_json()
 
         create_cmd = 'aks create --resource-group={resource_group} --name={name} ' \
+                     '--aks-custom-headers AKSHTTPCustomFeatures=Microsoft.ContainerService/EnableAPIServerVnetIntegrationPreview ' \
                      '--assign-identity {identity_id} ' \
                      '--enable-azure-keyvault-kms --azure-keyvault-kms-key-id={key_id} ' \
                      '--azure-keyvault-kms-key-vault-network-access=Private --azure-keyvault-kms-key-vault-resource-id {kv_resource_id} ' \
+                     '--enable-apiserver-vnet-integration=true ' \
                      '--ssh-key-value={ssh_key_value} -o json'
         self.cmd(create_cmd, checks=[
             self.check('provisioningState', 'Succeeded'),
@@ -9339,7 +9341,9 @@ def test_aks_update_with_azurekeyvaultkms_private_key_vault(self, resource_group
         ]).get_output_in_json()
 
         create_cmd = 'aks create --resource-group={resource_group} --name={name} ' \
+                     '--aks-custom-headers AKSHTTPCustomFeatures=Microsoft.ContainerService/EnableAPIServerVnetIntegrationPreview ' \
                      '--assign-identity {identity_id} ' \
+                     '--enable-apiserver-vnet-integration=true ' \
                      '--ssh-key-value={ssh_key_value} -o json'
         self.cmd(create_cmd, checks=[
             self.check('provisioningState', 'Succeeded'),
@@ -9438,9 +9442,11 @@ def test_aks_create_with_azurekeyvaultkms_private_cluster_v1_private_key_vault(s
         ]).get_output_in_json()
 
         create_cmd = 'aks create --resource-group={resource_group} --name={name} ' \
+                     '--aks-custom-headers AKSHTTPCustomFeatures=Microsoft.ContainerService/EnableAPIServerVnetIntegrationPreview ' \
                      '--assign-identity {identity_id} --enable-private-cluster ' \
                      '--enable-azure-keyvault-kms --azure-keyvault-kms-key-id={key_id} ' \
                      '--azure-keyvault-kms-key-vault-network-access=Private --azure-keyvault-kms-key-vault-resource-id {kv_resource_id} ' \
+                     '--enable-apiserver-vnet-integration=true ' \
                      '--ssh-key-value={ssh_key_value} -o json'
         self.cmd(create_cmd, checks=[
             self.check('provisioningState', 'Succeeded'),