gitlab-fog-azure-rm 0.6.0

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(Azure::Core::Http::HTTPError) 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_https_url.rb

@@ -0,0 +1,32 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Blob Class
+class TestPutBlobHttpUrl < Minitest::Test
+  # This class posesses the test cases for the requests of Blob service.
+  def setup
+    Fog.mock!
+    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    Fog.unmock!
+
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @blob_client = @service.instance_variable_get(:@blob_client)
+    @signature_client = @service.instance_variable_get(:@signature_client)
+
+    @url = ApiStub::Requests::Storage::File.blob_https_url
+    @token = ApiStub::Requests::Storage::File.blob_url_token
+  end
+
+  def test_put_blob_http_urls_success
+    @blob_client.stub :generate_uri, @url do
+      @signature_client.stub :generate_service_sas_token, @token do
+        assert_equal "#{@url}?#{@token}", @service.put_blob_https_url('test_container', 'test_blob', Time.now.utc + 3600)
+        assert_equal "#{@url}?#{@token}", @service.put_object_url('test_container', 'test_blob', Time.now.utc + 3600, {})
+      end
+    end
+  end
+
+  def test_put_blob_https_url_mock
+    assert_equal "#{@url}?#{@token}", @mock_service.put_blob_https_url('test_container', 'test_blob', Time.now.utc + 3600)
+    assert_equal "#{@url}?#{@token}", @mock_service.put_object_url('test_container', 'test_blob', Time.now.utc + 3600, {})
+  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(Azure::Core::Http::HTTPError) 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(Azure::Core::Http::HTTPError) 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(Azure::Core::Http::HTTPError) 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(Azure::Core::Http::HTTPError) 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(Azure::Core::Http::HTTPError) 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

data/test/requests/storage/test_release_blob_lease.rb

@@ -0,0 +1,34 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Blob Class
+class TestReleaseBlobLease < Minitest::Test
+  # This class posesses the test cases for the requests of releasing blob lease.
+  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_release_blob_lease_success
+    @blob_client.stub :release_blob_lease, true do
+      assert @service.release_blob_lease('test_container', 'test_blob', 'lease_id')
+    end
+  end
+
+  def test_release_blob_lease_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :release_blob_lease, http_exception do
+      assert_raises(Azure::Core::Http::HTTPError) do
+        @service.release_blob_lease('test_container', 'test_blob', 'lease_id')
+      end
+    end
+  end
+
+  def test_release_blob_lease_mock
+    assert @mock_service.release_blob_lease('test_container', 'test_blob', 'lease_id')
+  end
+end

data/test/requests/storage/test_release_container_lease.rb

@@ -0,0 +1,34 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Container Class
+class TestReleaseContainerLease < Minitest::Test
+  # This class posesses the test cases for the requests of releasing container lease.
+  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_release_container_lease_success
+    @blob_client.stub :release_container_lease, true do
+      assert @service.release_container_lease('test_container', 'lease_id')
+    end
+  end
+
+  def test_release_container_lease_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :release_container_lease, http_exception do
+      assert_raises(Azure::Core::Http::HTTPError) do
+        @service.release_container_lease('test_container', 'lease_id')
+      end
+    end
+  end
+
+  def test_release_container_lease_mock
+    assert @mock_service.release_container_lease('test_container', 'lease_id')
+  end
+end

data/test/requests/storage/test_save_page_blob.rb

