fog-azure-rm-temp 0.0.1

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

@@ -0,0 +1,46 @@
+module Fog
+  module Compute
+    class AzureRM
+      # Real class for Compute Request
+      class Real
+        def get_vm_extension(resource_group_name, virtual_machine_name, vm_extension_name)
+          msg = "Getting Extension #{vm_extension_name} of Virtual Machine #{virtual_machine_name} in Resource Group #{resource_group_name}"
+          Fog::Logger.debug msg
+          begin
+            vm_extension = @compute_mgmt_client.virtual_machine_extensions.get(resource_group_name, virtual_machine_name, vm_extension_name)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "#{msg} successful"
+          vm_extension
+        end
+      end
+
+      # Mock class for Compute Request
+      class Mock
+        def get_vm_extension(*)
+          body = {
+            'id' => '/subscriptions/########-####-####-####-############/resourceGroups/TestRG/providers/Microsoft.Compute/virtualMachines/TestVM/extensions/IaaSAntimalware',
+            'name' => 'IaasAntimalware',
+            'resource_group' => 'fog-test-rg',
+            'location' => 'West US',
+            'properties' => {
+              'publisher' => 'Microsoft.Azure.Security',
+              'type' => 'IaaSAntimalware',
+              'typeHandlerVersion' => '1.3',
+              'autoUpgradeMinorVersion' => 'true',
+              'forceUpdateTag' => 'RerunExtension',
+              'settings' => {
+                'AnitmalwareEnabled' => 'true',
+                'RealtimeProtectionEnabled' => 'false'
+              },
+              'protected_settings' => {}
+            }
+          }
+          extension_mapper = Azure::ARM::Compute::Models::VirtualMachineExtension.mapper
+          @compute_mgmt_client.deserialize(extension_mapper, body, 'result.body')
+        end
+      end
+    end
+  end
+end

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_availability_sets.rb

@@ -0,0 +1,40 @@
+module Fog
+  module Compute
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def list_availability_sets(resource_group)
+          msg = "Listing Availability Sets in Resource Group: #{resource_group}"
+          Fog::Logger.debug msg
+          begin
+            avail_sets = @compute_mgmt_client.availability_sets.list(resource_group)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Listing Availability Sets in Resource Group: #{resource_group} successful."
+          avail_sets.value
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def list_availability_sets(resource_group)
+          [
+            {
+              'id' => "/subscriptions/{subscription-id}/resourceGroups/#{resource_group}/providers/Microsoft.Compute/availabilitySets/test_availability_set",
+              'name' => 'test_availability_set',
+              'type' => 'Microsoft.Compute/availabilitySets',
+              'location' => 'westus',
+              'tags' =>  {},
+              'properties' =>
+              {
+                'platformUpdateDomainCount' => 5,
+                'platformFaultDomainCount' => 3,
+                'virtualMachines' => []
+              }
+            }
+          ]
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,64 @@
+module Fog
+  module Compute
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def list_available_sizes_for_virtual_machine(resource_group, name, async)
+          msg = "Listing sizes for Virtual Machine #{name} in Resource Group #{resource_group}"
+          Fog::Logger.debug msg
+          begin
+            if async
+              response = @compute_mgmt_client.virtual_machines.list_available_sizes_async(resource_group, name)
+            else
+              vm_sizes = @compute_mgmt_client.virtual_machines.list_available_sizes(resource_group, name)
+            end
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          if async
+            response
+          else
+            Fog::Logger.debug "Successfully listed sizes for Virtual Machine #{name} in Resource Group #{resource_group}"
+            vm_sizes.value
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def list_available_sizes_for_virtual_machine(*)
+          vm_sizes =
+            {
+              'value' => [
+                {
+                  'name' => 'Standard_A0',
+                  'numberOfCores' => 1,
+                  'osDiskSizeInMB' => 1_047_552,
+                  'resourceDiskSizeInMB' => 20_480,
+                  'memoryInMB' => 768,
+                  'maxDataDiskCount' => 1
+                },
+                {
+                  'name' => 'Standard_A1',
+                  'numberOfCores' => 1,
+                  'osDiskSizeInMB' => 1_047_552,
+                  'resourceDiskSizeInMB' => 71_680,
+                  'memoryInMB' => 1792,
+                  'maxDataDiskCount' => 2
+                },
+                {
+                  'name' => 'Standard_A2',
+                  'numberOfCores' => 2,
+                  'osDiskSizeInMB' => 1_047_552,
+                  'resourceDiskSizeInMB' => 138_240,
+                  'memoryInMB' => 3584,
+                  'maxDataDiskCount' => 4
+                }
+              ]
+            }
+          vm_mapper = Azure::ARM::Compute::Models::VirtualMachineSizeListResult.mapper
+          @compute_mgmt_client.deserialize(vm_mapper, vm_sizes, 'result.body').value
+        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/list_virtual_machines.rb

