Stress test usability feedback (Azure#18513)

azure-sdk · benbp · web-flow · commit ed6cc1e22b03 · 2021-11-03T15:29:08.000-07:00
Co-authored-by: Ben Broderick Phillips &lt;bebroder@microsoft.com&gt;
diff --git a/eng/common/scripts/stress-testing/deploy-stress-tests.ps1 b/eng/common/scripts/stress-testing/deploy-stress-tests.ps1
@@ -12,7 +12,10 @@ param(
     [switch]$Login,
 
     [Parameter(ParameterSetName = 'DoLogin')]
-    [string]$Subscription
+    [string]$Subscription,
+
+    # Default to true in Azure Pipelines environments
+    [switch] $CI = ($null -ne $env:SYSTEM_TEAMPROJECTID)
 )
 
 $ErrorActionPreference = 'Stop'
@@ -88,7 +91,7 @@ function DeployStressTests(
     Run helm repo update
     if ($LASTEXITCODE) { return $LASTEXITCODE }
 
-    $pkgs = FindStressPackages $searchDirectory $filters
+    $pkgs = FindStressPackages $searchDirectory $filters $CI
     Write-Host "" "Found $($pkgs.Length) stress test packages:"
     Write-Host $pkgs.Directory ""
     foreach ($pkg in $pkgs) {
@@ -106,6 +109,8 @@ function DeployStressTests(
         }
         exit 1
     }
+
+    Write-Host "`nStress test telemetry links (dashboard, fileshare, etc.): https://aka.ms/azsdk/stress/dashboard"
 }
 
 function DeployStressPackage(
diff --git a/eng/common/scripts/stress-testing/find-all-stress-packages.ps1 b/eng/common/scripts/stress-testing/find-all-stress-packages.ps1
@@ -9,7 +9,7 @@ class StressTestPackageInfo {
     [string]$ReleaseName
 }
 
-function FindStressPackages([string]$directory, [hashtable]$filters = @{}) {
+function FindStressPackages([string]$directory, [hashtable]$filters = @{}, [boolean]$CI = $false) {
     # Bare minimum filter for stress tests
     $filters['stressTest'] = 'true'
 
@@ -18,7 +18,7 @@ function FindStressPackages([string]$directory, [hashtable]$filters = @{}) {
     foreach ($chartFile in $chartFiles) {
         $chart = ParseChart $chartFile
         if (matchesAnnotations $chart $filters) {
-            $packages += NewStressTestPackageInfo $chart $chartFile
+            $packages += NewStressTestPackageInfo $chart $chartFile $CI
         }
     }
 
@@ -39,9 +39,17 @@ function MatchesAnnotations([hashtable]$chart, [hashtable]$filters) {
     return $true
 }
 
-function NewStressTestPackageInfo([hashtable]$chart, [System.IO.FileInfo]$chartFile) {
+function NewStressTestPackageInfo([hashtable]$chart, [System.IO.FileInfo]$chartFile, [boolean]$CI) {
+    $namespace = if ($CI) {
+        $chart.annotations.namespace
+    } else {
+        $namespace = if ($env:USER) { $env:USER } else { "${env:USERNAME}" }
+        # Remove spaces, etc. that may be in $namespace
+        $namespace -replace '\W'
+    }
+
     return [StressTestPackageInfo]@{
-        Namespace = $chart.annotations.namespace
+        Namespace = $namespace
         Directory = $chartFile.DirectoryName
         ReleaseName = $chart.name
     }

Original file line number	Diff line number	Diff line change
`@@ -12,7 +12,10 @@ param(`
`12`	`12`	`[switch]$Login,`
`13`	`13`
`14`	`14`	`[Parameter(ParameterSetName = 'DoLogin')]`
`15`		`- [string]$Subscription`
	`15`	`+ [string]$Subscription,`
	`16`	`+`
	`17`	`+ # Default to true in Azure Pipelines environments`
	`18`	`+ [switch] $CI = ($null -ne $env:SYSTEM_TEAMPROJECTID)`
`16`	`19`	`)`
`17`	`20`
`18`	`21`	`$ErrorActionPreference = 'Stop'`
`@@ -88,7 +91,7 @@ function DeployStressTests(`
`88`	`91`	`Run helm repo update`
`89`	`92`	`if ($LASTEXITCODE) { return $LASTEXITCODE }`
`90`	`93`
`91`		`- $pkgs = FindStressPackages $searchDirectory $filters`
	`94`	`+ $pkgs = FindStressPackages $searchDirectory $filters $CI`
`92`	`95`	`Write-Host "" "Found $($pkgs.Length) stress test packages:"`
`93`	`96`	`Write-Host $pkgs.Directory ""`
`94`	`97`	`foreach ($pkg in $pkgs) {`
`@@ -106,6 +109,8 @@ function DeployStressTests(`
`106`	`109`	`}`
`107`	`110`	`exit 1`
`108`	`111`	`}`
	`112`	`+`
	`113`	+ Write-Host "`nStress test telemetry links (dashboard, fileshare, etc.): https://aka.ms/azsdk/stress/dashboard"
`109`	`114`	`}`
`110`	`115`
`111`	`116`	`function DeployStressPackage(`