fog-azure-rm 0.1.0 → 0.1.1

data/test/requests/storage/test_list_blobs.rb

@@ -0,0 +1,77 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Blob Class
+class TestListBlobs < Minitest::Test
+  # This class posesses the test cases for the requests of listing blobs in storage containers.
+  def setup
+    Fog.mock!
+    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    Fog.unmock!
+    @mocked_response = mocked_storage_http_error
+
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @blob_client = @service.instance_variable_get(:@blob_client)
+
+    @blob_list = ApiStub::Requests::Storage::File.blob_list
+    @blobs1 = Azure::Service::EnumerationResults.new
+    @blobs1.continuation_token = 'marker'
+    @blobs1.push(@blob_list[0])
+    @blobs1.push(@blob_list[1])
+    @blobs2 = Azure::Service::EnumerationResults.new
+    @blobs2.push(@blob_list[2])
+    @blobs2.push(@blob_list[3])
+  end
+
+  def test_list_blobs_success
+    multiple_values = lambda do |_container_name, options|
+      return @blobs2 if options[:marker] == 'marker'
+      return @blobs1
+    end
+    @blob_client.stub :list_blobs, multiple_values do
+      result = @service.list_blobs('test_container')
+      assert result[:next_marker].nil?
+      assert_equal @blob_list, result[:blobs]
+    end
+  end
+
+  def test_list_blobs_with_max_results_success
+    @blob_client.stub :list_blobs, @blobs1 do
+      options = { max_results: 2 }
+      result = @service.list_blobs('test_container', options)
+      assert_equal 'marker', result[:next_marker]
+      assert_equal @blob_list[0, 2], result[:blobs]
+    end
+  end
+
+  def test_list_blobs_not_found
+    exception = ->(*) { raise StandardError.new('Not found(404). Not exist') }
+    @blob_client.stub :list_blobs, exception do
+      assert_raises('NotFound') do
+        @service.list_blobs('test_container')
+      end
+    end
+  end
+
+  def test_list_blobs_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :list_blobs, http_exception do
+      assert_raises(RuntimeError) do
+        @service.list_blobs('test_container')
+      end
+    end
+  end
+
+  def test_list_blobs_exception
+    @blob_client.stub :list_blobs, 'exception' do
+      assert_raises(RuntimeError) do
+        @service.list_blobs('test_container')
+      end
+    end
+  end
+
+  def test_list_blobs_mock
+    result = @mock_service.list_blobs('test_container')
+    assert_equal 'marker', result[:next_marker]
+    assert_equal @blob_list, result[:blobs]
+  end
+end

data/test/requests/storage/test_list_containers.rb

@@ -4,20 +4,51 @@ require File.expand_path '../../test_helper', __dir__
 class TestListContainers < Minitest::Test
   # This class posesses the test cases for the requests of listing storage containers.
   def setup
+    Fog.mock!
+    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    Fog.unmock!
+    @mocked_response = mocked_storage_http_error
+
     @service = Fog::Storage::AzureRM.new(storage_account_credentials)
     @blob_client = @service.instance_variable_get(:@blob_client)
-    @storage_container_object = ApiStub::Requests::Storage::Directory.list_containers
+
+    @containers = ApiStub::Requests::Storage::Directory.container_list
+    @containers1 = Azure::Service::EnumerationResults.new
+    @containers1.continuation_token = 'marker'
+    @containers1.push(@containers[0])
+    @containers1.push(@containers[1])
+    @containers2 = Azure::Service::EnumerationResults.new
+    @containers2.push(@containers[2])
+  end
+
+  def test_list_containers_success
+    multiple_values = lambda do |options|
+      return @containers2 if options[:marker] == 'marker'
+      return @containers1
+    end
+    @blob_client.stub :list_containers, multiple_values do
+      assert_equal @containers, @service.list_containers
+    end
   end
 
-  def test_list_containers_with_service_success
-    @blob_client.stub :list_containers, @storage_container_object do
-      assert @service.list_containers.size >= 1
+  def test_list_containers_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :list_containers, http_exception do
+      assert_raises(RuntimeError) do
+        @service.list_containers
+      end
     end
   end
 
-  def test_list_containers_with_internal_client_success
-    @blob_client.stub :list_containers, @storage_container_object do
-      assert @blob_client.list_containers.size >= 1
+  def test_list_containers_exception
+    @blob_client.stub :list_containers, 'exception' do
+      assert_raises(RuntimeError) do
+        @service.list_containers
+      end
     end
   end
+
+  def test_list_containers_mock
+    assert_equal @containers, @mock_service.list_containers
+  end
 end

data/test/requests/storage/test_multipart_save_block_blob.rb

