From 415c639a42c294d23316a3f3c1bca06f2a086ec8 Mon Sep 17 00:00:00 2001
From: Wes Haggard <weshaggard@users.noreply.github.com>
Date: Mon, 28 Oct 2024 09:58:54 -0700
Subject: [PATCH 1/2] Run perf tests under federated auth

Run the tests under the same federated auth used to deploy the tests and setup the variables needed to configure AzurePipelineCredential.
---
 eng/common/pipelines/templates/jobs/perf.yml | 37 +++++++++++++-------
 1 file changed, 24 insertions(+), 13 deletions(-)

diff --git a/eng/common/pipelines/templates/jobs/perf.yml b/eng/common/pipelines/templates/jobs/perf.yml
index 20ad5de1c789..3b4afc032f16 100644
--- a/eng/common/pipelines/templates/jobs/perf.yml
+++ b/eng/common/pipelines/templates/jobs/perf.yml
@@ -122,23 +122,34 @@ jobs:
       ResourceType: perf
       ServiceConnection: azure-sdk-tests-public
 
-  - script: >-
-      dotnet run -- run
-      --language ${{ parameters.Language }}
-      --language-version ${{ parameters.LanguageVersion }}
-      --repo-root $(System.DefaultWorkingDirectory)
-      --tests-file $(System.DefaultWorkingDirectory)/sdk/${{ parameters.ServiceDirectory }}/perf-tests.yml
-      --package-versions "${{ parameters.PackageVersions }}"
-      --tests "${{ parameters.Tests }}"
-      --arguments "${{ parameters.Arguments }}"
-      --iterations ${{ parameters.Iterations }}
-      $(Profile)
-      ${{ parameters.AdditionalArguments }}
+  - task: AzurePowerShell@5
+    inputs:
+      azureSubscription: azure-sdk-tests-public
+      azurePowerShellVersion: LatestVersion
+      pwsh: true
+      ScriptType: InlineScript
+      Inline: >-
+        $account = (Get-AzContext).Account;
+        $env:AZURESUBSCRIPTION_CLIENT_ID = $account.Id;
+        $env:AZURESUBSCRIPTION_TENANT_ID = $account.Tenants;
+
+        dotnet run -- run
+        --language ${{ parameters.Language }}
+        --language-version ${{ parameters.LanguageVersion }}
+        --repo-root $(System.DefaultWorkingDirectory)
+        --tests-file $(System.DefaultWorkingDirectory)/sdk/${{ parameters.ServiceDirectory }}/perf-tests.yml
+        --package-versions "${{ parameters.PackageVersions }}"
+        --tests "${{ parameters.Tests }}"
+        --arguments "${{ parameters.Arguments }}"
+        --iterations ${{ parameters.Iterations }}
+        $(Profile)
+        ${{ parameters.AdditionalArguments }}
     workingDirectory: azure-sdk-tools/tools/perf-automation/Azure.Sdk.Tools.PerfAutomation
+    displayName: Run perf tests
     env:
+      SYSTEM_ACCESSTOKEN: $(System.AccessToken)
       ${{ each var in parameters.EnvVars }}:
         ${{ var.key }}: ${{ var.value }}
-    displayName: Run perf tests
 
   - pwsh: |
       get-content results.txt

From 95dccbfd495f955f9865312cda6757f39720b761 Mon Sep 17 00:00:00 2001
From: Wes Haggard <weshaggard@users.noreply.github.com>
Date: Mon, 28 Oct 2024 17:58:46 -0700
Subject: [PATCH 2/2] Fix indention for working directory

---
 eng/common/pipelines/templates/jobs/perf.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/eng/common/pipelines/templates/jobs/perf.yml b/eng/common/pipelines/templates/jobs/perf.yml
index 3b4afc032f16..bd53833282c2 100644
--- a/eng/common/pipelines/templates/jobs/perf.yml
+++ b/eng/common/pipelines/templates/jobs/perf.yml
@@ -144,7 +144,7 @@ jobs:
         --iterations ${{ parameters.Iterations }}
         $(Profile)
         ${{ parameters.AdditionalArguments }}
-    workingDirectory: azure-sdk-tools/tools/perf-automation/Azure.Sdk.Tools.PerfAutomation
+      workingDirectory: azure-sdk-tools/tools/perf-automation/Azure.Sdk.Tools.PerfAutomation
     displayName: Run perf tests
     env:
       SYSTEM_ACCESSTOKEN: $(System.AccessToken)