1- [CmdletBinding (DefaultParameterSetName = ' Default'
2- param (
3- [string ]$SearchDirectory ,
4- [hashtable ]$Filters ,
5- [string ]$Environment ,
6- [string ]$Repository ,
7- [switch ]$PushImages ,
8- [string ]$ClusterGroup ,
9- [string ]$DeployId ,
10-
11- [Parameter (ParameterSetName = ' DoLogin' , Mandatory = $true )]
12- [switch ]$Login ,
13-
14- [Parameter (ParameterSetName = ' DoLogin'
15- [string ]$Subscription ,
16-
17- # Default to true in Azure Pipelines environments
18- [switch ] $CI = ($null -ne $env: SYSTEM_TEAMPROJECTID )
19- )
20-
211$ErrorActionPreference = ' Stop'
222
233. $PSScriptRoot / find-all - stress- packages.ps1
@@ -47,7 +27,7 @@ function RunOrExitOnFailure()
4727 }
4828}
4929
50- function Login ([string ]$subscription , [string ]$clusterGroup , [boolean ]$pushImages )
30+ function Login ([string ]$subscription , [string ]$clusterGroup , [switch ]$pushImages )
5131{
5232 Write-Host " Logging in to subscription, cluster and container registry"
5333 az account show * > $null
@@ -77,13 +57,38 @@ function DeployStressTests(
7757 [string ]$searchDirectory = ' .' ,
7858 [hashtable ]$filters = @ {},
7959 [string ]$environment = ' test' ,
80- [string ]$repository = ' images ' ,
81- [boolean ]$pushImages = $false ,
82- [string ]$clusterGroup = ' rg-stress-cluster-test ' ,
60+ [string ]$repository = ' ' ,
61+ [switch ]$pushImages ,
62+ [string ]$clusterGroup = ' ' ,
8363 [string ]$deployId = ' local' ,
84- [string ]$subscription = ' Azure SDK Developer Playground'
64+ [switch ]$login ,
65+ [string ]$subscription = ' ' ,
66+ [switch ]$ci
8567) {
86- if ($PSCmdlet.ParameterSetName -eq ' DoLogin'
68+ if ($environment -eq ' test'
69+ if ($clusterGroup -or $subscription ) {
70+ Write-Warning " Overriding cluster group and subscription with defaults for 'test' environment."
71+ }
72+ $clusterGroup = ' rg-stress-cluster-test'
73+ $subscription = ' Azure SDK Developer Playground'
74+ } elseif ($environment -eq ' prod'
75+ if ($clusterGroup -or $subscription ) {
76+ Write-Warning " Overriding cluster group and subscription with defaults for 'prod' environment."
77+ }
78+ $clusterGroup = ' rg-stress-cluster-prod'
79+ $subscription = ' Azure SDK Test Resources'
80+ }
81+
82+ if (! $repository ) {
83+ $repository = if ($env: USER ) { $env: USER } else { " ${env: USERNAME} "
84+ # Remove spaces, etc. that may be in $namespace
85+ $repository -replace ' \W'
86+ }
87+
88+ if ($login ) {
89+ if (! $clusterGroup -or ! $subscription ) {
90+ throw " clusterGroup and subscription parameters must be specified when logging into an environment that is not test or prod."
91+ }
8792 Login $subscription $clusterGroup $pushImages
8893 }
8994
@@ -96,7 +101,7 @@ function DeployStressTests(
96101 Write-Host $pkgs.Directory " "
97102 foreach ($pkg in $pkgs ) {
98103 Write-Host " Deploying stress test at '$ ( $pkg.Directory ) '"
99- DeployStressPackage $pkg $deployId $environment $repository $pushImages
104+ DeployStressPackage $pkg $deployId $environment $repository $pushImages $login
100105 }
101106
102107 Write-Host " Releases deployed by $deployId "
@@ -117,8 +122,9 @@ function DeployStressPackage(
117122 [object ]$pkg ,
118123 [string ]$deployId ,
119124 [string ]$environment ,
120- [string ]$repository ,
121- [boolean ]$pushImages
125+ [string ]$repositoryBase ,
126+ [switch ]$pushImages ,
127+ [switch ]$login
122128) {
123129 $registry = RunOrExitOnFailure az acr list - g $clusterGroup -- subscription $subscription - o json
124130 $registryName = ($registry | ConvertFrom-Json ).name
@@ -131,26 +137,23 @@ function DeployStressPackage(
131137 if ($LASTEXITCODE ) { return }
132138 }
133139
140+ $imageTag = " ${registryName} .azurecr.io"
141+ if ($repositoryBase ) {
142+ $imageTag += " /$repositoryBase "
143+ }
144+ $imageTag += " /$ ( $pkg.Namespace ) /$ ( $pkg.ReleaseName ) :${deployId} "
145+
134146 if ($pushImages ) {
135- $dockerFiles = Get-ChildItem " $ ( $pkg.Directory ) /Dockerfile*"
136- foreach ($dockerFile in $dockerFiles ) {
137- # Infer docker image name from parent directory name, if file is named `Dockerfile`
138- # or from suffix, is file is named like `Dockerfile.myimage` (for multiple dockerfiles).
139- $prefix , $imageName = $dockerFile.Name.Split (" ."
140- if (! $imageName ) {
141- $imageName = $dockerFile.Directory.Name
142- }
143- $imageTag = " ${registryName} .azurecr.io/$ ( $repository.ToLower ()) /$ ( $imageName ) :$deployId "
144- Write-Host " Building and pushing stress test docker image '$imageTag '"
145- Run docker build - t $imageTag -f $dockerFile.FullName $dockerFile.DirectoryName
146- if ($LASTEXITCODE ) { return }
147- Run docker push $imageTag
148- if ($LASTEXITCODE ) {
149- if ($PSCmdlet.ParameterSetName -ne ' DoLogin'
150- Write-Warning " If docker push is failing due to authentication issues, try calling this script with '-Login'"
151- }
152- return
147+ Write-Host " Building and pushing stress test docker image '$imageTag '"
148+ $dockerFile = Get-ChildItem " $ ( $pkg.Directory ) /Dockerfile"
149+ Run docker build - t $imageTag -f $dockerFile.FullName $dockerFile.DirectoryName
150+ if ($LASTEXITCODE ) { return }
151+ Run docker push $imageTag
152+ if ($LASTEXITCODE ) {
153+ if ($login ) {
154+ Write-Warning " If docker push is failing due to authentication issues, try calling this script with '-Login'"
153155 }
156+ return
154157 }
155158 }
156159
@@ -162,8 +165,7 @@ function DeployStressPackage(
162165 Run helm upgrade $pkg.ReleaseName $pkg.Directory `
163166 - n $pkg.Namespace `
164167 -- install `
165- -- set repository= $registryName.azurecr.io / $repository `
166- -- set tag= $deployId `
168+ -- set image= $imageTag `
167169 -- set stress- test-addons .env= $environment
168170 if ($LASTEXITCODE ) {
169171 # Issues like 'UPGRADE FAILED: another operation (install/upgrade/rollback) is in progress'
0 commit comments