/sdk/compute/mgmt/src/test/java/com/azure/management/compute/VirtualMachineBootDiagnosticsTests.java

http://github.com/WindowsAzure/azure-sdk-for-java · Java · 285 lines · 252 code · 27 blank · 6 comment · 0 complexity · 579a66d073bd80657fa2c42600cd7b4d MD5 · raw file 

    

  1. // Copyright (c) Microsoft Corporation. All rights reserved.
    2. 

  2. // Licensed under the MIT License.
    3. 

  3. 

  4. package com.azure.management.compute;
    5. 

  5. 

  6. import com.azure.core.http.HttpPipeline;
    7. 

  7. import com.azure.management.resources.fluentcore.arm.Region;
    8. 

  8. import com.azure.management.resources.fluentcore.model.Creatable;
    9. 

  9. import com.azure.management.resources.fluentcore.profile.AzureProfile;
    10. 

  10. import com.azure.management.storage.StorageAccount;
    11. 

  11. import org.junit.jupiter.api.Assertions;
    12. 

  12. import org.junit.jupiter.api.Test;
    13. 

  13. 

  14. public class VirtualMachineBootDiagnosticsTests extends ComputeManagementTest {
    15. 

  15.     private String rgName = "";
    16. 

  16.     private final Region region = Region.US_SOUTH_CENTRAL;
    17. 

  17.     private final String vmName = "javavm";
    18. 

  18. 

  19.     @Override
    20. 

  20.     protected void initializeClients(HttpPipeline httpPipeline, AzureProfile profile) {
    21. 

  21.         rgName = generateRandomResourceName("javacsmrg", 15);
    22. 

  22.         super.initializeClients(httpPipeline, profile);
    23. 

  23.     }
    24. 

  24. 

  25.     @Override
    26. 

  26.     protected void cleanUpResources() {
    27. 

  27.         resourceManager.resourceGroups().deleteByName(rgName);
    28. 

  28.     }
    29. 

  29. 

  30.     @Test
    31. 

  31.     public void canEnableBootDiagnosticsWithImplicitStorageOnManagedVMCreation() {
    32. 

  32.         VirtualMachine virtualMachine =
    33. 

  33.             computeManager
    34. 

  34.                 .virtualMachines()
    35. 

  35.                 .define(vmName)
    36. 

  36.                 .withRegion(region)
    37. 

  37.                 .withNewResourceGroup(rgName)
    38. 

  38.                 .withNewPrimaryNetwork("10.0.0.0/28")
    39. 

  39.                 .withPrimaryPrivateIPAddressDynamic()
    40. 

  40.                 .withoutPrimaryPublicIPAddress()
    41. 

  41.                 .withPopularLinuxImage(KnownLinuxVirtualMachineImage.UBUNTU_SERVER_16_04_LTS)
    42. 

  42.                 .withRootUsername("Foo12")
    43. 

  43.                 .withRootPassword("abc!@#F0orL")
    44. 

  44.                 .withBootDiagnostics()
    45. 

  45.                 .create();
    46. 

  46. 

  47.         Assertions.assertNotNull(virtualMachine);
    48. 

  48.         Assertions.assertTrue(virtualMachine.isBootDiagnosticsEnabled());
    49. 

  49.         Assertions.assertNotNull(virtualMachine.bootDiagnosticsStorageUri());
    50. 

  50.     }
    51. 

  51. 

  52.     @Test
    53. 

  53.     public void canEnableBootDiagnosticsWithCreatableStorageOnManagedVMCreation() {
    54. 

  54.         final String storageName = sdkContext.randomResourceName("st", 14);
    55. 

  55.         Creatable<StorageAccount> creatableStorageAccount =
    56. 

  56.             storageManager.storageAccounts().define(storageName).withRegion(region).withNewResourceGroup(rgName);
    57. 

  57. 

  58.         VirtualMachine virtualMachine =
    59. 

  59.             computeManager
    60. 

  60.                 .virtualMachines()
    61. 

  61.                 .define(vmName)
    62. 

  62.                 .withRegion(region)
    63. 

  63.                 .withNewResourceGroup(rgName)
    64. 

  64.                 .withNewPrimaryNetwork("10.0.0.0/28")
    65. 

  65.                 .withPrimaryPrivateIPAddressDynamic()
    66. 

  66.                 .withoutPrimaryPublicIPAddress()
    67. 

  67.                 .withPopularLinuxImage(KnownLinuxVirtualMachineImage.UBUNTU_SERVER_16_04_LTS)
    68. 

  68.                 .withRootUsername("Foo12")
    69. 

  69.                 .withRootPassword("abc!@#F0orL")
    70. 

  70.                 .withBootDiagnostics(creatableStorageAccount)
    71. 

  71.                 .create();
    72. 

  72.         Assertions.assertNotNull(virtualMachine);
    73. 

  73.         Assertions.assertTrue(virtualMachine.isBootDiagnosticsEnabled());
    74. 

  74.         Assertions.assertNotNull(virtualMachine.bootDiagnosticsStorageUri());
    75. 

  75.         Assertions.assertTrue(virtualMachine.bootDiagnosticsStorageUri().contains(storageName));
    76. 

  76.     }
    77. 

  77. 

  78.     @Test
    79. 

  79.     public void canEnableBootDiagnosticsWithExplicitStorageOnManagedVMCreation() {
    80. 

  80.         final String storageName = sdkContext.randomResourceName("st", 14);
    81. 

  81.         StorageAccount storageAccount =
    82. 

  82.             storageManager
    83. 

  83.                 .storageAccounts()
    84. 

  84.                 .define(storageName)
    85. 

  85.                 .withRegion(region)
    86. 

  86.                 .withNewResourceGroup(rgName)
    87. 

  87.                 .create();
    88. 

  88. 

  89.         VirtualMachine virtualMachine =
    90. 

  90.             computeManager
    91. 

  91.                 .virtualMachines()
    92. 

  92.                 .define(vmName)
    93. 

  93.                 .withRegion(region)
    94. 

  94.                 .withNewResourceGroup(rgName)
    95. 

  95.                 .withNewPrimaryNetwork("10.0.0.0/28")
    96. 

  96.                 .withPrimaryPrivateIPAddressDynamic()
    97. 

  97.                 .withoutPrimaryPublicIPAddress()
    98. 

  98.                 .withPopularLinuxImage(KnownLinuxVirtualMachineImage.UBUNTU_SERVER_16_04_LTS)
    99. 

  99.                 .withRootUsername("Foo12")
    100. 

  100.                 .withRootPassword("abc!@#F0orL")
    101. 

  101.                 .withBootDiagnostics(storageAccount)
    102. 

  102.                 .create();
    103. 

  103. 

  104.         Assertions.assertNotNull(virtualMachine);
    105. 

  105.         Assertions.assertTrue(virtualMachine.isBootDiagnosticsEnabled());
    106. 

  106.         Assertions.assertNotNull(virtualMachine.bootDiagnosticsStorageUri());
    107. 

  107.         Assertions.assertTrue(virtualMachine.bootDiagnosticsStorageUri().contains(storageName));
    108. 

  108.     }
    109. 

  109. 

  110.     @Test
    111. 

  111.     public void canDisableBootDiagnostics() {
    112. 

  112.         VirtualMachine virtualMachine =
    113. 

  113.             computeManager
    114. 

  114.                 .virtualMachines()
    115. 

  115.                 .define(vmName)
    116. 

  116.                 .withRegion(region)
    117. 

  117.                 .withNewResourceGroup(rgName)
    118. 

  118.                 .withNewPrimaryNetwork("10.0.0.0/28")
    119. 

  119.                 .withPrimaryPrivateIPAddressDynamic()
    120. 

  120.                 .withoutPrimaryPublicIPAddress()
    121. 

  121.                 .withPopularLinuxImage(KnownLinuxVirtualMachineImage.UBUNTU_SERVER_16_04_LTS)
    122. 

  122.                 .withRootUsername("Foo12")
    123. 

  123.                 .withRootPassword("abc!@#F0orLX")
    124. 

  124.                 .withBootDiagnostics()
    125. 

  125.                 .create();
    126. 

  126. 

  127.         Assertions.assertNotNull(virtualMachine);
    128. 

  128.         Assertions.assertTrue(virtualMachine.isBootDiagnosticsEnabled());
    129. 

  129.         Assertions.assertNotNull(virtualMachine.bootDiagnosticsStorageUri());
    130. 

  130. 

  131.         virtualMachine.update().withoutBootDiagnostics().apply();
    132. 

  132. 

  133.         Assertions.assertFalse(virtualMachine.isBootDiagnosticsEnabled());
    134. 

  134.         // Disabling boot diagnostics will not remove the storage uri from the vm payload.
    135. 

  135.         Assertions.assertNotNull(virtualMachine.bootDiagnosticsStorageUri());
    136. 

  136.     }
    137. 

  137. 

  138.     @Test
    139. 

  139.     public void bootDiagnosticsShouldUsesOSUnManagedDiskImplicitStorage() {
    140. 

  140.         VirtualMachine virtualMachine =
    141. 

  141.             computeManager
    142. 

  142.                 .virtualMachines()
    143. 

  143.                 .define(vmName)
    144. 

  144.                 .withRegion(region)
    145. 

  145.                 .withNewResourceGroup(rgName)
    146. 

  146.                 .withNewPrimaryNetwork("10.0.0.0/28")
    147. 

  147.                 .withPrimaryPrivateIPAddressDynamic()
    148. 

  148.                 .withoutPrimaryPublicIPAddress()
    149. 

  149.                 .withPopularLinuxImage(KnownLinuxVirtualMachineImage.UBUNTU_SERVER_16_04_LTS)
    150. 

  150.                 .withRootUsername("Foo12")
    151. 

  151.                 .withRootPassword("abc!@#F0orL")
    152. 

  152.                 .withUnmanagedDisks() // The implicit storage account for OS disk should be used for boot diagnostics as
    153. 

  153.                                       // well
    154. 

  154.                 .withBootDiagnostics()
    155. 

  155.                 .create();
    156. 

  156. 

  157.         Assertions.assertNotNull(virtualMachine);
    158. 

  158.         Assertions.assertTrue(virtualMachine.isBootDiagnosticsEnabled());
    159. 

  159.         Assertions.assertNotNull(virtualMachine.bootDiagnosticsStorageUri());
    160. 

  160.         Assertions.assertNotNull(virtualMachine.osUnmanagedDiskVhdUri());
    161. 

  161.         Assertions
    162. 

  162.             .assertTrue(
    163. 

  163.                 virtualMachine
    164. 

  164.                     .osUnmanagedDiskVhdUri()
    165. 

  165.                     .toLowerCase()
    166. 

  166.                     .startsWith(virtualMachine.bootDiagnosticsStorageUri().toLowerCase()));
    167. 

  167.     }
    168. 

  168. 

  169.     @Test
    170. 

  170.     public void bootDiagnosticsShouldUseUnManagedDisksExplicitStorage() {
    171. 

  171.         final String storageName = sdkContext.randomResourceName("st", 14);
    172. 

  172.         StorageAccount storageAccount =
    173. 

  173.             storageManager
    174. 

  174.                 .storageAccounts()
    175. 

  175.                 .define(storageName)
    176. 

  176.                 .withRegion(region)
    177. 

  177.                 .withNewResourceGroup(rgName)
    178. 

  178.                 .create();
    179. 

  179. 

  180.         VirtualMachine virtualMachine =
    181. 

  181.             computeManager
    182. 

  182.                 .virtualMachines()
    183. 

  183.                 .define(vmName)
    184. 

  184.                 .withRegion(region)
    185. 

  185.                 .withNewResourceGroup(rgName)
    186. 

  186.                 .withNewPrimaryNetwork("10.0.0.0/28")
    187. 

  187.                 .withPrimaryPrivateIPAddressDynamic()
    188. 

  188.                 .withoutPrimaryPublicIPAddress()
    189. 

  189.                 .withPopularLinuxImage(KnownLinuxVirtualMachineImage.UBUNTU_SERVER_16_04_LTS)
    190. 

  190.                 .withRootUsername("Foo12")
    191. 

  191.                 .withRootPassword("abc!@#F0orL")
    192. 

  192.                 .withUnmanagedDisks()
    193. 

  193.                 .withBootDiagnostics()
    194. 

  194.                 .withExistingStorageAccount(
    195. 

  195.                     storageAccount) // This storage account must be shared by disk and boot diagnostics
    196. 

  196.                 .create();
    197. 

  197. 

  198.         Assertions.assertNotNull(virtualMachine);
    199. 

  199.         Assertions.assertTrue(virtualMachine.isBootDiagnosticsEnabled());
    200. 

  200.         Assertions.assertNotNull(virtualMachine.bootDiagnosticsStorageUri());
    201. 

  201.         Assertions.assertTrue(virtualMachine.bootDiagnosticsStorageUri().contains(storageName));
    202. 

  202.     }
    203. 

  203. 

  204.     @Test
    205. 

  205.     public void canEnableBootDiagnosticsWithImplicitStorageOnUnManagedVMCreation() {
    206. 

  206.         VirtualMachine virtualMachine1 =
    207. 

  207.             computeManager
    208. 

  208.                 .virtualMachines()
    209. 

  209.                 .define(vmName)
    210. 

  210.                 .withRegion(region)
    211. 

  211.                 .withNewResourceGroup(rgName)
    212. 

  212.                 .withNewPrimaryNetwork("10.0.0.0/28")
    213. 

  213.                 .withPrimaryPrivateIPAddressDynamic()
    214. 

  214.                 .withoutPrimaryPublicIPAddress()
    215. 

  215.                 .withPopularLinuxImage(KnownLinuxVirtualMachineImage.UBUNTU_SERVER_16_04_LTS)
    216. 

  216.                 .withRootUsername("Foo12")
    217. 

  217.                 .withRootPassword("abc!@#F0orL")
    218. 

  218.                 .withUnmanagedDisks()
    219. 

  219.                 .create();
    220. 

  220. 

  221.         String osDiskVhd = virtualMachine1.osUnmanagedDiskVhdUri();
    222. 

  222.         Assertions.assertNotNull(osDiskVhd);
    223. 

  223.         computeManager.virtualMachines().deleteById(virtualMachine1.id());
    224. 

  224. 

  225.         VirtualMachine virtualMachine2 =
    226. 

  226.             computeManager
    227. 

  227.                 .virtualMachines()
    228. 

  228.                 .define(vmName)
    229. 

  229.                 .withRegion(region)
    230. 

  230.                 .withNewResourceGroup(rgName)
    231. 

  231.                 .withNewPrimaryNetwork("10.0.0.0/28")
    232. 

  232.                 .withPrimaryPrivateIPAddressDynamic()
    233. 

  233.                 .withoutPrimaryPublicIPAddress()
    234. 

  234.                 .withSpecializedOSUnmanagedDisk(osDiskVhd, OperatingSystemTypes.LINUX)
    235. 

  235.                 .withBootDiagnostics() // A new storage account should be created and used
    236. 

  236.                 .create();
    237. 

  237. 

  238.         Assertions.assertNotNull(virtualMachine2);
    239. 

  239.         Assertions.assertTrue(virtualMachine2.isBootDiagnosticsEnabled());
    240. 

  240.         Assertions.assertNotNull(virtualMachine2.bootDiagnosticsStorageUri());
    241. 

  241.         Assertions.assertNotNull(virtualMachine2.osUnmanagedDiskVhdUri());
    242. 

  242.         Assertions
    243. 

  243.             .assertFalse(
    244. 

  244.                 virtualMachine2
    245. 

  245.                     .osUnmanagedDiskVhdUri()
    246. 

  246.                     .toLowerCase()
    247. 

  247.                     .startsWith(virtualMachine2.bootDiagnosticsStorageUri().toLowerCase()));
    248. 

  248.     }
    249. 

  249. 

  250.     @Test
    251. 

  251.     public void canEnableBootDiagnosticsWithCreatableStorageOnUnManagedVMCreation() {
    252. 

  252.         final String storageName = sdkContext.randomResourceName("st", 14);
    253. 

  253.         Creatable<StorageAccount> creatableStorageAccount =
    254. 

  254.             storageManager.storageAccounts().define(storageName).withRegion(region).withNewResourceGroup(rgName);
    255. 

  255. 

  256.         VirtualMachine virtualMachine =
    257. 

  257.             computeManager
    258. 

  258.                 .virtualMachines()
    259. 

  259.                 .define(vmName)
    260. 

  260.                 .withRegion(region)
    261. 

  261.                 .withNewResourceGroup(rgName)
    262. 

  262.                 .withNewPrimaryNetwork("10.0.0.0/28")
    263. 

  263.                 .withPrimaryPrivateIPAddressDynamic()
    264. 

  264.                 .withoutPrimaryPublicIPAddress()
    265. 

  265.                 .withPopularLinuxImage(KnownLinuxVirtualMachineImage.UBUNTU_SERVER_16_04_LTS)
    266. 

  266.                 .withRootUsername("Foo12")
    267. 

  267.                 .withRootPassword("abc!@#F0orL")
    268. 

  268.                 .withUnmanagedDisks()
    269. 

  269.                 .withBootDiagnostics(
    270. 

  270.                     creatableStorageAccount) // This storage account should be used for BDiagnostics not OS disk storage
    271. 

  271.                                              // account
    272. 

  272.                 .create();
    273. 

  273.         Assertions.assertNotNull(virtualMachine);
    274. 

  274.         Assertions.assertTrue(virtualMachine.isBootDiagnosticsEnabled());
    275. 

  275.         Assertions.assertNotNull(virtualMachine.bootDiagnosticsStorageUri());
    276. 

  276.         Assertions.assertTrue(virtualMachine.bootDiagnosticsStorageUri().contains(storageName));
    277. 

  277.         // There should be a different storage account created for the OS Disk
    278. 

  278.         Assertions
    279. 

  279.             .assertFalse(
    280. 

  280.                 virtualMachine
    281. 

  281.                     .osUnmanagedDiskVhdUri()
    282. 

  282.                     .toLowerCase()
    283. 

  283.                     .startsWith(virtualMachine.bootDiagnosticsStorageUri().toLowerCase()));
    284. 

  284.     }
    285. 

  285. }
    286.