fog-azure-rm 0.0.6 → 0.0.8

data/test/models/storage/test_file.rb

@@ -0,0 +1,258 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Storage Container Model
+class TestFile < Minitest::Test
+  def setup
+    Fog.mock!
+    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @mock_file = storage_blob(@mock_service)
+    Fog.unmock!
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @file = storage_blob(@service)
+    @cloud_blob = storage_cloud_blob
+    @blob_client = @service.instance_variable_get(:@blob_client)
+    @mocked_response = mocked_storage_http_error
+    @content = Array.new(1024 * 1024) { 'a' }.join
+    @get_metadata_result = ApiStub::Models::Storage::File.test_get_blob_metadata
+    @create_result = ApiStub::Models::Storage::File.upload_block_blob_from_file
+    @download_result = ApiStub::Models::Storage::File.download_blob_to_file
+    @get_properties_result = ApiStub::Models::Storage::File.get_blob_properties
+  end
+
+  def test_model_methods
+    methods = [
+      :save,
+      :save_to_file,
+      :get_properties,
+      :set_properties,
+      :set_metadata,
+      :get_metadata,
+      :destroy
+    ]
+    methods.each do |method|
+      assert @file.respond_to? method, true
+    end
+  end
+
+  def test_model_attributes
+    attributes = [
+      :key,
+      :accept_ranges,
+      :cache_control,
+      :committed_block_count,
+      :content_length,
+      :content_type,
+      :content_md5,
+      :content_encoding,
+      :content_language,
+      :content_disposition,
+      :copy_completion_time,
+      :copy_status,
+      :copy_status_description,
+      :copy_id,
+      :copy_progress,
+      :copy_source,
+      :directory,
+      :etag,
+      :file_path,
+      :last_modified,
+      :lease_duration,
+      :lease_state,
+      :lease_status,
+      :metadata,
+      :sequence_number,
+      :blob_type
+    ]
+    attributes.each do |attribute|
+      assert_respond_to @file, attribute
+    end
+  end
+
+  def test_save_method_response
+    @file.file_path = 'test.txt'
+    @service.stub :upload_block_blob_from_file, @create_result do
+      assert_instance_of Fog::Storage::AzureRM::File, @file.save
+    end
+    @service.stub :upload_block_blob_from_file, @create_result do
+      assert_instance_of Fog::Storage::AzureRM::File, @file.create
+    end
+  end
+
+  def test_save_small_blob_method_response
+    small_file_name = 'small_test_file.dat'
+    small_file = File.new(small_file_name, 'w')
+    small_file.puts(@content)
+    small_file.close
+    @file.file_path = small_file_name
+    @blob_client.stub :create_block_blob, @create_result do
+      assert_instance_of Fog::Storage::AzureRM::File, @file.create
+    end
+    io_exception = -> (_container_name, _blob_name, _file_path, _option) { raise IOError.new }
+    @blob_client.stub :create_block_blob, io_exception do
+      assert_raises(RuntimeError) do
+        @file.create
+      end
+    end
+    http_exception = -> (_container_name, _blob_name, _file_path, _option) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :create_block_blob, http_exception do
+      assert_raises(RuntimeError) do
+        @file.create
+      end
+    end
+    File.delete(small_file_name)
+  end
+
+  def test_save_large_blob_method_response
+    large_file_name = 'large_test_file.dat'
+    large_file = File.new(large_file_name, 'w')
+    33.times do
+      large_file.puts(@content)
+    end
+    large_file.close
+    @file.file_path = large_file_name
+    @blob_client.stub :put_blob_block, true do
+      @blob_client.stub :commit_blob_blocks, @create_result do
+        assert_instance_of Fog::Storage::AzureRM::File, @file.create
+      end
+    end
+    File.delete(large_file_name)
+  end
+
+  def test_save_method_response_mock
+    @mock_file.file_path = 'mock_test.txt'
+    assert_instance_of Fog::Storage::AzureRM::File, @mock_file.save
+  end
+
+  def test_save_to_file
+    @service.stub :download_blob_to_file, @download_result do
+      assert_instance_of Fog::Storage::AzureRM::File, @file.save_to_file('test.txt')
+    end
+  end
+
+  def test_save_to_file_io_exception
+    io_exception = -> (_container_name, _blob_name, _option) { raise IOError.new }
+    @blob_client.stub :get_blob, io_exception do
+      assert_raises(RuntimeError) do
+        @file.save_to_file('test.txt')
+      end
+    end
+  end
+
+  def test_save_to_file_http_exception
+    http_exception = -> (_container_name, _blob_name, _option) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :get_blob, http_exception do
+      assert_raises(RuntimeError) do
+        @file.save_to_file('test.txt')
+      end
+    end
+  end
+
+  def test_save_to_file_mock
+    assert_instance_of Fog::Storage::AzureRM::File, @mock_file.save_to_file('test.txt')
+  end
+
+  def test_get_blob_properties
+    @service.stub :get_blob_properties, @get_properties_result do
+      assert_instance_of Fog::Storage::AzureRM::File, @file.get_properties
+    end
+  end
+
+  def test_get_cloud_blob_properties
+    @service.stub :get_blob_properties, @cloud_blob do
+      assert_instance_of Fog::Storage::AzureRM::File, @file.get_properties
+    end
+  end
+
+  def test_get_blob_properties_http_exception
+    http_exception = -> (_container_name, _blob_name, _option) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :get_blob_properties, http_exception do
+      assert_raises(RuntimeError) do
+        @file.get_properties
+      end
+    end
+  end
+
+  def test_get_blob_properties_mock
+    assert_instance_of Fog::Storage::AzureRM::File, @mock_file.get_properties
+  end
+
+  def test_set_blob_properties
+    @service.stub :set_blob_properties, true do
+      assert @file.set_properties
+    end
+  end
+
+  def test_set_blob_properties_http_exception
+    http_exception = -> (_container_name, _blob_name, _properties) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :set_blob_properties, http_exception do
+      assert_raises(RuntimeError) do
+        @file.set_properties
+      end
+    end
+  end
+
+  def test_set_blob_properties_mock
+    assert @mock_file.set_properties
+  end
+
+  def test_get_blob_metadata
+    @service.stub :get_blob_metadata, @get_metadata_result do
+      file = @file.get_metadata
+      assert_instance_of Fog::Storage::AzureRM::File, file
+      assert_equal @get_metadata_result, file.metadata
+    end
+  end
+
+  def test_get_blob_metadata_http_exception
+    http_exception = -> (_container_name, _blob_name, _option) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :get_blob_metadata, http_exception do
+      assert_raises(RuntimeError) do
+        @file.get_metadata
+      end
+    end
+  end
+
+  def test_get_blob_metadata_mock
+    blob = @mock_file.get_metadata
+    assert_instance_of Fog::Storage::AzureRM::File, blob
+    assert_equal @get_metadata_result, blob.metadata
+  end
+
+  def test_set_blob_metadata
+    @service.stub :set_blob_metadata, true do
+      assert @file.set_metadata(@get_metadata_result)
+    end
+  end
+
+  def test_set_blob_metadata_http_exception
+    http_exception = -> (_container_name, _blob_name, _metadata, _option) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :set_blob_metadata, http_exception do
+      assert_raises(RuntimeError) do
+        @file.set_metadata @get_metadata_result
+      end
+    end
+  end
+
+  def test_set_blob_metadata_mock
+    assert @mock_file.set_metadata(@get_metadata_result)
+  end
+
+  def test_delete_method_true_response
+    @service.stub :delete_blob, true do
+      assert @file.destroy
+    end
+  end
+
+  def test_delete_method_exception
+    exception = -> (_container_name, _blob_name, _option) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :delete_blob, exception do
+      assert_raises(RuntimeError) do
+        assert @file.destroy
+      end
+    end
+  end
+
+  def test_delete_method_response_mock
+    assert @mock_file.destroy
+  end
+end

