compute.rb in fog-azure-rm-0.0.3

- old
+ new

@@ -1,7 +1,6 @@
 require 'fog/azurerm/core'
-# rubocop:disable LineLength
 module Fog
   module Compute
     # This class registers models, requests and collections
     class AzureRM < Fog::Service
       requires :tenant_id
@@ -22,16 +21,21 @@
       request :deallocate_virtual_machine
       request :power_off_virtual_machine
       request :redeploy_virtual_machine
       request :restart_virtual_machine
       request :start_virtual_machine
+      request :check_vm_status
+      request :attach_data_disk_to_vm
+      request :detach_data_disk_from_vm
 
       model_path 'fog/azurerm/models/compute'
       model :availability_set
       collection :availability_sets
       model :server
       collection :servers
+      model :data_disk
+
       # This class provides the mock implementation for unit tests.
       class Mock
         def initialize(_options = {})
           begin
             require 'azure_mgmt_compute'
@@ -44,17 +48,21 @@
       # This class provides the actual implemention for service calls.
       class Real
         def initialize(options)
           begin
             require 'azure_mgmt_compute'
+            require 'azure_mgmt_storage'
+            require 'azure/storage'
           rescue LoadError => e
             retry if require('rubygems')
             raise e.message
           end
 
           credentials = Fog::Credentials::AzureRM.get_credentials(options[:tenant_id], options[:client_id], options[:client_secret])
           @compute_mgmt_client = ::Azure::ARM::Compute::ComputeManagementClient.new(credentials)
           @compute_mgmt_client.subscription_id = options[:subscription_id]
+          @storage_mgmt_client = ::Azure::ARM::Storage::StorageManagementClient.new(credentials)
+          @storage_mgmt_client.subscription_id = options[:subscription_id]
         end
       end
     end
   end
 end