fog-azure-rm-downgraded 0.3.0

data/test/models/storage/test_files.rb

@@ -0,0 +1,234 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Blob Collection
+class TestFiles < Minitest::Test
+  def setup
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @directory = directory(@service)
+    @files = @directory.files
+
+    @mocked_response = mocked_storage_http_error
+    @container = ApiStub::Models::Storage::File.container
+    @blob = ApiStub::Models::Storage::File.blob
+    @blob_list = ApiStub::Models::Storage::File.blob_list
+    @blob_https_url = ApiStub::Models::Storage::File.blob_https_url
+  end
+
+  def test_collection_methods
+    methods = [
+      :all,
+      :each,
+      :each_file_this_page,
+      :get,
+      :get_url,
+      :get_http_url,
+      :get_https_url,
+      :head,
+      :new
+    ]
+    methods.each do |method|
+      assert_respond_to @files, method
+    end
+  end
+
+  def test_collection_attributes
+    attributes = [
+      :directory,
+      :delimiter,
+      :marker,
+      :max_results,
+      :next_marker,
+      :prefix
+    ]
+    attributes.each do |attribute|
+      assert_respond_to @files, attribute
+    end
+  end
+
+  def test_all_method
+    @service.stub :get_container_properties, @container do
+      @service.stub :list_blobs, @blob_list do
+        files = @files.all
+        assert_instance_of Fog::Storage::AzureRM::Files, files
+        assert_equal @blob_list[:blobs].size, files.size
+      end
+    end
+  end
+
+  def test_all_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @files.attributes.delete(:directory)
+      @files.all
+    end
+  end
+
+  def test_all_method_container_not_found
+    @directory.collection.stub :get, nil do
+      assert @files.all.nil?
+    end
+  end
+
+  def test_each_method_with_all
+    @blob_list[:next_marker] = nil
+    @blob_list[:marker] = nil
+    @service.stub :get_container_properties, @container do
+      @service.stub :list_blobs, @blob_list do
+        @files.next_marker = nil
+        j = 0
+        @files.each do |file|
+          assert_instance_of Fog::Storage::AzureRM::File, file
+          j += 1
+        end
+        assert_equal @blob_list[:blobs].size, j
+      end
+    end
+  end
+
+  def test_each_method_with_parts
+    assert_instance_of Fog::Storage::AzureRM::Files, @files.each
+
+    multiple_values = lambda do |_container_name, options|
+      if options[:marker] == 'marker'
+        return {
+          next_marker: nil,
+          blobs: @blob_list[:blobs][2, 2]
+        }
+      end
+      return {
+        next_marker: 'marker',
+        blobs: @blob_list[:blobs][0, 2]
+      }
+    end
+
+    @service.stub :get_container_properties, @container do
+      @service.stub :list_blobs, multiple_values do
+        j = 0
+        @files.each do |file|
+          assert_instance_of Fog::Storage::AzureRM::File, file
+          j += 1
+        end
+        assert_equal @blob_list[:blobs].size, j
+      end
+    end
+  end
+
+  def test_get_method_success
+    content = 'data'
+    @service.stub :get_blob, [@blob, content] do
+      file = @files.get('test_blob')
+      assert_instance_of Fog::Storage::AzureRM::File, file
+      assert_equal content, file.body
+    end
+  end
+
+  def test_get_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @files.attributes.delete(:directory)
+      @files.get('test_blob')
+    end
+  end
+
+  def test_get_method_blob_not_found
+    exception = ->(*) { raise 'NotFound' }
+    @service.stub :get_blob, exception do
+      assert @files.get('test_blob').nil?
+    end
+  end
+
+  def test_get_method_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @service.stub :get_blob, http_exception do
+      assert_raises(Azure::Core::Http::HTTPError) do
+        @files.get('test_blob')
+      end
+    end
+  end
+
+  def test_get_url_method_success
+    @files.stub :get_https_url, @blob_https_url do
+      assert_equal @blob_https_url, @files.get_url('test_blob', Time.now + 3600)
+    end
+
+    options = { scheme: 'http' }
+    http_url = @blob_https_url.gsub('https:', 'http:')
+    @files.stub :get_http_url, http_url do
+      assert_equal http_url, @files.get_url('test_blob', Time.now + 3600, options)
+    end
+  end
+
+  def test_get_url_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @files.attributes.delete(:directory)
+      @files.get_url('test_blob', Time.now + 3600)
+    end
+  end
+
+  def test_get_http_url_method_success
+    http_url = @blob_https_url.gsub('https:', 'http:')
+    @service.stub :get_blob_http_url, http_url do
+      assert_equal http_url, @files.get_http_url('test_blob', Time.now + 3600)
+    end
+  end
+
+  def test_get_http_url_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @files.attributes.delete(:directory)
+      @files.get_http_url('test_blob', Time.now + 3600)
+    end
+  end
+
+  def test_get_https_url_method_success
+    @service.stub :get_blob_https_url, @blob_https_url do
+      assert_equal @blob_https_url, @files.get_https_url('test_blob', Time.now + 3600)
+    end
+  end
+
+  def test_get_https_url_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @files.attributes.delete(:directory)
+      @files.get_https_url('test_blob', Time.now + 3600)
+    end
+  end
+
+  def test_head_method_success
+    @service.stub :get_blob_properties, @blob do
+      file = @files.head('test_blob')
+      assert_instance_of Fog::Storage::AzureRM::File, file
+      assert file.attributes[:body].nil?
+    end
+  end
+
+  def test_head_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @files.attributes.delete(:directory)
+      @files.head('test_blob')
+    end
+  end
+
+  def test_head_method_blob_not_found
+    exception = ->(*) { raise 'NotFound' }
+    @service.stub :get_blob_properties, exception do
+      assert @files.head('test_blob').nil?
+    end
+  end
+
+  def test_head_method_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @service.stub :get_blob_properties, http_exception do
+      assert_raises(Azure::Core::Http::HTTPError) do
+        @files.head('test_blob')
+      end
+    end
+  end
+
+  def test_new_method_success
+    assert_instance_of Fog::Storage::AzureRM::File, @files.new
+  end
+
+  def test_new_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @files.attributes.delete(:directory)
+      @files.new
+    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/models/storage/test_storage_account.rb

