fog-azure-rm 0.2.7 → 0.3.0

data/lib/fog/azurerm/requests/compute/grant_access_to_managed_disk.rb

@@ -0,0 +1,30 @@
+module Fog
+  module Compute
+    class AzureRM
+      # Real class for Compute Request
+      class Real
+        def grant_access_to_managed_disk(resource_group_name, disk_name, access_type, duration_in_sec)
+          msg = "Granting access to Managed Disk: #{disk_name}"
+          Fog::Logger.debug msg
+          access_data = Azure::ARM::Compute::Models::GrantAccessData.new
+          access_data.access = access_type
+          access_data.duration_in_seconds = duration_in_sec
+          begin
+            access_uri = @compute_mgmt_client.disks.grant_access(resource_group_name, disk_name, access_data)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Access granted to managed disk: #{disk_name} successfully."
+          access_uri.access_sas
+        end
+      end
+
+      # Mock class for Compute Request
+      class Mock
+        def grant_access_to_managed_disk(*)
+          'ACCESS URI'
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/compute/list_managed_disks_by_rg.rb

@@ -0,0 +1,66 @@
+module Fog
+  module Compute
+    class AzureRM
+      # Real class for Compute Request
+      class Real
+        def list_managed_disks_by_rg(resource_group_name)
+          msg = "Listing all Managed Disks in resource group: #{resource_group_name}"
+          Fog::Logger.debug msg
+          begin
+            managed_disks = @compute_mgmt_client.disks.list_by_resource_group(resource_group_name)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug 'Managed Disks listed successfully.'
+          managed_disks
+        end
+      end
+
+      # Mock class for Compute Request
+      class Mock
+        def list_managed_disks_by_rg(*)
+          disks = [
+            {
+              'accountType' => 'Standard_LRS',
+              'properties' => {
+                'osType' => 'Windows',
+                'creationData' => {
+                  'createOption' => 'Empty'
+                },
+                'diskSizeGB' => 10,
+                'encryptionSettings' => {
+                  'enabled' => true,
+                  'diskEncryptionKey' => {
+                    'sourceVault' => {
+                      'id' => '/subscriptions/{subscriptionId}/resourceGroups/myResourceGroup/providers/Microsoft.KeyVault/vaults/myVMVault'
+                    },
+                    'secretUrl' => 'https://myvmvault.vault-int.azure-int.net/secrets/{secret}'
+                  },
+                  'keyEncryptionKey' => {
+                    'sourceVault' => {
+                      'id' => '/subscriptions/{subscriptionId}/resourceGroups/myResourceGroup/providers/Microsoft.KeyVault/vaults/myVMVault'
+                    },
+                    'keyUrl' => 'https://myvmvault.vault-int.azure-int.net/keys/{key}'
+                  }
+                },
+                'timeCreated' => '2016-12-28T02:46:21.3322041+00:00',
+                'provisioningState' => 'Succeeded',
+                'diskState' => 'Unattached'
+              },
+              'type' => 'Microsoft.Compute/disks',
+              'location' => 'westus',
+              'tags' => {
+                'department' => 'Development',
+                'project' => 'ManagedDisks'
+              },
+              'id' => '/subscriptions/{subscriptionId}/resourceGroups/myResourceGroup/providers/Microsoft.Compute/disks/myManagedDisk1',
+              'name' => 'myManagedDisk1'
+            }
+          ]
+          disk_mapper = Azure::ARM::Compute::Models::DiskList.mapper
+          @compute_mgmt_client.deserialize(disk_mapper, disks, 'result.body').value
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/compute/list_managed_disks_in_subscription.rb

