fog-azure-rm 0.0.9 → 0.1.0

data/test/models/storage/test_storage_account.rb

@@ -28,7 +28,8 @@ class TestStorageAccount < Minitest::Test
       :location,
       :resource_group,
       :sku_name,
-      :replication
+      :replication,
+      :encryption
     ]
     attributes.each do |attribute|
       assert_respond_to @storage_account, attribute
@@ -56,6 +57,12 @@ class TestStorageAccount < Minitest::Test
     end
   end
 
+  def test_update_method_response
+    @service.stub :update_storage_account, @storage_account_response do
+      assert_instance_of Fog::Storage::AzureRM::StorageAccount, @storage_account.update({})
+    end
+  end
+
   def test_get_access_keys_method_response
     key1 = Azure::ARM::Storage::Models::StorageAccountKey.new
     key1.key_name = 'key1'

data/test/requests/sql/test_create_or_update_firewall_rule.rb

@@ -0,0 +1,27 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Create Sql Server Firewall Rule
+class TestCreateOrUpdateFirewallRule < Minitest::Test
+  def setup
+    @service = Fog::Sql::AzureRM.new(credentials)
+    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+  end
+
+  def test_create_or_update_sql_server_firewall_rule_success
+    firewall_rule_response = ApiStub::Requests::Sql::FirewallRule.create_firewall_rule_response
+    data_hash = ApiStub::Requests::Sql::FirewallRule.firewall_rule_hash
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      RestClient.stub :put, firewall_rule_response do
+        assert_equal @service.create_or_update_firewall_rule(data_hash), JSON.parse(firewall_rule_response)
+      end
+    end
+  end
+
+  def test_create_or_update_sql_server_firewall_rule_failure
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      assert_raises ArgumentError do
+        @service.create_or_update_firewall_rule('test-resource-group', 'test-server-name')
+      end
+    end
+  end
+end

data/test/requests/sql/test_delete_firewall_rule.rb

@@ -0,0 +1,34 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Delete Sql Server Firewall Rule
+class TestDeleteFirewallRule < Minitest::Test
+  def setup
+    @service = Fog::Sql::AzureRM.new(credentials)
+    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+  end
+
+  def test_delete_sql_server_firewall_rule_success
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      RestClient.stub :delete, true do
+        assert @service.delete_firewall_rule('fog-test-rg', 'fog-test-server-name', 'rule-name')
+      end
+    end
+  end
+
+  def test_delete_sql_server_firewall_rule_failure
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      assert_raises ArgumentError do
+        @service.delete_firewall_rule('fog-test-name')
+      end
+    end
+  end
+
+  def test_delete_sql_server_firewall_rule_exception
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @token_provider.stub :get_authentication_header, response do
+      assert_raises Exception do
+        assert @service.delete_firewall_rule('fog-test-rg', 'fog-test-server-name', 'rule-name')
+      end
+    end
+  end
+end

data/test/requests/sql/test_get_firewall_rule.rb

@@ -0,0 +1,35 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Get Sql Server Firewall Rule
+class TestGetFirewallRule < Minitest::Test
+  def setup
+    @service = Fog::Sql::AzureRM.new(credentials)
+    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+  end
+
+  def test_get_sql_server_firewall_rule_success
+    create_response = ApiStub::Requests::Sql::FirewallRule.create_firewall_rule_response
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      RestClient.stub :get, create_response do
+        assert_equal @service.get_firewall_rule('fog-test-rg', 'fog-test-server-name', 'rule-name'), JSON.parse(create_response)
+      end
+    end
+  end
+
+  def test_get_sql_server_firewall_rule_failure
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      assert_raises ArgumentError do
+        @service.get_firewall_rule('fog-test-rg')
+      end
+    end
+  end
+
+  def test_get_sql_server_firewall_rule_exception
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @token_provider.stub :get_authentication_header, response do
+      assert_raises Exception do
+        @service.get_firewall_rule('fog-test-rg', 'fog-test-server-name', 'rule-name')
+      end
+    end
+  end
+end

data/test/requests/sql/test_list_firewall_rules.rb

@@ -0,0 +1,35 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for List Sql Server Firewall Rule Request
+class TestListFirewallRules < Minitest::Test
+  def setup
+    @service = Fog::Sql::AzureRM.new(credentials)
+    @token_provider = Fog::Credentials::AzureRM.instance_variable_get(:@token_provider)
+  end
+
+  def test_list_sql_server_firewall_rules_success
+    list_response = ApiStub::Requests::Sql::FirewallRule.list_firewall_rule_response
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      RestClient.stub :get, list_response do
+        assert_equal @service.list_firewall_rules('fog-test-rg', 'server-name'), JSON.parse(list_response)['value']
+      end
+    end
+  end
+
+  def test_list_sql_servers_firewall_rules_failure
+    @token_provider.stub :get_authentication_header, 'Bearer <some-token>' do
+      assert_raises ArgumentError do
+        @service.list_firewall_rules('fog-test-rg')
+      end
+    end
+  end
+
+  def test_list_sql_servers_firewall_rules_exception
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @token_provider.stub :get_authentication_header, response do
+      assert_raises Exception do
+        @service.list_firewall_rules('fog-test-rg', 'server-name')
+      end
+    end
+  end
+end

data/test/requests/storage/test_create_disk.rb

@@ -0,0 +1,17 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Data Disk Class
+class TestCreateDisk < Minitest::Test
+  # This class posesses the test cases for the request of create disk.
+  def setup
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @blob_client = @service.instance_variable_get(:@blob_client)
+    @storage_data_disk_object = ApiStub::Requests::Storage::File.upload_block_blob_from_file
+  end
+
+  def test_create_disk_success
+    @service.stub :upload_block_blob_from_file, @storage_data_disk_object do
+      assert @service.create_disk('test_blob', options = {})
+    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_disk.rb

@@ -0,0 +1,23 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Data Disk Class
+class TestDeleteDisk < Minitest::Test
+  # This class posesses the test cases for the requests of deleting data disks.
+  def setup
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @blob_client = @service.instance_variable_get(:@blob_client)
+    @storage_data_disk_object = ApiStub::Requests::Storage::File.get_blob_properties
+  end
+
+  def test_delete_disk_success
+    @service.stub :delete_blob, nil do
+      assert @service.delete_disk('testblob1')
+    end
+  end
+
+  def test_delete_disk_failure
+    @service.stub :delete_blob, @storage_data_disk_object do
+      assert !@service.delete_disk('testblob1')
+    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

data/test/requests/storage/test_get_backup_container.rb

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