data/test/models/storage/test_files.rb

@@ -0,0 +1,67 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Blob Collection
+class TestFiles < Minitest::Test
+  def setup
+    Fog.mock!
+    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @mock_files = Fog::Storage::AzureRM::Files.new(directory: 'test-container', service: @mock_service)
+    Fog.unmock!
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @blob_client = @service.instance_variable_get(:@blob_client)
+    @files = Fog::Storage::AzureRM::Files.new(directory: 'test-container', service: @service)
+    @mocked_response = mocked_storage_http_error
+    @list_results = ApiStub::Models::Storage::File.list_blobs
+  end
+
+  def test_collection_methods
+    methods = [
+      :all,
+      :get
+    ]
+    methods.each do |method|
+      assert @files.respond_to? method, true
+    end
+  end
+
+  def test_all_method
+    @service.stub :list_blobs, @list_results do
+      assert_instance_of Fog::Storage::AzureRM::Files, @files.all
+      assert @files.all.size >= 1
+      @files.all.each do |file|
+        assert_instance_of Fog::Storage::AzureRM::File, file
+      end
+    end
+  end
+
+  def test_all_method_http_exception
+    http_exception = -> (_container_name, _option) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :list_blobs, http_exception do
+      assert_raises(RuntimeError) do
+        @files.all
+      end
+    end
+  end
+
+  def test_all_method_mock
+    assert_instance_of Fog::Storage::AzureRM::Files, @mock_files.all
+    assert @mock_files.all.size >= 1
+    @mock_files.all.each do |file|
+      assert_instance_of Fog::Storage::AzureRM::File, file
+    end
+  end
+
+  def test_get_method
+    @service.stub :list_blobs, @list_results do
+      assert_instance_of Fog::Storage::AzureRM::File, @files.get('test-container', 'testblob1')
+      assert !@files.get('test-container', 'wrong-name').nil?
+    end
+  end
+
+  def test_get_method_mock
+    @service.stub :list_blobs, @list_results do
+      assert_instance_of Fog::Storage::AzureRM::File, @mock_files.get('test-container', 'testblob1')
+      assert !@mock_files.get('test-container', 'wrong-name').nil?
+    end
+  end
+end

