Merge pull request Azure#6198 from maddieclayton/recordAks

Fix and rerecord Aks test

Author: Maddie Clayton

src/ResourceManager/Aks/Commands.Aks.Test/ScenarioTests/KubernetesTests.cs

@@ -15,9 +15,8 @@ public KubernetesTests(ITestOutputHelper output)
             TestExecutionHelpers.SetUpSessionAndProfile();
         }
 
-        [Fact(Skip = "Need service team to re-record test after changes to the ClientRuntime.")]
+        [Fact]
         [Trait(Category.AcceptanceType, Category.CheckIn)]
-        [Trait("Re-record", "ClientRuntime changes")]
         public void TestAzureKubernetes()
         {
             TestController.NewInstance.RunPowerShellTest("Test-AzureRmKubernetes");

src/ResourceManager/Aks/Commands.Aks.Test/ScenarioTests/TestController.cs

@@ -74,6 +74,27 @@ public void RunPowerShellTest(params string[] scripts)
                     string jsonOutput = @"{""" + subscription + @""":{ ""service_principal"":""foo"",""client_secret"":""bar""}}";
                     AzureSession.Instance.DataStore.WriteFile(Path.Combine(home, ".azure", "acsServicePrincipal.json"), jsonOutput);
                 }
+                else if (HttpMockServer.GetCurrentMode() == HttpRecorderMode.Record)
+                {
+                    AzureSession.Instance.DataStore = new MemoryDataStore();
+                    var home = Environment.GetFolderPath(Environment.SpecialFolder.UserProfile);
+                    var dir = Path.GetDirectoryName(new Uri(Assembly.GetExecutingAssembly().CodeBase).AbsolutePath);
+                    var subscription = HttpMockServer.Variables["SubscriptionId"];
+                    var currentEnvironment = TestEnvironmentFactory.GetTestEnvironment();
+                    string spn = null;
+                    string spnSecret = null;
+                    if (currentEnvironment.ConnectionString.KeyValuePairs.ContainsKey("ServicePrincipal"))
+                    {
+                        spn = currentEnvironment.ConnectionString.KeyValuePairs["ServicePrincipal"];
+                    }
+                    if (currentEnvironment.ConnectionString.KeyValuePairs.ContainsKey("ServicePrincipalSecret"))
+                    {
+                        spnSecret = currentEnvironment.ConnectionString.KeyValuePairs["ServicePrincipalSecret"];
+                    }
+                    AzureSession.Instance.DataStore.WriteFile(Path.Combine(home, ".ssh", "id_rsa.pub"), File.ReadAllText(dir + "/Fixtures/id_rsa.pub"));
+                    string jsonOutput = @"{""" + subscription + @""":{ ""service_principal"":""" + spn + @""",""client_secret"":"""+ spnSecret + @"""}}";
+                    AzureSession.Instance.DataStore.WriteFile(Path.Combine(home, ".azure", "acsServicePrincipal.json"), jsonOutput);
+                }
 
                 helper.RunPowerShellTest(scripts);
             }