@@ -0,0 +1,66 @@
+module Fog
+  module Compute
+    class AzureRM
+      # Real class for Compute Request
+      class Real
+        def list_managed_disks_in_subscription
+          msg = 'Listing all Managed Disks'
+          Fog::Logger.debug msg
+          begin
+            managed_disks = @compute_mgmt_client.disks.list
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug 'Managed Disks listed successfully.'
+          managed_disks
+        end
+      end
+
+      # Mock class for Compute Request
+      class Mock
+        def list_managed_disks_in_subscription
+          disks = [
+            {
+              'accountType' => 'Standard_LRS',
+              'properties' => {
+                'osType' => 'Windows',
+                'creationData' => {
+                  'createOption' => 'Empty'
+                },
+                'diskSizeGB' => 10,
+                'encryptionSettings' => {
+                  'enabled' => true,
+                  'diskEncryptionKey' => {
+                    'sourceVault' => {
+                      'id' => '/subscriptions/{subscriptionId}/resourceGroups/myResourceGroup/providers/Microsoft.KeyVault/vaults/myVMVault'
+                    },
+                    'secretUrl' => 'https://myvmvault.vault-int.azure-int.net/secrets/{secret}'
+                  },
+                  'keyEncryptionKey' => {
+                    'sourceVault' => {
+                      'id' => '/subscriptions/{subscriptionId}/resourceGroups/myResourceGroup/providers/Microsoft.KeyVault/vaults/myVMVault'
+                    },
+                    'keyUrl' => 'https://myvmvault.vault-int.azure-int.net/keys/{key}'
+                  }
+                },
+                'timeCreated' => '2016-12-28T02:46:21.3322041+00:00',
+                'provisioningState' => 'Succeeded',
+                'diskState' => 'Unattached'
+              },
+              'type' => 'Microsoft.Compute/disks',
+              'location' => 'westus',
+              'tags' => {
+                'department' => 'Development',
+                'project' => 'ManagedDisks'
+              },
+              'id' => '/subscriptions/{subscriptionId}/resourceGroups/myResourceGroup/providers/Microsoft.Compute/disks/myManagedDisk1',
+              'name' => 'myManagedDisk1'
+            }
+          ]
+          disk_mapper = Azure::ARM::Compute::Models::DiskList.mapper
+          @compute_mgmt_client.deserialize(disk_mapper, disks, 'result.body').value
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/compute/revoke_access_to_managed_disk.rb

@@ -0,0 +1,33 @@
+module Fog
+  module Compute
+    class AzureRM
+      # Real class for Compute Request
+      class Real
+        def revoke_access_to_managed_disk(resource_group_name, disk_name)
+          msg = "Revoking access to Managed Disk: #{disk_name}"
+          Fog::Logger.debug msg
+          begin
+            response = @compute_mgmt_client.disks.revoke_access(resource_group_name, disk_name)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Access revoked to managed disk: #{disk_name} successfully."
+          response
+        end
+      end
+
+      # Mock class for Compute Request
+      class Mock
+        def revoke_access_to_managed_disk(*)
+          response = {
+            'name' => 'revoke',
+            'status' => 200,
+            'error' => 'Error Details'
+          }
+          response_mapper = Azure::ARM::Compute::Models::OperationStatusResponse.mapper
+          @compute_mgmt_client.deserialize(response_mapper, response, 'result.body')
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/dns/check_zone_exists.rb

@@ -9,7 +9,11 @@ module Fog
           begin
             zone = @dns_client.zones.get(resource_group, name)
           rescue MsRestAzure::AzureOperationError => e
-            raise_azure_exception(e, msg)
+            if !e.body['error'].nil? && e.body['error']['code'] == ERROR_CODE_RESOURCE_NOT_FOUND
+              zone = nil
+            else
+              raise_azure_exception(e, msg)
+            end
           rescue => e
             Fog::Logger.debug e[:error][:code]
           end

data/lib/fog/azurerm/requests/network/check_net_sec_rule_exists.rb

@@ -11,7 +11,7 @@ module Fog
             Fog::Logger.debug "Network Security Rule #{security_rule_name} exists."
             true
           rescue MsRestAzure::AzureOperationError => e
-            if e.body['error']['code'] == 'ResourceNotFound'
+            if !e.body['error'].nil? && (e.body['error']['code'] == ERROR_CODE_RESOURCE_NOT_FOUND || e.body['error']['code'] == ERROR_CODE_NOT_FOUND)
               Fog::Logger.debug "Network Security Rule #{security_rule_name} doesn't exist."
               false
             else

data/lib/fog/azurerm/requests/network/check_subnet_exists.rb

@@ -11,7 +11,7 @@ module Fog
             Fog::Logger.debug "Subnet #{subnet_name} exists."
             true
           rescue MsRestAzure::AzureOperationError => e
-            if e.body['error']['code'] == 'ResourceNotFound'
+            if !e.body['error'].nil? && (e.body['error']['code'] == ERROR_CODE_RESOURCE_NOT_FOUND || e.body['error']['code'] == ERROR_CODE_NOT_FOUND)
               Fog::Logger.debug "Subnet #{subnet_name} doesn't exist."
               false
             else

data/lib/fog/azurerm/utilities/general.rb

@@ -47,22 +47,25 @@ def get_record_type(type)
 end
 
 def raise_azure_exception(exception, msg)