@@ -0,0 +1,88 @@
+require File.expand_path '../../test_helper', __dir__
+# Test class for Storage Account Model
+class TestStorageAccount < Minitest::Test
+  def setup
+    @service = Fog::Storage::AzureRM.new(credentials)
+    @storage_mgmt_client = @service.instance_variable_get(:@storage_mgmt_client)
+    @storage_account = storage_account(@service)
+    @standard_lrs_storage_account = standard_lrs(@service)
+    @standard_invalid_replication = standard_check_for_invalid_replications(@service)
+    @premium_invalid_replication  = premium_check_for_invalid_replications(@service)
+    @storage_account_response = ApiStub::Models::Storage::StorageAccount.create_storage_account(@storage_mgmt_client)
+  end
+
+  def test_model_methods
+    methods = [
+      :save,
+      :destroy,
+      :get_access_keys
+    ]
+    methods.each do |method|
+      assert_respond_to @storage_account, method
+    end
+  end
+
+  def test_model_attributes
+    attributes = [
+      :name,
+      :location,
+      :resource_group,
+      :sku_name,
+      :replication,
+      :encryption
+    ]
+    attributes.each do |attribute|
+      assert_respond_to @storage_account, attribute
+    end
+  end
+
+  def test_save_method_response
+    @service.stub :create_storage_account, @storage_account_response do
+      assert_instance_of Fog::Storage::AzureRM::StorageAccount, @storage_account.save
+    end
+    @service.stub :create_storage_account, @storage_account_response do
+      assert_raises RuntimeError do
+        @standard_lrs_storage_account.save
+      end
+    end
+    @service.stub :create_storage_account, @storage_account_response do
+      assert_raises RuntimeError do
+        @standard_invalid_replication.save
+      end
+    end
+    @service.stub :create_storage_account, @storage_account_response do
+      assert_raises RuntimeError do
+        @premium_invalid_replication.save
+      end
+    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'
+    key1.value = 'YWNjZXNzLWtleQ=='
+    key1.permissions = 'Full'
+    response = [key1]
+    @service.stub :get_storage_access_keys, response do
+      assert_equal @storage_account.get_access_keys, response
+    end
+  end
+
+  def test_destroy_method_true_response
+    @service.stub :delete_storage_account, true do
+      assert @storage_account.destroy
+    end
+  end
+
+  def test_destroy_method_false_response
+    @service.stub :delete_storage_account, false do
+      assert !@storage_account.destroy
+    end
+  end
+end

