fog-azure-rm-temp 0.0.3 → 0.0.4

data/test/models/{storage → compute}/test_data_disk.rb

@@ -3,8 +3,8 @@ require File.expand_path '../../test_helper', __dir__
 # Test class for Data Disk Model
 class TestDataDisk < Minitest::Test
   def setup
-    @service = Fog::Storage::AzureRM.new(credentials)
-    @data_disk = Fog::Storage::AzureRM::DataDisk.new
+    @service = Fog::Compute::AzureRM.new(credentials)
+    @data_disk = Fog::Compute::AzureRM::DataDisk.new
   end
 
   def test_model_attributes

data/test/models/compute/test_server.rb

@@ -20,7 +20,9 @@ class TestServer < Minitest::Test
       :redeploy,
       :list_available_sizes,
       :attach_data_disk,
-      :detach_data_disk
+      :detach_data_disk,
+      :attach_managed_disk,
+      :detach_managed_disk
     ]
     methods.each do |method|
       assert_respond_to @server, method
@@ -198,4 +200,28 @@ class TestServer < Minitest::Test
       assert_instance_of Fog::AzureRM::AsyncResponse, @server.detach_data_disk('disk1', true)
     end
   end
+
+  def test_attach_managed_disk_response
+    response = ApiStub::Models::Compute::Server.attach_managed_disk_response(@compute_client)
+    @service.stub :attach_data_disk_to_vm, response do
+      assert_instance_of Fog::Compute::AzureRM::Server, @server.attach_managed_disk('disk_name', 'resoure_group')
+    end
+
+    async_response = Concurrent::Promise.execute { 10 }
+    @service.stub :attach_data_disk_to_vm, async_response do
+      assert_instance_of Fog::AzureRM::AsyncResponse, @server.attach_managed_disk('managed_disk_name', 'resoure_group', true)
+    end
+  end
+
+  def test_detach_managed_disk_response
+    response = ApiStub::Models::Compute::Server.create_linux_virtual_machine_response(@compute_client)
+    @service.stub :detach_data_disk_from_vm, response do
+      assert_instance_of Fog::Compute::AzureRM::Server, @server.detach_managed_disk('managed_disk_name')
+    end
+
+    async_response = Concurrent::Promise.execute { 10 }
+    @service.stub :detach_data_disk_from_vm, async_response do
+      assert_instance_of Fog::AzureRM::AsyncResponse, @server.detach_managed_disk('managed_disk_name', true)
+    end
+  end
 end

data/test/models/storage/test_recovery_vault.rb

@@ -0,0 +1,61 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Recovery Vault model
+class TestRecoveryVault < Minitest::Test
+  def setup
+    @service = Fog::Storage::AzureRM.new(credentials)
+    @recovery_vault = recovery_vault(@service)
+  end
+
+  def test_model_methods
+    methods = [
+      :save,
+      :enable_backup_protection,
+      :disable_backup_protection,
+      :start_backup,
+      :destroy
+    ]
+    methods.each do |method|
+      assert_respond_to @recovery_vault, method
+    end
+  end
+
+  def test_model_attributes
+    attributes = [
+      :id,
+      :name,
+      :resource_group,
+      :location,
+      :type,
+      :sku_name
+    ]
+    attributes.each do |attribute|
+      assert_respond_to @recovery_vault, attribute
+    end
+  end
+
+  def test_save_method_response
+    create_response = ApiStub::Models::Storage::RecoveryVault.create_method_response
+    @service.stub :create_or_update_recovery_vault, create_response do
+      assert_instance_of Fog::Storage::AzureRM::RecoveryVault, @recovery_vault.save
+    end
+  end
+
+  def test_enable_backup_protection_method_response
+    @service.stub :enable_backup_protection, true do
+      assert @recovery_vault.enable_backup_protection('test-vm', 'test-vm-rg')
+    end
+  end
+
+  def test_disable_backup_protection_method_response
+    @service.stub :disable_backup_protection, true do
+      assert @recovery_vault.disable_backup_protection('test-vm', 'test-vm-rg')
+    end
+  end
+
+  def test_destroy_method_response
+    @service.stub :delete_recovery_vault, true do
+      assert @recovery_vault.destroy
+    end
+  end
+end

data/test/models/storage/test_recovery_vaults.rb