data/test/models/storage/test_storage_account.rb

@@ -30,10 +30,8 @@ class TestStorageAccount < Minitest::Test
       :sku_name,
       :replication
     ]
-    @service.stub :create_storage_account, @storage_account_response do
-      attributes.each do |attribute|
-        assert_respond_to @storage_account, attribute
-      end
+    attributes.each do |attribute|
+      assert_respond_to @storage_account, attribute
     end
   end
 

data/test/requests/compute/test_create_virtual_machine.rb

@@ -9,6 +9,7 @@ class TestCreateVirtualMachine < Minitest::Test
     @response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_response(compute_client)
     @custom_data_response = ApiStub::Requests::Compute::VirtualMachine.create_virtual_machine_with_custom_data_response(compute_client)
     @linux_virtual_machine_hash = ApiStub::Requests::Compute::VirtualMachine.linux_virtual_machine_params
+    @windows_virtual_machine_hash = ApiStub::Requests::Compute::VirtualMachine.windows_virtual_machine_params
   end
 
   def test_create_linux_virtual_machine_success
@@ -18,23 +19,36 @@ class TestCreateVirtualMachine < Minitest::Test
   end
 
   def test_create_windows_virtual_machine_success
-    @windows_virtual_machine_hash = ApiStub::Requests::Compute::VirtualMachine.windows_virtual_machine_params
     @virtual_machines.stub :create_or_update, @response do
       assert_equal @service.create_virtual_machine(@windows_virtual_machine_hash), @response
     end
   end
 
+  def test_create_linux_virtual_machine_from_custom_image_success
+    linux_virtual_machine_with_custom_image_hash = ApiStub::Requests::Compute::VirtualMachine.linux_virtual_machine_with_custom_image_params
+    @virtual_machines.stub :create_or_update, @response do
+      assert_equal @service.create_virtual_machine(linux_virtual_machine_with_custom_image_hash), @response
+    end
+  end
+
+  def test_create_windows_virtual_machine_from_custom_image_success
+    windows_virtual_machine_with_custom_image_hash = ApiStub::Requests::Compute::VirtualMachine.windows_virtual_machine_with_custom_image_params
+    @virtual_machines.stub :create_or_update, @response do
+      assert_equal @service.create_virtual_machine(windows_virtual_machine_with_custom_image_hash), @response
+    end
+  end
+
   def test_create_linux_virtual_machine_with_custom_data_success
