gitlab-fog-azure-rm 0.6.0

data/test/requests/storage/test_get_blob_http_url.rb

@@ -0,0 +1,30 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Blob Class
+class TestGetBlobHttpUrl < 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.gsub('https:', 'http:')
+    @token = ApiStub::Requests::Storage::File.blob_url_token
+  end
+
+  def test_get_blob_http_url_success
+    @blob_client.stub :generate_uri, @url do
+      @signature_client.stub :generate_service_sas_token, @token do
+        assert_equal "#{@url}?#{@token}", @service.get_blob_http_url('test_container', 'test_blob', Time.now.utc + 3600)
+      end
+    end
+  end
+
+  def test_get_blob_http_url_mock
+    assert_equal "#{@url}?#{@token}", @mock_service.get_blob_http_url('test_container', 'test_blob', Time.now.utc + 3600)
+  end
+end

data/test/requests/storage/test_get_blob_https_url.rb

@@ -0,0 +1,32 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Blob Class
+class TestGetBlobHttpsUrl < 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_get_blob_https_url_success
+    @blob_client.stub :generate_uri, @url do
+      @signature_client.stub :generate_service_sas_token, @token do
+        assert_equal "#{@url}?#{@token}", @service.get_blob_https_url('test_container', 'test_blob', Time.now.utc + 3600)
+        assert_equal "#{@url}?#{@token}", @service.get_object_url('test_container', 'test_blob', Time.now.utc + 3600)
+      end
+    end
+  end
+
+  def test_get_blob_https_url_mock
+    assert_equal "#{@url}?#{@token}", @mock_service.get_blob_https_url('test_container', 'test_blob', Time.now.utc + 3600)
+    assert_equal "#{@url}?#{@token}", @mock_service.get_object_url('test_container', 'test_blob', Time.now.utc + 3600)
+  end
+end

data/test/requests/storage/test_get_blob_properties.rb

@@ -0,0 +1,47 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Blob Class
+class TestGetBlobProperties < Minitest::Test
+  # This class posesses the test cases for the requests of getting storage 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)
+
+    @blob = ApiStub::Requests::Storage::File.blob
+  end
+
+  def test_get_blob_properties_success
+    @blob_client.stub :get_blob_properties, @blob do
+      assert_equal @blob, @service.get_blob_properties('test_container', 'test_blob')
+    end
+  end
+
+  def test_get_blob_properties_not_found
+    exception = ->(*) { raise StandardError.new('Not found(404). Not exist') }
+    @blob_client.stub :get_blob_properties, exception do
+      assert_raises('NotFound') do
+        @service.get_blob_properties('test_container', 'test_blob')
+      end
+    end
+  end
+
+  def test_get_blob_properties_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :get_blob_properties, http_exception do
+      assert_raises(Azure::Core::Http::HTTPError) do
+        @service.get_blob_properties('test_container', 'test_blob')
+      end
+    end
+  end
+
+  def test_get_blob_properties_mock
+    mock_blob_properties = @mock_service.get_blob_properties('test_container', 'test_blob')
+    assert_equal @blob.name, mock_blob_properties.name
+    assert_equal @blob.properties, mock_blob_properties.properties
+  end
+end

data/test/requests/storage/test_get_blob_url.rb

@@ -0,0 +1,75 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Blob Class
+class TestGetBlobUrl < 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!
+  end
+
+  def test_get_blob_url_success
+    service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    blob_client = service.instance_variable_get(:@blob_client)
+    url = ApiStub::Requests::Storage::File.blob_https_url
+
+    blob_client.stub :generate_uri, url do
+      assert_equal url, service.get_blob_url('test_container', 'test_blob')
+
+      options = { scheme: 'http' }
+      assert_equal url.gsub('https:', 'http:'), service.get_blob_url('test_container', 'test_blob', options)
+    end
+  end
+
+  def test_get_blob_url_for_china_success
+    china_storage_account_credentials = storage_account_credentials.merge(environment: ENVIRONMENT_AZURE_CHINA_CLOUD)
+    service = Fog::Storage::AzureRM.new(china_storage_account_credentials)
+    blob_client = service.instance_variable_get(:@blob_client)
+    url = ApiStub::Requests::Storage::File.blob_https_url(ENVIRONMENT_AZURE_CHINA_CLOUD)
+
+    blob_client.stub :generate_uri, url do
+      assert_equal url, service.get_blob_url('test_container', 'test_blob')
+
+      options = { scheme: 'http' }
+      assert_equal url.gsub('https:', 'http:'), service.get_blob_url('test_container', 'test_blob', options)
+    end
+  end
+
+  def test_get_blob_url_for_us_success
+    us_storage_account_credentials = storage_account_credentials.merge(environment: ENVIRONMENT_AZURE_US_GOVERNMENT)
+    service = Fog::Storage::AzureRM.new(us_storage_account_credentials)
+    blob_client = service.instance_variable_get(:@blob_client)
+    url = ApiStub::Requests::Storage::File.blob_https_url(ENVIRONMENT_AZURE_US_GOVERNMENT)
+
+    blob_client.stub :generate_uri, url do
+      assert_equal url, service.get_blob_url('test_container', 'test_blob')
+
+      options = { scheme: 'http' }
+      assert_equal url.gsub('https:', 'http:'), service.get_blob_url('test_container', 'test_blob', options)
+    end
+  end
+
+  def test_get_blob_url_for_german_success
+    german_storage_account_credentials = storage_account_credentials.merge(environment: ENVIRONMENT_AZURE_GERMAN_CLOUD)
+    service = Fog::Storage::AzureRM.new(german_storage_account_credentials)
+    blob_client = service.instance_variable_get(:@blob_client)
+    url = ApiStub::Requests::Storage::File.blob_https_url(ENVIRONMENT_AZURE_GERMAN_CLOUD)
+
+    blob_client.stub :generate_uri, url do
+      assert_equal url, service.get_blob_url('test_container', 'test_blob')
+
+      options = { scheme: 'http' }
+      assert_equal url.gsub('https:', 'http:'), service.get_blob_url('test_container', 'test_blob', options)
+    end
+  end
+
+  def test_get_blob_url_mock
+    url = ApiStub::Requests::Storage::File.blob_https_url
+    assert_equal url, @mock_service.get_blob_url('test_container', 'test_blob')
+
+    options = { scheme: 'http' }
+    http_url = url.gsub('https:', 'http:')
+    assert_equal http_url, @mock_service.get_blob_url('test_container', 'test_blob', options)
+  end
+end