@@ -0,0 +1,47 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test Class for RecoveryVault Collections
+class TestRecoveryVaults < Minitest::Test
+  def setup
+    @service = Fog::Storage::AzureRM.new(credentials)
+    @recovery_vaults = Fog::Storage::AzureRM::RecoveryVaults.new(resource_group: 'fog-test-rg', service: @service)
+  end
+
+  def test_collection_methods
+    methods = [
+      :all,
+      :get
+    ]
+    methods.each do |method|
+      assert_respond_to @recovery_vaults, method
+    end
+  end
+
+  def test_collection_attributes
+    attributes = [
+      :resource_group,
+      :name
+    ]
+    attributes.each do |attribute|
+      assert_respond_to @recovery_vaults, attribute
+    end
+  end
+
+  def test_all_method_response
+    response = [ApiStub::Models::Storage::RecoveryVault.create_method_response]
+    @service.stub :list_recovery_vaults, response do
+      assert_instance_of Fog::Storage::AzureRM::RecoveryVaults, @recovery_vaults.all
+      assert @recovery_vaults.all.size >= 1
+      @recovery_vaults.all.each do |recovery_vault|
+        assert_instance_of Fog::Storage::AzureRM::RecoveryVault, recovery_vault
+      end
+    end
+  end
+
+  def test_get_method_response
+    response = ApiStub::Models::Storage::RecoveryVault.create_method_response
+    @service.stub :get_recovery_vault, response do
+      assert_instance_of Fog::Storage::AzureRM::RecoveryVault, @recovery_vaults.get('fog-test-rg', 'fog-test-vault')
+    end
+  end
+end

data/test/requests/compute/test_attach_data_disk_to_vm.rb

@@ -11,6 +11,10 @@ class TestAttachDataDiskToVM < Minitest::Test
     @get_vm_response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_response(@compute_client)
     @update_vm_response = ApiStub::Requests::Compute::VirtualMachine.update_virtual_machine_response(@compute_client)
     @storage_access_keys_response = ApiStub::Requests::Storage::StorageAccount.list_keys_response
+    @disks = @compute_client.disks
+    @get_managed_disk_response = ApiStub::Requests::Compute::VirtualMachine.get_managed_disk_response(@compute_client)
+    @get_vm_managed_disk_response = ApiStub::Requests::Compute::VirtualMachine.get_vm_with_managed_disk_response(@compute_client)
+    @input_params = {vm_name: "fog-test-vm", vm_resource_group: "fog-test-rg", disk_name: "disk1", disk_size_gb: 1, storage_account_name: "mystorage1"}
   end
 
   def test_attach_data_disk_to_vm_success
@@ -20,7 +24,7 @@ class TestAttachDataDiskToVM < Minitest::Test
         @virtual_machines.stub :get, @get_vm_response do
           @storage_accounts.stub :list_keys, @storage_access_keys_response do
             @virtual_machines.stub :create_or_update, @update_vm_response do
-              assert_equal @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1', false), @update_vm_response
+              assert_equal @service.attach_data_disk_to_vm(@input_params, false), @update_vm_response
               blob_service.verify
             end
           end
@@ -29,13 +33,24 @@ class TestAttachDataDiskToVM < Minitest::Test
     end
   end
 
+  def test_attach_managed_disk_to_vm_success
+    @virtual_machines.stub :get, @get_vm_response do
+      @disks.stub :get, @get_managed_disk_response do
+        @virtual_machines.stub :create_or_update, @get_vm_managed_disk_response do
+          input_params = {vm_name: "ManagedVM", vm_resource_group: "ManagedRG", disk_name: "ManagedDataDisk1", disk_size_gb: 100, disk_resource_group: "ManagedRG"}
+          assert_equal @service.attach_data_disk_to_vm(input_params, false), @get_vm_managed_disk_response
+        end
+      end
+    end
+  end
+
   def test_attach_data_disk_to_vm_failure
     get_vm_response = ApiStub::Requests::Compute::VirtualMachine.virtual_machine_15_data_disks_response(@compute_client)
     @virtual_machines.stub :get, get_vm_response do
       @storage_accounts.stub :list_keys, @storage_access_keys_response do
         @virtual_machines.stub :create_or_update, @update_vm_response do
           assert_raises RuntimeError do
-            @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1', false)
+            @service.attach_data_disk_to_vm(@input_params, false)
           end
         end
       end
@@ -48,7 +63,7 @@ class TestAttachDataDiskToVM < Minitest::Test
       @storage_accounts.stub :list_keys, @storage_access_keys_response do
         @virtual_machines.stub :create_or_update, update_vm_response do
           assert_raises RuntimeError do
-            @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1', false)
+            @service.attach_data_disk_to_vm(@input_params, false)
           end
         end
       end