@@ -0,0 +1,115 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Blob Class
+class TestSavePageBlob < Minitest::Test
+  # This class posesses the test cases for the requests of saving storage page blob.
+  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_save_page_blob_success
+    body = 'd' * 5 * 1024 * 1024 # MAXIMUM_CHUNK_SIZE is 4 * 1024 * 1024
+
+    @service.stub :create_page_blob, true do
+      @service.stub :put_blob_pages, true do
+        assert @service.save_page_blob('test_container', 'test_blob', body, worker_thread_num: 6, content_md5: 'oafL1+HS78x65+e39PGIIg==')
+      end
+    end
+  end
+
+  def test_save_page_blob_with_file_handle_success
+    data_length = 5 * 1024 * 1024 # MAXIMUM_CHUNK_SIZE is 4 * 1024 * 1024
+    i = 0
+    multiple_values = lambda do |*|
+      i += 1
+      return 'd' * 4 * 1024 * 1024 if i == 1
+      return 'd' * 1 * 1024 * 1024 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 :size, data_length do
+          file_handle.stub :rewind, nil do
+            @service.stub :create_page_blob, true do
+              @service.stub :put_blob_pages, true do
+                assert @service.save_page_blob('test_container', 'test_blob', file_handle, worker_thread_num: 0)
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+
+  def test_save_page_blob_with_file_handle_do_not_support_rewind_success
+    data_length = 5 * 1024 * 1024 # MAXIMUM_CHUNK_SIZE is 4 * 1024 * 1024
+    i = 0
+    multiple_values = lambda do |*|
+      i += 1
+      return 'd' * 4 * 1024 * 1024 if i == 1
+      return 'd' * 1 * 1024 * 1024 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 :size, data_length do
+          file_handle.stub :rewind, exception do
+            @service.stub :create_page_blob, true do
+              @service.stub :put_blob_pages, true do
+                assert @service.save_page_blob('test_container', 'test_blob', file_handle, worker_thread_num: 'invalid')
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+
+  def test_save_page_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_page_blob, http_exception do
+      @service.stub :delete_blob, true do
+        assert_raises(Azure::Core::Http::HTTPError) do
+          @service.save_page_blob('test_container', 'test_blob', body, {})
+        end
+      end
+    end
+  end
+
+  def test_save_page_blob_invalid_size_exception
+    body = 'd' * 1025 # The page blob size must be aligned to a 512-byte boundary.
+
+    @service.stub :delete_blob, true do
+      assert_raises(RuntimeError) do
+        @service.save_page_blob('test_container', 'test_blob', body, {})
+      end
+    end
+  end
+
+  def test_save_page_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_page_blob, http_exception do
+      @service.stub :delete_blob, http_exception do
+        assert_raises(Azure::Core::Http::HTTPError) do
+          @service.save_page_blob('test_container', 'test_blob', body, {})
+        end
+      end
+    end
+  end
+
+  def test_save_page_blob_mock
+    assert @mock_service.save_page_blob('test_container', 'test_blob', 'content', {})
+  end
+end

data/test/requests/storage/test_wait_blob_copy_operation_to_finish.rb