@@ -0,0 +1,105 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Blob Class
+class TestMultipartSaveBlockBlob < Minitest::Test
+  # This class posesses the test cases for the requests of saving storage block blob with multiple parts.
+  def setup
+    Fog.mock!
+    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    Fog.unmock!
+    @mocked_response = mocked_storage_http_error
+
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+  end
+
+  def test_multipart_save_block_blob_success
+    body = 'd' * 5 * 1024 * 1024 # MAXIMUM_CHUNK_SIZE is 4 * 1024 * 1024
+
+    @service.stub :create_block_blob, true do
+      @service.stub :put_blob_block, true do
+        @service.stub :commit_blob_blocks, true do
+          assert @service.multipart_save_block_blob('test_container', 'test_blob', body, worker_thread_num: 6, content_md5: 'oafL1+HS78x65+e39PGIIg==')
+        end
+      end
+    end
+  end
+
+  def test_multipart_save_block_blob_with_file_handle_success
+    i = 0
+    multiple_values = lambda do |*|
+      i += 1
+      return 'd' * 4 * 1024 * 1024 if i == 1
+      return 'd' * (1 * 1024 * 1024 + 1) if i == 2
+      return nil
+    end
+    temp_file = '/dev/null'
+    File.open(temp_file, 'r') do |file_handle|
+      file_handle.stub :read, multiple_values do
+        file_handle.stub :rewind, nil do
+          @service.stub :create_block_blob, @raw_cloud_blob do
+            @service.stub :put_blob_block, true do
+              @service.stub :commit_blob_blocks, true do
+                assert @service.multipart_save_block_blob('test_container', 'test_blob', file_handle, worker_thread_num: 0)
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+
+  def test_multipart_save_block_blob_with_file_handle_do_not_support_rewind_success
+    i = 0
+    multiple_values = lambda do |*|
+      i += 1
+      return 'd' * 4 * 1024 * 1024 if i == 1
+      return 'd' * (1 * 1024 * 1024 + 1) if i == 2
+      return nil
+    end
+    temp_file = '/dev/null'
+    exception = ->(*) { raise 'do not support rewind' }
+    File.open(temp_file, 'r') do |file_handle|
+      file_handle.stub :read, multiple_values do
+        file_handle.stub :rewind, exception do
+          @service.stub :create_block_blob, @raw_cloud_blob do
+            @service.stub :put_blob_block, true do
+              @service.stub :commit_blob_blocks, true do
+                assert @service.multipart_save_block_blob('test_container', 'test_blob', file_handle, worker_thread_num: 'invalid')
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+
+  def test_multipart_save_block_blob_http_exception
+    body = 'd' * 5 * 1024 * 1024 # MAXIMUM_CHUNK_SIZE is 4 * 1024 * 1024
+
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @service.stub :create_block_blob, http_exception do
+      @service.stub :delete_blob, true do
+        assert_raises(Azure::Core::Http::HTTPError) do
+          @service.multipart_save_block_blob('test_container', 'test_blob', body, {})
+        end
+      end
+    end
+  end
+
+  def test_multipart_save_block_blob_fail_when_delete_blob_http_exception
+    body = 'd' * 5 * 1024 * 1024 # MAXIMUM_CHUNK_SIZE is 4 * 1024 * 1024
+
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @service.stub :create_block_blob, http_exception do
+      @service.stub :delete_blob, http_exception do
+        assert_raises(Azure::Core::Http::HTTPError) do
+          @service.multipart_save_block_blob('test_container', 'test_blob', body, {})
+        end
+      end
+    end
+  end
+
+  def test_multipart_save_block_blob_mock
+    assert @mock_service.multipart_save_block_blob('test_container', 'test_blob', 'content', {})
+  end
+end

data/test/requests/storage/test_put_blob_block.rb

@@ -0,0 +1,34 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Blob Class
+class TestPutBlobBlock < Minitest::Test
+  # This class posesses the test cases for the requests of putting blob block.
+  def setup
+    Fog.mock!
+    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    Fog.unmock!
+    @mocked_response = mocked_storage_http_error
+
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @blob_client = @service.instance_variable_get(:@blob_client)
+  end
+
+  def test_put_blob_block_success
+    @blob_client.stub :put_blob_block, true do
+      assert @service.put_blob_block('test_container', 'test_blob', 'id', 'data')
+    end
+  end
+
+  def test_put_blob_block_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :put_blob_block, http_exception do
+      assert_raises(RuntimeError) do
+        @service.put_blob_block('test_container', 'test_blob', 'id', 'data')
+      end
+    end
+  end
+
+  def test_put_blob_block_mock
+    assert @mock_service.put_blob_block('test_container', 'test_blob', 'id', 'data')
+  end
+end

data/test/requests/storage/test_put_blob_metadata.rb