-  message = if exception.respond_to? 'body'
-              "Exception in #{msg} #{exception.body['error']['message'] unless exception.body['error']['message'].nil?} Type: #{exception.class}\n#{exception.backtrace.join("\n")}"
-            else
-              "#{exception.inspect}\n#{exception.backtrace.join("\n")}"
-            end
+  if exception.respond_to? 'body'
+    message = if exception.body['error'].nil?
+                exception.body['message']
+              else
+                exception.body['error']['message']
+              end
+    exception_message = "Exception in #{msg} #{message} Type: #{exception.class}\n#{exception.backtrace.join("\n")}"
+  else
+    "#{exception.inspect}\n#{exception.backtrace.join("\n")}"
+  end
+
   Fog::Logger.debug exception.backtrace
-  raise message
+  raise exception_message
 end
 
 # Make sure if input_params(Hash) contains all keys present in required_params(Array)
 def validate_params(required_params, input_params)
   missing_params = required_params.select { |param| param unless input_params.key?(param) }
-
-  if missing_params.any?
-    raise(ArgumentError, "Missing Parameters: #{missing_params.join(', ')} required for this operation")
-  end
+  raise(ArgumentError, "Missing Parameters: #{missing_params.join(', ')} required for this operation") if missing_params.any?
 end
 
 def get_resource_from_resource_id(resource_id, position)

data/lib/fog/azurerm/version.rb

@@ -1,5 +1,5 @@
 module Fog
   module AzureRM
-    VERSION = '0.2.7'.freeze
+    VERSION = '0.3.0'.freeze
   end
 end

data/test/api_stub.rb

@@ -5,6 +5,7 @@ module ApiStub
       autoload :Server, File.expand_path('api_stub/models/compute/server', __dir__)
       autoload :AvailabilitySet, File.expand_path('api_stub/models/compute/availability_set', __dir__)
       autoload :VirtualMachineExtension, File.expand_path('api_stub/models/compute/virtual_machine_extension', __dir__)
+      autoload :ManagedDisk, File.expand_path('api_stub/models/compute/managed_disk', __dir__)
     end
 
     module Resources
@@ -67,6 +68,7 @@ module ApiStub
       autoload :AvailabilitySet, File.expand_path('api_stub/requests/compute/availability_set', __dir__)
       autoload :VirtualMachine, File.expand_path('api_stub/requests/compute/virtual_machine', __dir__)
       autoload :VirtualMachineExtension, File.expand_path('api_stub/requests/compute/virtual_machine_extension', __dir__)
+      autoload :ManagedDisk, File.expand_path('api_stub/requests/compute/managed_disk', __dir__)
     end
 
     module Resources

data/test/api_stub/models/compute/managed_disk.rb

@@ -0,0 +1,59 @@
+module ApiStub
+  module Models
+    module Compute
+      # Mock class for ManagedDisk Model
+      class ManagedDisk
+        def self.create_managed_disk_response(sdk_compute_client)
+          disk = {
+            'accountType' => 'Standard_LRS',
+            'properties' => {
+              'osType' => 'Windows',
+              'creationData' => {
+                'createOption' => 'Empty'
+              },
+              'diskSizeGB' => 10,
+              'encryptionSettings' => {
+                'enabled' => true,
+                'diskEncryptionKey' => {
+                  'sourceVault' => {
+                    'id' => '/subscriptions/{subscriptionId}/resourceGroups/myResourceGroup/providers/Microsoft.KeyVault/vaults/myVMVault'
+                  },
+                  'secretUrl' => 'https://myvmvault.vault-int.azure-int.net/secrets/{secret}'
+                },
+                'keyEncryptionKey' => {
+                  'sourceVault' => {
+                    'id' => '/subscriptions/{subscriptionId}/resourceGroups/myResourceGroup/providers/Microsoft.KeyVault/vaults/myVMVault'
+                  },
+                  'keyUrl' => 'https://myvmvault.vault-int.azure-int.net/keys/{key}'
+                }
+              },
+              'timeCreated' => '2016-12-28T02:46:21.3322041+00:00',
+              'provisioningState' => 'Succeeded',
+              'diskState' => 'Unattached'
+            },
+            'type' => 'Microsoft.Compute/disks',
+            'location' => 'westus',
+            'tags' => {
+              'department' => 'Development',
+              'project' => 'ManagedDisks'
+            },
+            'id' => '/subscriptions/{subscriptionId}/resourceGroups/myResourceGroup/providers/Microsoft.Compute/disks/myManagedDisk1',
+            'name' => 'myManagedDisk1'
+          }
+          result_mapper = Azure::ARM::Compute::Models::Disk.mapper
+          sdk_compute_client.deserialize(result_mapper, disk, 'result.body')
+        end
+
+        def self.operation_status_response(sdk_compute_client)
+          response = {
+            'name' => 'xxxx-xxxxx-xxxx',
+            'status' => 'success',
+            'error' => 'ERROR'
+          }
+          response_mapper = Azure::ARM::Compute::Models::OperationStatusResponse.mapper
+          sdk_compute_client.deserialize(response_mapper, response, 'result.body')
+        end
+      end
+    end
+  end
+end

