add import

microsoft · Jan 18, 2025 · 018a9fb · 018a9fb
1 parent 2966317
commit 018a9fb
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/scripts/cleanup_ebpf_cicd_tests.ps1 b/scripts/cleanup_ebpf_cicd_tests.ps1
@@ -13,6 +13,7 @@ Push-Location $WorkingDirectory
 
 Import-Module .\common.psm1 -Force -ArgumentList ($LogFileName) -WarningAction SilentlyContinue
 # $TestVMCredential = Get-AzureKeyVaultCredential -SecretName 'Administrator'
+Import-Module CredentialManager -ErrorAction Stop
 $TestVMCredential = Get-StoredCredential -Target $Target -ErrorAction Stop
 
 # Read the test execution json.

diff --git a/scripts/execute_ebpf_cicd_tests.ps1 b/scripts/execute_ebpf_cicd_tests.ps1
@@ -17,7 +17,7 @@ param ([Parameter(Mandatory = $false)][string] $AdminTarget = "TEST_VM",
 Push-Location $WorkingDirectory
 
 Import-Module $WorkingDirectory\common.psm1 -Force -ArgumentList ($LogFileName) -ErrorAction Stop
-
+Import-Module CredentialManager -ErrorAction Stop
 # $AdminTestVMCredential = Get-AzureKeyVaultCredential -SecretName 'Administrator'
 # $StandardUserTestVMCredential = Get-AzureKeyVaultCredential -SecretName 'VMStandardUser'
 $AdminTestVMCredential = Get-StoredCredential -Target $AdminTarget -ErrorAction Stop

diff --git a/scripts/setup_ebpf_cicd_tests.ps1 b/scripts/setup_ebpf_cicd_tests.ps1
@@ -15,13 +15,14 @@ param ([parameter(Mandatory=$false)][string] $Target = "TEST_VM",
 
 Push-Location $WorkingDirectory
 
-$TestVMCredential = Get-StoredCredential -Target $Target -ErrorAction Stop
+Import-Module CredentialManager -ErrorAction Stop
 
 # Load other utility modules.
 Import-Module .\common.psm1 -Force -ArgumentList ($LogFileName) -WarningAction SilentlyContinue
 # $TestVMCredential = Get-AzureKeyVaultCredential -SecretName 'Administrator'
 
 Import-Module .\config_test_vm.psm1 -Force -ArgumentList ($TestVMCredential.UserName, $TestVMCredential.Password, $WorkingDirectory, $LogFileName) -WarningAction SilentlyContinue
+$TestVMCredential = Get-StoredCredential -Target $Target -ErrorAction Stop
 
 # Read the test execution json.
 $Config = Get-Content ("{0}\{1}" -f $PSScriptRoot, $TestExecutionJsonFileName) | ConvertFrom-Json