@@ -0,0 +1,36 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Blob Class
+class TestPutBlobMetadata < Minitest::Test
+  # This class posesses the test cases for the requests of setting blob metadata.
+  def setup
+    Fog.mock!
+    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    Fog.unmock!
+    @mocked_response = mocked_storage_http_error
+
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @blob_client = @service.instance_variable_get(:@blob_client)
+
+    @metadata = ApiStub::Requests::Storage::File.blob_metadata
+  end
+
+  def test_put_blob_metadata_success
+    @blob_client.stub :set_blob_metadata, true do
+      assert @service.put_blob_metadata('test_container', 'test_blob', @metadata)
+    end
+  end
+
+  def test_put_blob_metadata_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :set_blob_metadata, http_exception do
+      assert_raises(RuntimeError) do
+        @service.put_blob_metadata('test_container', 'test_blob', @metadata)
+      end
+    end
+  end
+
+  def test_put_blob_metadata_mock
+    assert @mock_service.put_blob_metadata('test_container', 'test_blob', @metadata)
+  end
+end

data/test/requests/storage/test_put_blob_pages.rb

@@ -0,0 +1,34 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Blob Class
+class TestPutBlobPages < Minitest::Test
+  # This class posesses the test cases for the requests of putting blob page.
+  def setup
+    Fog.mock!
+    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    Fog.unmock!
+    @mocked_response = mocked_storage_http_error
+
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @blob_client = @service.instance_variable_get(:@blob_client)
+  end
+
+  def test_put_blob_pages_success
+    @blob_client.stub :put_blob_pages, true do
+      assert @service.put_blob_pages('test_container', 'test_blob', 0, 1024, 'data')
+    end
+  end
+
+  def test_put_blob_pages_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :put_blob_pages, http_exception do
+      assert_raises(RuntimeError) do
+        @service.put_blob_pages('test_container', 'test_blob', 0, 1024, 'data')
+      end
+    end
+  end
+
+  def test_put_blob_pages_mock
+    assert @mock_service.put_blob_pages('test_container', 'test_blob', 0, 1024, 'data')
+  end
+end

data/test/requests/storage/test_put_blob_properties.rb

@@ -0,0 +1,39 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Blob Class
+class TestPutBlobProperties < Minitest::Test
+  # This class posesses the test cases for the requests of setting blob properties.
+  def setup
+    Fog.mock!
+    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    Fog.unmock!
+    @mocked_response = mocked_storage_http_error
+
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @blob_client = @service.instance_variable_get(:@blob_client)
+
+    @properties = {
+      content_language: 'english',
+      content_disposition: 'attachment'
+    }
+  end
+
+  def test_put_blob_properties_success
+    @blob_client.stub :set_blob_properties, true do
+      assert @service.put_blob_properties('test_container', 'test_blob', @properties)
+    end
+  end
+
+  def test_put_blob_properties_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :set_blob_properties, http_exception do
+      assert_raises(RuntimeError) do
+        @service.put_blob_properties('test_container', 'test_blob', @properties)
+      end
+    end
+  end
+
+  def test_put_blob_properties_mock
+    assert @mock_service.put_blob_properties('test_container', 'test_blob', @properties)
+  end
+end

data/test/requests/storage/test_put_container_acl.rb

@@ -0,0 +1,34 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Container Class
+class TestPutContainerACL < Minitest::Test
+  # This class posesses the test cases for the requests of setting storage container acl.
+  def setup
+    Fog.mock!
+    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    Fog.unmock!
+    @mocked_response = mocked_storage_http_error
+
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @blob_client = @service.instance_variable_get(:@blob_client)
+  end
+
+  def test_put_container_acl_success
+    @blob_client.stub :set_container_acl, true do
+      assert @service.put_container_acl('test_container', 'container')
+    end
+  end
+
+  def test_put_container_acl_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :set_container_acl, http_exception do
+      assert_raises(RuntimeError) do
+        @service.put_container_acl('test_container', 'container')
+      end
+    end
+  end
+
+  def test_put_container_acl_mock
+    assert @mock_service.put_container_acl('test_container', 'container')
+  end
+end

data/test/requests/storage/test_put_container_metadata.rb

@@ -0,0 +1,36 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Container Class
+class TestPutContainerMetadata < Minitest::Test
+  # This class posesses the test cases for the requests of container service.
+  def setup
+    Fog.mock!
+    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    Fog.unmock!
+    @mocked_response = mocked_storage_http_error
+
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @blob_client = @service.instance_variable_get(:@blob_client)
+
+    @metadata = ApiStub::Requests::Storage::Directory.container_metadata
+  end
+
+  def test_put_container_metadata_success
+    @blob_client.stub :set_container_metadata, true do
+      assert @service.put_container_metadata('test_container', @metadata)
+    end
+  end
+
+  def test_put_container_metadata_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :set_container_metadata, http_exception do
+      assert_raises(RuntimeError) do
+        @service.put_container_metadata('test_container', @metadata)
+      end
+    end
+  end
+
+  def test_put_container_metadata_mock
+    assert @mock_service.put_container_metadata('test_container', @metadata)
+  end
+end