data/test/models/storage/test_storage_accounts.rb

@@ -0,0 +1,78 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Availability Set Collection
+class TestStorageAccounts < Minitest::Test
+  def setup
+    @service = Fog::Storage::AzureRM.new(credentials)
+    @storage_mgmt_client = @service.instance_variable_get(:@storage_mgmt_client)
+    @storage_accounts = Fog::Storage::AzureRM::StorageAccounts.new(resource_group: 'fog-test-rg', service: @service)
+    @list_storage_account_response = [ApiStub::Models::Storage::StorageAccount.create_storage_account(@storage_mgmt_client)]
+  end
+
+  def test_collection_methods
+    methods = [
+      :all,
+      :get,
+      :check_storage_account_exists
+    ]
+    methods.each do |method|
+      assert_respond_to @storage_accounts, method
+    end
+  end
+
+  def test_collection_attributes
+    assert_respond_to @storage_accounts, :resource_group
+  end
+
+  def test_all_method_response_for_rg
+    @service.stub :list_storage_account_for_rg, @list_storage_account_response do
+      assert_instance_of Fog::Storage::AzureRM::StorageAccounts, @storage_accounts.all
+      assert @storage_accounts.all.size >= 1
+      @storage_accounts.all.each do |s|
+        assert_instance_of Fog::Storage::AzureRM::StorageAccount, s
+      end
+    end
+  end
+
+  def test_all_method_response
+    storage_accounts = Fog::Storage::AzureRM::StorageAccounts.new(service: @service)
+    @service.stub :list_storage_accounts, @list_storage_account_response do
+      assert_instance_of Fog::Storage::AzureRM::StorageAccounts, storage_accounts.all
+      assert storage_accounts.all.size >= 1
+      storage_accounts.all.each do |s|
+        assert_instance_of Fog::Storage::AzureRM::StorageAccount, s
+      end
+    end
+  end
+
+  def test_get_method_response
+    response = ApiStub::Models::Storage::StorageAccount.create_storage_account(@storage_mgmt_client)
+    @service.stub :get_storage_account, response do
+      assert_instance_of Fog::Storage::AzureRM::StorageAccount, @storage_accounts.get('fog-test-rg', 'fog-test-storage-account')
+    end
+  end
+
+  def test_check_name_availability_true_case
+    @service.stub :check_storage_account_name_availability, true do
+      assert @storage_accounts.check_name_availability('fog-test-storage-account')
+    end
+  end
+
+  def test_check_name_availability_false_case
+    @service.stub :check_storage_account_name_availability, false do
+      assert !@storage_accounts.check_name_availability('fog-test-storage-account')
+    end
+  end
+
+  def test_check_storage_account_exists_true_response
+    @service.stub :check_storage_account_exists, true do
+      assert @storage_accounts.check_storage_account_exists('fog-test-rg', 'fog-test-storage-account')
+    end
+  end
+
+  def test_check_storage_account_exists_false_response
+    @service.stub :check_storage_account_exists, false do
+      assert !@storage_accounts.check_storage_account_exists('fog-test-rg', 'fog-test-storage-account')
+    end
+  end
+end