fix issue with environment variables getting cleared (#36513)

Author: scbedd

eng/pipelines/templates/steps/build-test.yml

@@ -82,16 +82,17 @@ steps:
         pwsh: true
         ScriptType: InlineScript
         Inline: >-
+          $account = (Get-AzContext).Account;
+          $env:AZURESUBSCRIPTION_CLIENT_ID = $account.Id;
+          $env:AZURESUBSCRIPTION_TENANT_ID = $account.Tenants;
+
           if ($IsWindows) {
             . $(VENV_LOCATION)/Scripts/Activate.ps1
           }
           else {
             . $(VENV_LOCATION)/bin/activate.ps1
           }
           Write-Host (Get-Command python).Source
-          $account = (Get-AzContext).Account;
-          $env:AZURESUBSCRIPTION_CLIENT_ID = $account.Id;
-          $env:AZURESUBSCRIPTION_TENANT_ID = $account.Tenants;
 
           python scripts/devops_tasks/dispatch_tox.py
           "$(TargetingString)"
@@ -166,16 +167,17 @@ steps:
         pwsh: true
         ScriptType: InlineScript
         Inline: >-
+          $account = (Get-AzContext).Account;
+          $env:AZURESUBSCRIPTION_CLIENT_ID = $account.Id;
+          $env:AZURESUBSCRIPTION_TENANT_ID = $account.Tenants;
+
           if ($IsWindows) {
             . $(VENV_LOCATION)/Scripts/Activate.ps1
           }
           else {
             . $(VENV_LOCATION)/bin/activate.ps1
           }
           Write-Host (Get-Command python).Source
-          $account = (Get-AzContext).Account;
-          $env:AZURESUBSCRIPTION_CLIENT_ID = $account.Id;
-          $env:AZURESUBSCRIPTION_TENANT_ID = $account.Tenants;
 
           scripts/devops_tasks/dispatch_tox.py
           "$(TargetingString)"