@@ -61,7 +76,7 @@ class TestAttachDataDiskToVM < Minitest::Test
       @storage_accounts.stub :list_keys, @storage_access_keys_response do
         @virtual_machines.stub :create_or_update, update_vm_response do
           assert_raises RuntimeError do
-            @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1', false)
+            @service.attach_data_disk_to_vm(@input_params, false)
           end
         end
       end
@@ -72,7 +87,7 @@ class TestAttachDataDiskToVM < Minitest::Test
     get_vm_response = proc { fail MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
     @virtual_machines.stub :get, get_vm_response do
       assert_raises RuntimeError do
-        @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1', false)
+        @service.attach_data_disk_to_vm(@input_params, false)
       end
     end
   end
@@ -82,7 +97,7 @@ class TestAttachDataDiskToVM < Minitest::Test
     @virtual_machines.stub :get, @get_vm_response do
       @storage_accounts.stub :list_keys, storage_access_keys_response do
         assert_raises RuntimeError do
-          @service.attach_data_disk_to_vm('fog-test-rg', 'fog-test-vm', 'disk1', 1, 'mystorage1', false)
+          @service.attach_data_disk_to_vm(@input_params, false)
         end
       end
     end

data/test/requests/storage/test_create_recovery_vault.rb

@@ -0,0 +1,35 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Create Recovery Vault request
+class TestCreateRecoveryVault < Minitest::Test
+  def setup
+    @service = Fog::Storage::AzureRM.new(credentials)
+    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+  end
+
+  def test_create_recovery_vault_success
+    response = ApiStub::Requests::Storage::RecoveryVault.create_recovery_vault_response
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      RestClient.stub :put, response do
+        assert_equal @service.create_or_update_recovery_vault('fog-test-rg', 'westus', 'fog-test-vault'), JSON.parse(response)
+      end
+    end
+  end
+
+  def test_create_recovery_vault_argument_error
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      assert_raises ArgumentError do
+        @service.create_or_update_recovery_vault('fog-test-rg', 'fog-test-vault')
+      end
+    end
+  end
+
+  def test_create_recovery_vault_exception
+    response = -> { fail Exception.new('mocked exception') }
+    @token_provider.stub :get_authentication_header, response do
+      assert_raises Exception do
+        @service.create_or_update_recovery_vault('fog-test-rg', 'westus', 'fog-test-vault')
+      end
+    end
+  end
+end

data/test/requests/storage/test_delete_recovery_vault.rb

@@ -0,0 +1,34 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Delete Recovery Vault request
+class TestDeleteRecoveryVault < Minitest::Test
+  def setup
+    @service = Fog::Storage::AzureRM.new(credentials)
+    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+  end
+
+  def test_delete_recovery_vault_success
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      RestClient.stub :delete, true do
+        assert @service.delete_recovery_vault('fog-test-rg', 'fog-test-vault'), true
+      end
+    end
+  end
+
+  def test_delete_recovery_vault_argument_error
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      assert_raises ArgumentError do
+        @service.delete_recovery_vault('fog-test-rg')
+      end
+    end
+  end
+
+  def test_delete_recovery_vault_exception
+    response = -> { fail Exception.new('mocked exception') }
+    @token_provider.stub :get_authentication_header, response do
+      assert_raises Exception do
+        assert @service.delete_recovery_vault('fog-test-rg')
+      end
+    end
+  end
+end

data/test/requests/storage/test_disable_backup_protection.rb

@@ -0,0 +1,52 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Disable Backup Protection request
+class TestDisableBackupProtection < Minitest::Test
+  def setup
+    @service = Fog::Storage::AzureRM.new(credentials)
+    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+
+    @compute_service = Fog::Compute::AzureRM.new(credentials)
+    @compute_client = @compute_service.instance_variable_get(:@compute_mgmt_client)
+  end
+
+  def test_disable_backup_protection_success
+    response = ApiStub::Requests::Compute::VirtualMachine.virtual_machine_response(@compute_client)
+    @service.stub :set_recovery_vault_context, true do
+      @service.stub :get_virtual_machine_id, response.id do
+        @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+          RestClient.stub :put, true do
+            assert_equal @service.disable_backup_protection('fog-test-vault', 'fog-test-rg', 'fog-test-vm', 'fog-test-vm-rg'), true
+          end
+        end
+      end
+    end
+  end
+
+  def test_disable_backup_protection_argument_error
+    response = ApiStub::Requests::Compute::VirtualMachine.virtual_machine_response(@compute_client)
+    @service.stub :set_recovery_vault_context, true do
+      @service.stub :get_virtual_machine_id, response.id do
+        @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+          assert_raises ArgumentError do
+            @service.disable_backup_protection('fog-test-vault', 'fog-test-rg')
+          end
+        end
+      end
+    end
+  end
+
+  def test_disable_backup_protection_exception
+    response = ApiStub::Requests::Compute::VirtualMachine.virtual_machine_response(@compute_client)
+    rest_client_response = -> { fail RestClient::Exception.new("'body': {'error': {'code': 'ResourceNotFound', 'message': 'mocked exception message'}}") }
+    @service.stub :set_recovery_vault_context, true do
+      @service.stub :get_virtual_machine_id, response.id do
+        @token_provider.stub :get_authentication_header, rest_client_response do
+          assert_raises Exception do
+            @service.disable_backup_protection('fog-test-vault', 'fog-test-rg', 'fog-test-vm', 'fog-test-vm-rg')
+          end
+        end
+      end
+    end
+  end
+end

data/test/requests/storage/test_enable_backup_protection.rb

@@ -0,0 +1,66 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Enable Backup Protection request
+class TestEnableBackupProtection < Minitest::Test
+  def setup
+    @service = Fog::Storage::AzureRM.new(credentials)
+    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+
+    @compute_service = Fog::Compute::AzureRM.new(credentials)
+    @compute_client = @compute_service.instance_variable_get(:@compute_mgmt_client)
+
+    @response = ApiStub::Requests::Compute::VirtualMachine.virtual_machine_response(@compute_client)
+    @policy_response = ApiStub::Requests::Storage::RecoveryVault.get_backup_protection_policy_response
+  end
+
+  def test_enable_backup_protection_success
+    @service.stub :set_recovery_vault_context, true do
+      @service.stub :get_backup_protection_policy, JSON.parse(@policy_response)['value'] do
+        @service.stub :get_virtual_machine_id, @response.id do
+          @service.stub :get_backup_job_for_vm, nil do
+            @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+              RestClient.stub :put, true do
+                assert_equal @service.enable_backup_protection('fog-test-vault', 'fog-test-rg', 'fog-test-vm', 'fog-test-vm-rg'), true
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+
+  def test_enable_backup_protection_argument_error
+    @service.stub :set_recovery_vault_context, true do
+      @service.stub :get_backup_protection_policy, JSON.parse(@policy_response)['value'] do
+        @service.stub :get_virtual_machine_id, @response.id do
+          @service.stub :get_backup_job_for_vm, nil do
+            @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+              assert_raises ArgumentError do
+                @service.enable_backup_protection('fog-test-vault', 'fog-test-rg')
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+
+  def test_enable_backup_protection_exception
+    response = ApiStub::Requests::Compute::VirtualMachine.virtual_machine_response(@compute_client)
+    policy_response = ApiStub::Requests::Storage::RecoveryVault.get_backup_protection_policy_response
+    rest_client_response = -> { fail RestClient::Exception.new("'body': {'error': {'code': 'ResourceNotFound', 'message': 'mocked exception message'}}") }
+    @service.stub :set_recovery_vault_context, true do
+      @service.stub :get_backup_protection_policy, JSON.parse(policy_response)['value'] do
+        @service.stub :get_virtual_machine_id, response.id do
+          @service.stub :get_backup_job_for_vm, nil do
+            @token_provider.stub :get_authentication_header, rest_client_response do
+              assert_raises Exception do
+                @service.enable_backup_protection('fog-test-vault', 'fog-test-rg', 'fog-test-vm', 'fog-test-vm-rg')
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/test/requests/storage/test_get_all_backup_jobs.rb

@@ -0,0 +1,35 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Get All Backup Jobs request
+class TestGetAllBackupJobs < Minitest::Test
+  def setup
+    @service = Fog::Storage::AzureRM.new(credentials)
+    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+  end
+
+  def test_get_all_backup_jobs_success
+    response = ApiStub::Requests::Storage::RecoveryVault.get_all_backup_jobs_response
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      RestClient.stub :get, response do
+        assert_equal @service.get_all_backup_jobs('fog-test-vault', 'fog-test-rg'), JSON.parse(response)['value']
+      end
+    end
+  end
+
+  def test_get_all_backup_jobs_argument_error
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      assert_raises ArgumentError do
+        @service.get_all_backup_jobs('fog-test-vault')
+      end
+    end
+  end
+
+  def test_get_all_backup_jobs_exception
+    response = -> { fail Exception.new('mocked exception') }
+    @token_provider.stub :get_authentication_header, response do
+      assert_raises Exception do
+        @service.get_all_backup_jobs('fog-test-vault', 'fog-test-rg')
+      end
+    end
+  end
+end