@@ -0,0 +1,148 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Blob Class
+class TestWaitBlobCopyOperationToFinish < Minitest::Test
+  # This class posesses the test cases for the requests of waiting storage blob copy operation to finish.
+  def setup
+    Fog.mock!
+    @mock_service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    Fog.unmock!
+    @mocked_response = mocked_storage_http_error
+    @blob = storage_blob
+
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @blob_client = @service.instance_variable_get(:@blob_client)
+  end
+
+  def test_wait_blob_copy_operation_to_finish_with_success
+    copy_id = @blob.properties[:copy_id]
+    copy_status = 'pending'
+
+    i = 0
+    multiple_values = lambda do |*|
+      i += 1
+      if i == 1
+        @blob.properties[:copy_status] = copy_status
+        @blob.properties[:copy_progress] = '0/1024'
+        @blob.properties[:copy_status_description] = 'in progress'
+        return @blob
+      end
+
+      if i == 2
+        @blob.properties[:copy_status] = copy_status
+        @blob.properties[:copy_progress] = '2/1024'
+        @blob.properties[:copy_status_description] = 'in progress'
+        return @blob
+      end
+
+      if i == 3
+        @blob.properties[:copy_status] = copy_status
+        @blob.properties[:copy_progress] = '1023/1024'
+        @blob.properties[:copy_status_description] = 'in progress'
+        return @blob
+      end
+
+      @blob.properties[:copy_status] = 'success'
+      @blob.properties[:copy_progress] = '1024/1024'
+      @blob.properties[:copy_status_description] = 'finish'
+      @blob
+    end
+
+    @service.stub :get_blob_properties, multiple_values do
+      assert @service.wait_blob_copy_operation_to_finish('test_container', 'test_blob', copy_id, copy_status)
+    end
+  end
+
+  def test_wait_blob_copy_operation_to_finish_with_copy_finished_success
+    copy_id = nil
+    copy_status = 'success'
+
+    @service.stub :get_blob_properties, @blob do
+      assert @service.wait_blob_copy_operation_to_finish('test_container', 'test_blob', copy_id, copy_status)
+    end
+  end
+
+  def test_wait_blob_copy_operation_to_finish_with_copy_id_is_nil
+    copy_id = nil
+    copy_status = 'pending'
+
+    @service.stub :get_blob_properties, @blob do
+      assert @service.wait_blob_copy_operation_to_finish('test_container', 'test_blob', copy_id, copy_status)
+    end
+  end
+
+  def test_wait_blob_copy_operation_to_finish_with_copy_id_not_match_exception
+    copy_id = 'copy_id'
+    copy_status = 'pending'
+
+    @service.stub :get_blob_properties, @blob do
+      @service.stub :delete_blob, true do
+        assert_raises(RuntimeError) do
+          @service.wait_blob_copy_operation_to_finish('test_container', 'test_blob', copy_id, copy_status)
+        end
+      end
+    end
+  end
+
+  def test_wait_blob_copy_operation_to_finish_with_timeout_exception
+    copy_id = @blob.properties[:copy_id]
+    copy_status = 'pending'
+
+    i = 0
+    multiple_values = lambda do |*|
+      i += 1
+      if i == 1
+        @blob.properties[:copy_status] = copy_status
+        @blob.properties[:copy_progress] = '0/1024'
+        @blob.properties[:copy_status_description] = 'in progress'
+        return @blob
+      end
+
+      @blob.properties[:copy_status] = copy_status
+      @blob.properties[:copy_progress] = '0/1024'
+      @blob.properties[:copy_status_description] = 'in progress'
+      @blob
+    end
+
+    @service.stub :get_blob_properties, multiple_values do
+      @service.stub :delete_blob, true do
+        assert_raises(TimeoutError) do
+          @service.wait_blob_copy_operation_to_finish('test_container', 'test_blob', copy_id, copy_status, 2)
+        end
+      end
+    end
+  end
+
+  def test_wait_blob_copy_operation_to_finish_with_fail_exception
+    copy_id = @blob.properties[:copy_id]
+    copy_status = 'pending'
+
+    @blob.properties[:copy_status] = 'failed'
+    @service.stub :get_blob_properties, @blob do
+      @service.stub :delete_blob, true do
+        assert_raises(RuntimeError) do
+          @service.wait_blob_copy_operation_to_finish('test_container', 'test_blob', copy_id, copy_status)
+        end
+      end
+    end
+  end
+
+  def test_wait_blob_copy_operation_to_finish_with_fail_when_delete_blob_http_exception
+    copy_id = @blob.properties[:copy_id]
+    copy_status = 'pending'
+
+    @blob.properties[:copy_status] = 'failed'
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @service.stub :get_blob_properties, @blob do
+      @service.stub :delete_blob, http_exception do
+        assert_raises(RuntimeError) do
+          @service.wait_blob_copy_operation_to_finish('test_container', 'test_blob', copy_id, copy_status)
+        end
+      end
+    end
+  end
+
+  def test_wait_blob_copy_operation_to_finish_mock
+    assert @mock_service.wait_blob_copy_operation_to_finish('test_container', 'test_blob', 'copy_id', 'pending')
+  end
+end