data/test/api_stub/requests/compute/managed_disk.rb

@@ -0,0 +1,102 @@
+module ApiStub
+  module Requests
+    module Compute
+      # Mock class for Managed Disk Requests
+      class ManagedDisk
+        def self.create_or_update_managed_disk_response(compute_client)
+          body = '{
+            "accountType": "Standard_LRS",
+            "properties": {
+              "osType": "Windows",
+              "creationData": {
+                "createOption": "Empty"
+              },
+              "diskSizeGB": 10,
+              "encryptionSettings": {
+                "enabled": true,
+                "diskEncryptionKey": {
+                  "sourceVault": {
+                    "id": "/subscriptions/{subscriptionId}/resourceGroups/myResourceGroup/providers/Microsoft.KeyVault/vaults/myVMVault"
+                  },
+                  "secretUrl": "https://myvmvault.vault-int.azure-int.net/secrets/{secret}"
+                },
+                "keyEncryptionKey": {
+                  "sourceVault": {
+                    "id": "/subscriptions/{subscriptionId}/resourceGroups/myResourceGroup/providers/Microsoft.KeyVault/vaults/myVMVault"
+                  },
+                  "keyUrl": "https://myvmvault.vault-int.azure-int.net/keys/{key}"
+                }
+              },
+              "timeCreated": "2016-12-28T02:46:21.3322041+00:00",
+              "provisioningState": "Succeeded",
+              "diskState": "Unattached"
+            },
+            "type": "Microsoft.Compute/disks",
+            "location": "westus",
+            "tags": {
+              "department": "Development",
+              "project": "ManagedDisks"
+            },
+            "id": "/subscriptions/{subscriptionId}/resourceGroups/myResourceGroup/providers/Microsoft.Compute/disks/myManagedDisk1",
+            "name": "myManagedDisk1"
+          }'
+          disk_mapper = Azure::ARM::Compute::Models::Disk.mapper
+          compute_client.deserialize(disk_mapper, Fog::JSON.decode(body), 'result.body')
+        end
+
+        def self.get_managed_disk_response(sdk_compute_client)
+          body = '{
+             "value": [ {
+                "accountType": "Standard_LRS",
+                "properties": {
+                  "osType": "Windows",
+                  "creationData": {
+                    "createOption": "Empty"
+                  },
+                  "diskSizeGB": 10,
+                  "encryptionSettings": {
+                    "enabled": true,
+                    "diskEncryptionKey": {
+                      "sourceVault": {
+                        "id": "/subscriptions/{subscriptionId}/resourceGroups/myResourceGroup/providers/Microsoft.KeyVault/vaults/myVMVault"
+                      },
+                      "secretUrl": "https://myvmvault.vault-int.azure-int.net/secrets/{secret}"
+                    },
+                    "keyEncryptionKey": {
+                      "sourceVault": {
+                        "id": "/subscriptions/{subscriptionId}/resourceGroups/myResourceGroup/providers/Microsoft.KeyVault/vaults/myVMVault"
+                      },
+                      "keyUrl": "https://myvmvault.vault-int.azure-int.net/keys/{key}"
+                    }
+                  },
+                  "timeCreated": "2016-12-28T02:46:21.3322041+00:00",
+                  "provisioningState": "Succeeded",
+                  "diskState": "Unattached"
+                },
+                "type": "Microsoft.Compute/disks",
+                "location": "westus",
+                "tags": {
+                  "department": "Development",
+                  "project": "ManagedDisks"
+                },
+                "id": "/subscriptions/{subscriptionId}/resourceGroups/myResourceGroup/providers/Microsoft.Compute/disks/myManagedDisk1",
+                "name": "myManagedDisk1"
+             } ]
+          }'
+          disk_mapper = Azure::ARM::Compute::Models::Disk.mapper
+          sdk_compute_client.deserialize(disk_mapper, Fog::JSON.decode(body), 'result.body')
+        end
+
+        def self.operation_status_response(sdk_compute_client)
+          response = {
+            'name' => 'xxxx-xxxxx-xxxx',
+            'status' => 'success',
+            'error' => 'ERROR'
+          }
+          response_mapper = Azure::ARM::Compute::Models::OperationStatusResponse.mapper
+          sdk_compute_client.deserialize(response_mapper, response, 'result.body')
+        end
+      end
+    end
+  end
+end