-    @linux_virtual_machine_with_custom_data_hash = ApiStub::Requests::Compute::VirtualMachine.linux_virtual_machine_with_custom_data_params
+    linux_virtual_machine_with_custom_data_hash = ApiStub::Requests::Compute::VirtualMachine.linux_virtual_machine_with_custom_data_params
     @virtual_machines.stub :create_or_update, @custom_data_response do
-      assert_equal @service.create_virtual_machine(@linux_virtual_machine_with_custom_data_hash), @custom_data_response
+      assert_equal @service.create_virtual_machine(linux_virtual_machine_with_custom_data_hash), @custom_data_response
     end
   end
 
   def test_create_windows_virtual_machine_with_custom_data_success
-    @windows_virtual_machine_with_custom_data_hash = ApiStub::Requests::Compute::VirtualMachine.windows_virtual_machine_with_custom_data_params
+    windows_virtual_machine_with_custom_data_hash = ApiStub::Requests::Compute::VirtualMachine.windows_virtual_machine_with_custom_data_params
     @virtual_machines.stub :create_or_update, @custom_data_response do
-      assert_equal @service.create_virtual_machine(@windows_virtual_machine_with_custom_data_hash), @custom_data_response
+      assert_equal @service.create_virtual_machine(windows_virtual_machine_with_custom_data_hash), @custom_data_response
     end
   end
 

data/test/requests/compute/test_create_vm_extension.rb

@@ -0,0 +1,35 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Create VM Extension request
+class TestCreateVMExtension < Minitest::Test
+  def setup
+    @service = Fog::Compute::AzureRM.new(credentials)
+    compute_client = @service.instance_variable_get(:@compute_mgmt_client)
+    @vm_extension = compute_client.virtual_machine_extensions
+    @response = ApiStub::Requests::Compute::VirtualMachineExtension.create_vm_extension_response(compute_client)
+    @vm_extension_params = ApiStub::Requests::Compute::VirtualMachineExtension.vm_extension_params
+  end
+
+  def test_create_vm_extension_success
+    @vm_extension.stub :create_or_update, @response do
+      assert_equal @service.create_or_update_vm_extension(@vm_extension_params), @response
+    end
+  end
+
+  def test_create_vm_extension_argument_error
+    @vm_extension.stub :create_or_update, @response do
+      assert_raises ArgumentError do
+        @service.create_or_update_vm_extension
+      end
+    end
+  end
+
+  def test_create_vm_extension_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @vm_extension.stub :create_or_update, response do
+      assert_raises RuntimeError do
+        @service.create_or_update_vm_extension(@vm_extension_params)
+      end
+    end
+  end
+end

data/test/requests/compute/test_get_vm_extension.rb

@@ -0,0 +1,26 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Get VM Extension request
+class TestGetVMExtension < Minitest::Test
+  def setup
+    @service = Fog::Compute::AzureRM.new(credentials)
+    @compute_client = @service.instance_variable_get(:@compute_mgmt_client)
+    @vm_extension = @compute_client.virtual_machine_extensions
+  end
+
+  def test_get_vm_extension_success
+    response = ApiStub::Requests::Compute::VirtualMachineExtension.create_vm_extension_response(@compute_client)
+    @vm_extension.stub :get, response do
+      assert_equal @service.get_vm_extension('fog-test-rg', 'fog-test-vm', 'fog-test-extension'), response
+    end
+  end
+
+  def test_update_vm_extension_failure
+    response = proc { raise MsRestAzure::AzureOperationError.new(nil, nil, 'error' => { 'message' => 'mocked exception' }) }
+    @vm_extension.stub :get, response do
+      assert_raises RuntimeError do
+        @service.get_vm_extension('fog-test-rg', 'fog-test-vm', 'fog-test-extension')
+      end
+    end
+  end
+end