data_disk.rb in fog-azure-rm-0.1.1

- old
+ new
@@ -21,57 +21,68 @@
   client_secret: azure_credentials['client_secret'],
   subscription_id: azure_credentials['subscription_id']
 )
 
 ########################################################################################################################
-######################                                 Prerequisites                              ######################
+######################                               Resource names                                #####################
 ########################################################################################################################
 
-rs.resource_groups.create(
-  name: 'TestRG-DD',
-  location: 'eastus'
-)
+time = current_time
+resource_group_name = "Blob-RG-#{time}"
+storage_account_name = "sa#{time}"
+container_name = 'vhds'
+test_container_name = 'disks'
 
-storage_account = storage.storage_accounts.create(
-  name: 'fogstorageac',
-  location: 'eastus',
-  resource_group: 'TestRG-DD',
-  account_type: 'Standard',
-  replication: 'LRS'
-)
-
-access_key = storage_account.get_access_keys[0].value
-Fog::Logger.debug access_key.inspect
-storage_data = Fog::Storage.new(
-  provider: 'AzureRM',
-  azure_storage_account_name: storage_account.name,
-  azure_storage_access_key: access_key
-)
-storage_data.directories.create(
-  name: 'vhds',
-  key: access_key
-)
-
 ########################################################################################################################
-######################                               Create Disk                                  ######################
+######################                                 Prerequisites                              ######################
 ########################################################################################################################
 
-storage_data.create_disk('datadisk1', options = {})
+begin
+  resource_group = rs.resource_groups.create(
+    name: resource_group_name,
+    location: LOCATION
+  )
 
-########################################################################################################################
-######################                                Delete Data Disk                            ######################
-########################################################################################################################
+  storage_account = storage.storage_accounts.create(
+    name: storage_account_name,
+    location: LOCATION,
+    resource_group: resource_group_name
+  )
 
-storage_data.delete_disk('datadisk1')
+  storage_data = Fog::Storage.new(
+    provider: 'AzureRM',
+    azure_storage_account_name: storage_account.name,
+    azure_storage_access_key: storage_account.get_access_keys[0].value
+  )
+  storage_data.directories.create(
+    key: container_name,
+    public: false
+  )
+  storage_data.directories.create(
+    key: test_container_name,
+    public: false
+  )
 
-########################################################################################################################
-######################                                   CleanUp                                  ######################
-########################################################################################################################
+  ########################################################################################################################
+  ######################                               Create Disk                                  ######################
+  ########################################################################################################################
 
-container = storage_data.directories.get('vhds')
-container.destroy
+  storage_data.create_disk('datadisk1', 10)
+  puts 'Created a disk in default container vhds'
 
-storage = storage.storage_accounts.get('TestRG-DD', 'fogstorageac')
-storage.destroy
+  storage_data.create_disk('datadisk2', 10, container_name: test_container_name)
+  puts 'Created a disk in non-default container'
 
-resource_group = rs.resource_groups.get('TestRG-DD')
-resource_group.destroy
+  ########################################################################################################################
+  ######################                                Delete Data Disk                            ######################
+  ########################################################################################################################
+
+  storage_data.delete_disk('datadisk1')
+  puts 'Deleted a disk in default container vhds'
+
+  storage_data.delete_disk('datadisk2', container_name: test_container_name)
+  puts 'Deleted a disk in non-default container'
+rescue
+  puts 'Integration Test for data disk is failing'
+ensure
+  resource_group.destroy unless resource_group.nil?
+end