Trusted Launch Test

sdk/compute/Microsoft.Azure.Management.Compute/tests/ScenarioTests/VMInstanceViewPatchStatusTests.cs

@@ -1,4 +1,4 @@
-// Copyright (c) Microsoft Corporation. All rights reserved.
+// Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License. See License.txt in the project root for license information.
 
 using Compute.Tests;

sdk/compute/Microsoft.Azure.Management.Compute/tests/ScenarioTests/VMScenarioTests.cs

@@ -105,6 +105,27 @@ public void TestVMScenarioOperations_EncryptionAtHost()
                 Environment.SetEnvironmentVariable("AZURE_VM_TEST_LOCATION", originalTestLocation);
             }
         }
+
+        /// <summary>
+        /// To record this test case, you need to run it in region which support Encryption at host
+        /// </summary>
+        [Fact]
+        [Trait("Name", "TestVMScenarioOperations_TrustedLaunch")]
+        public void TestVMScenarioOperations_TrustedLaunch()
+        {
+            string originalTestLocation = Environment.GetEnvironmentVariable("AZURE_VM_TEST_LOCATION");
+            try
+            {
+                ImageReference image = new ImageReference(publisher: "MicrosoftWindowsServer", offer: "windowsserver-gen2preview-preview", version: "18363.592.2001092016", sku: "windows10-tvm");
+                Environment.SetEnvironmentVariable("AZURE_VM_TEST_LOCATION", "eastus2euap");
+                TestVMScenarioOperationsInternal("TestVMScenarioOperations_TrustedLaunch", vmSize: VirtualMachineSizeTypes.StandardD2sV3, hasManagedDisks: true,
+                    osDiskStorageAccountType: StorageAccountTypes.StandardSSDLRS, securityType: "TrustedLaunch", imageReference: image, validateListAvailableSize: false);
+            }
+            finally
+            {
+                Environment.SetEnvironmentVariable("AZURE_VM_TEST_LOCATION", originalTestLocation);
+            }
+        }
 
         /// <summary>
         /// To record this test case, you need to run it in region which support DiskEncryptionSet resource for the Disks
@@ -241,7 +262,7 @@ public void TestVMScenarioOperations_PpgScenario()
         private void TestVMScenarioOperationsInternal(string methodName, bool hasManagedDisks = false, IList<string> zones = null, string vmSize = "Standard_A1_v2",
             string osDiskStorageAccountType = "Standard_LRS", string dataDiskStorageAccountType = "Standard_LRS", bool? writeAcceleratorEnabled = null,
             bool hasDiffDisks = false, bool callUpdateVM = false, bool isPpgScenario = false, string diskEncryptionSetId = null, bool? encryptionAtHostEnabled = null,
-            bool isAutomaticPlacementOnDedicatedHostGroupScenario = false, ImageReference imageReference = null, bool validateListAvailableSize = true)
+            string securityType = null, bool isAutomaticPlacementOnDedicatedHostGroupScenario = false, ImageReference imageReference = null, bool validateListAvailableSize = true)
         {
             using (MockContext context = MockContext.Start(this.GetType(), methodName))
             {
@@ -280,7 +301,7 @@ private void TestVMScenarioOperationsInternal(string methodName, bool hasManaged
 
                     CreateVM(rgName, asName, storageAccountName, imageRef, out inputVM, hasManagedDisks: hasManagedDisks,hasDiffDisks: hasDiffDisks, vmSize: vmSize, osDiskStorageAccountType: osDiskStorageAccountType,
                         dataDiskStorageAccountType: dataDiskStorageAccountType, writeAcceleratorEnabled: writeAcceleratorEnabled, zones: zones, ppgName: ppgName, 
-                        diskEncryptionSetId: diskEncryptionSetId, encryptionAtHostEnabled: encryptionAtHostEnabled, dedicatedHostGroupReferenceId: dedicatedHostGroupReferenceId,
+                        diskEncryptionSetId: diskEncryptionSetId, encryptionAtHostEnabled: encryptionAtHostEnabled, securityType: securityType, dedicatedHostGroupReferenceId: dedicatedHostGroupReferenceId,
                         dedicatedHostGroupName: dedicatedHostGroupName, dedicatedHostName: dedicatedHostName);
 
                     // Instance view is not completely populated just after VM is provisioned. So we wait here for a few minutes to 
@@ -326,6 +347,12 @@ private void TestVMScenarioOperationsInternal(string methodName, bool hasManaged
                         listVMSizesResponse = m_CrpClient.AvailabilitySets.ListAvailableSizes(rgName, asName);
                         Helpers.ValidateVirtualMachineSizeListResponse(listVMSizesResponse, hasAZ: zones != null, writeAcceleratorEnabled: writeAcceleratorEnabled, hasDiffDisks: hasDiffDisks);
                     }
+
+                    if(securityType != null && securityType.Equals("TrustedLaunch"))
+                    {
+                        Assert.True(inputVM.SecurityProfile.UefiSettings.VTpmEnabled);
+                        Assert.True(inputVM.SecurityProfile.UefiSettings.SecureBootEnabled);
+                    }
 
                     if(isPpgScenario)
                     {

sdk/compute/Microsoft.Azure.Management.Compute/tests/ScenarioTests/VMTestBase.cs

@@ -235,6 +235,7 @@ protected VirtualMachine CreateVM(
             string ppgName = null,
             string diskEncryptionSetId = null,
             bool? encryptionAtHostEnabled = null,
+            string securityType = null,
             string dedicatedHostGroupReferenceId = null,
             string dedicatedHostGroupName = null,
             string dedicatedHostName = null)
@@ -296,6 +297,30 @@ protected VirtualMachine CreateVM(
                         EncryptionAtHost = encryptionAtHostEnabled.Value
                     };
                 }
+
+                if (securityType != null && securityType.Equals("TrustedLaunch"))
+                {
+                    if(inputVM.SecurityProfile != null)
+                    {
+                        inputVM.SecurityProfile.UefiSettings = new UefiSettings
+                        {
+                            VTpmEnabled = true,
+                            SecureBootEnabled = true
+                        };
+                    }
+                    else
+                    {
+                        inputVM.SecurityProfile = new SecurityProfile
+                        {
+                            UefiSettings = new UefiSettings
+                            {
+                                VTpmEnabled = true,
+                                SecureBootEnabled = true
+                            }
+                        };
+                    }
+
+                }
 
                 if (zones != null)
                 {