data/test/requests/storage/test_get_container_acl.rb

@@ -0,0 +1,37 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Container Class
+class TestGetContainerAcl < Minitest::Test
+  # This class posesses the test cases for the requests of getting 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)
+
+    @raw_container_acl = ApiStub::Requests::Storage::Directory.raw_container_acl
+    @container_acl = ApiStub::Requests::Storage::Directory.container_acl
+  end
+
+  def test_get_container_acl_success
+    @blob_client.stub :get_container_acl, @raw_container_acl do
+      assert_equal @container_acl, @service.get_container_acl('test_container')
+    end
+  end
+
+  def test_get_container_acl_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :get_container_acl, http_exception do
+      assert_raises(Azure::Core::Http::HTTPError) do
+        @service.get_container_acl('test_container')
+      end
+    end
+  end
+
+  def test_get_container_acl_mock
+    assert_equal @container_acl, @mock_service.get_container_acl('test_container')
+  end
+end

data/test/requests/storage/test_get_container_properties.rb

@@ -0,0 +1,45 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Container Class
+class TestGetContainerProperties < Minitest::Test
+  # This class posesses the test cases for the requests of getting storage container 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)
+
+    @container = ApiStub::Requests::Storage::Directory.container
+  end
+
+  def test_get_container_properties_success
+    @blob_client.stub :get_container_properties, @container do
+      assert_equal @container, @service.get_container_properties('test_container')
+    end
+  end
+
+  def test_get_container_properties_not_found
+    exception = ->(_name, _option) { raise StandardError.new('Not found(404). Not exist') }
+    @blob_client.stub :get_container_properties, exception do
+      assert_raises('NotFound') do
+        @service.get_container_properties('test_container')
+      end
+    end
+  end
+
+  def test_get_container_properties_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :get_container_properties, http_exception do
+      assert_raises(Azure::Core::Http::HTTPError) do
+        @service.get_container_properties('test_container')
+      end
+    end
+  end
+
+  def test_get_container_properties_mock
+    assert_equal @container, @mock_service.get_container_properties('test_container')
+  end
+end

data/test/requests/storage/test_get_container_url.rb

@@ -0,0 +1,33 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Container Class
+class TestGetContainerUrl < Minitest::Test
+  # This class posesses the test cases for the requests of getting storage container url.
+  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)
+
+    @url = ApiStub::Requests::Storage::Directory.container_https_url
+  end
+
+  def test_get_container_url_success
+    @blob_client.stub :generate_uri, @url do
+      assert_equal @url, @service.get_container_url('test_container')
+
+      options = { scheme: 'http' }
+      assert_equal @url.gsub('https:', 'http:'), @service.get_container_url('test_container', options)
+    end
+  end
+
+  def test_get_container_url_mock
+    assert_equal @url, @mock_service.get_container_url('test_container')
+
+    options = { scheme: 'http' }
+    http_url = @url.gsub('https:', 'http:')
+    assert_equal http_url, @mock_service.get_container_url('test_container', options)
+  end
+end

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

@@ -0,0 +1,54 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Storage Container Class
+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)
+
+    @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_http_exception
+    http_exception = ->(*) { raise Azure::Core::Http::HTTPError.new(@mocked_response) }
+    @blob_client.stub :list_containers, http_exception do
+      assert_raises(Azure::Core::Http::HTTPError) do
+        @service.list_containers
+      end
+    end
+  end
+
+  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