@@ -0,0 +1,69 @@
+module Fog
+  module Compute
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def list_virtual_machines(resource_group)
+          msg = "Listing Virtual Machines in Resource Group '#{resource_group}'"
+          Fog::Logger.debug msg
+          begin
+            virtual_machines = @compute_mgmt_client.virtual_machines.list_as_lazy(resource_group)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "listing Virtual Machines in Resource Group '#{resource_group}' successful"
+          virtual_machines.value
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def list_virtual_machines(*)
+          vms = {
+            'value' => [
+              {
+                'id' => '/subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.Compute/virtualMachines/fog-test-server',
+                'name' => 'fog-test-server',
+                'location' => 'West US',
+                'properties' => {
+                  'hardwareProfile' => {
+                    'vmSize' => 'Basic_A0'
+                  },
+                  'storageProfile' => {
+                    'imageReference' => {
+                      'publisher' => 'Canonical',
+                      'offer' => 'UbuntuServer',
+                      'sku' => '14.04.2-LTS',
+                      'version' => 'latest'
+                    },
+                    'osDisk' => {
+                      'name' => 'fog-test-server_os_disk',
+                      'vhd' => {
+                        'uri' => 'http://storageAccount.blob.core.windows.net/vhds/fog-test-server_os_disk.vhd'
+                      }
+                    }
+                  },
+                  'osProfile' => {
+                    'computerName' => 'fog-test-server',
+                    'adminUsername' => 'shaffan',
+                    'linuxConfiguration' => {
+                      'disablePasswordAuthentication' => false
+                    }
+                  },
+                  'networkProfile' => {
+                    'networkInterfaces' => [
+                      {
+                        'id' => '/subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.Network/networkInterfaces/fog-test-vnet'
+                      }
+                    ]
+                  }
+                }
+              }
+            ]
+          }
+          vm_mapper = Azure::ARM::Compute::Models::VirtualMachineListResult.mapper
+          @compute_mgmt_client.deserialize(vm_mapper, vms, 'result.body').value
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,35 @@
+module Fog
+  module Compute
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def power_off_virtual_machine(resource_group, name, async)
+          msg = "Powering off Virtual Machine #{name} in Resource Group #{resource_group}"
+          Fog::Logger.debug msg
+          begin
+            if async 
+              response = @compute_mgmt_client.virtual_machines.power_off_async(resource_group, name)
+            else
+              @compute_mgmt_client.virtual_machines.power_off(resource_group, name)
+            end
+          rescue  MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          if async
+            response
+          else
+            Fog::Logger.debug "Virtual Machine #{name} Powered off Successfully."
+            true
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def power_off_virtual_machine(*)
+          Fog::Logger.debug 'Virtual Machine fog-test-server from Resource group fog-test-rg Powered off successfully.'
+          true
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,35 @@
+module Fog
+  module Compute
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def redeploy_virtual_machine(resource_group, name, async)
+          msg = "Redeploying Virtual Machine #{name} in Resource Group #{resource_group}."
+          Fog::Logger.debug msg
+          begin
+            if async
+              response = @compute_mgmt_client.virtual_machines.redeploy_async(resource_group, name)
+            else
+              @compute_mgmt_client.virtual_machines.redeploy(resource_group, name)
+            end
+          rescue  MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          if async
+            response
+          else
+            Fog::Logger.debug "Virtual Machine #{name} Redeployed Successfully."
+            true
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def redeploy_virtual_machine(*)
+          Fog::Logger.debug 'Virtual Machine fog-test-server from Resource group fog-test-rg Redeployed successfully.'
+          true
+        end
+      end
+    end
+  end
+end