far-gem 0.5.1

data/test/models/storage/test_file.rb

@@ -0,0 +1,294 @@
+require File.expand_path '../../test_helper', __dir__
+
+# Test class for Storage Container Model
+class TestFile < Minitest::Test
+  def setup
+    @service = Fog::Storage::AzureRM.new(storage_account_credentials)
+    @directory = directory(@service)
+    @file = file(@service)
+    @raw_cloud_blob = storage_blob
+
+    @mocked_response = mocked_storage_http_error
+    @blob = ApiStub::Models::Storage::File.blob
+    @metadata = ApiStub::Models::Storage::File.blob_metadata
+    @blob_list = ApiStub::Models::Storage::File.blob_list
+    @blob_https_url = ApiStub::Models::Storage::File.blob_https_url
+  end
+
+  def test_model_methods
+    methods = [
+      :save,
+      :body,
+      :body=,
+      :copy,
+      :copy_from_uri,
+      :destroy,
+      :public?,
+      :public_url,
+      :url
+    ]
+    methods.each do |method|
+      assert_respond_to @file, method
+    end
+  end
+
+  def test_model_attributes
+    attributes = [
+      :key,
+      :body,
+      :directory,
+      :accept_ranges,
+      :content_length,
+      :content_type,
+      :content_md5,
+      :content_encoding,
+      :content_language,
+      :cache_control,
+      :content_disposition,
+      :copy_completion_time,
+      :copy_status,
+      :copy_status_description,
+      :copy_id,
+      :copy_progress,
+      :copy_source,
+      :etag,
+      :last_modified,
+      :lease_duration,
+      :lease_state,
+      :lease_status,
+      :sequence_number,
+      :blob_type,
+      :metadata
+    ]
+    attributes.each do |attribute|
+      assert_respond_to @file, attribute
+    end
+  end
+
+  def test_save_method_with_small_block_blob_success
+    @file.body = 'd' * 1025 # SINGLE_BLOB_PUT_THRESHOLD is 32 * 1024 * 1024
+
+    @service.stub :create_block_blob, @raw_cloud_blob do
+      assert @file.save
+    end
+  end
+
+  def test_save_method_with_large_block_blob_success
+    @file.body = 'd' * (32 * 1024 * 1024 + 1) # SINGLE_BLOB_PUT_THRESHOLD is 32 * 1024 * 1024
+
+    @service.stub :multipart_save_block_blob, true do
+      @service.stub :get_blob_properties, @raw_cloud_blob do
+        assert @file.save
+      end
+    end
+  end
+
+  def test_save_method_with_page_blob_success
+    options = { blob_type: 'PageBlob' }
+    @file.metadata = @metadata
+    @file.body = 'd' * 5 * 1024 * 1024 # MAXIMUM_CHUNK_SIZE is 4 * 1024 * 1024
+
+    @service.stub :save_page_blob, true do
+      @service.stub :get_blob_properties, @raw_cloud_blob do
+        assert @file.save(options)
+      end
+    end
+  end
+
+  def test_save_method_with_not_update_body_success
+    options = { update_body: false }
+    @file.metadata = @metadata
+
+    @service.stub :put_blob_metadata, true do
+      @service.stub :put_blob_properties, true do
+        @service.stub :get_blob_properties, @raw_cloud_blob do
+          assert @file.save(options)
+        end
+      end
+    end
+  end
+
+  def test_save_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @file.attributes.delete(:directory)
+      @file.save
+    end
+  end
+
+  def test_save_method_without_key_exception
+    assert_raises(ArgumentError) do
+      @file.attributes.delete(:key)
+      @file.save
+    end
+  end
+
+  def test_save_method_create_without_body_exception
+    assert_raises(ArgumentError) do
+      @file.attributes.delete(:body)
+      @file.save
+    end
+  end
+
+  def test_get_body_method_local_success
+    @file.attributes[:body] = 'body'
+    assert_equal 'body', @file.body
+
+    @file.attributes[:body] = nil
+    @file.attributes[:last_modified] = nil
+    assert_equal '', @file.body
+  end
+
+  def test_get_body_method_remote_success
+    @file.attributes[:body] = nil
+    remote_file = @file.dup
+    body_content = 'data'
+    @file.collection.stub :get, remote_file do
+      remote_file.stub :body, body_content do
+        body = @file.body
+        assert_equal body_content, body
+        assert_equal body_content, @file.attributes[:body]
+      end
+    end
+  end
+
+  def test_get_body_method_remote_not_found_success
+    @file.attributes[:body] = nil
+    @file.collection.stub :get, nil do
+      body = @file.body
+      assert_equal '', body
+      assert_equal '', @file.attributes[:body]
+    end
+  end
+
+  def test_set_body_method_success
+    @file.body = 'new_body'
+    assert_equal 'new_body', @file.attributes[:body]
+  end
+
+  def test_copy_method_success
+    copy_id = @raw_cloud_blob.properties[:copy_id]
+    copy_status = 'pending'
+
+    @service.stub :get_blob_properties, @raw_cloud_blob do
+      @service.stub :copy_blob, [copy_id, copy_status] do
+        @service.stub :wait_blob_copy_operation_to_finish, true do
+          target_file = @file.copy('target_container', 'target_blob')
+          assert_instance_of Fog::Storage::AzureRM::File, target_file
+        end
+      end
+    end
+  end
+
+  def test_copy_from_uri_method_success
+    copy_id = @raw_cloud_blob.properties[:copy_id]
+    copy_status = 'pending'
+
+    @service.stub :get_blob_properties, @raw_cloud_blob do
+      @service.stub :copy_blob_from_uri, [copy_id, copy_status] do
+        @service.stub :wait_blob_copy_operation_to_finish, true do
+          assert @file.copy_from_uri('source_uri')
+        end
+      end
+    end
+  end
+
+  def test_copy_from_uri_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @file.attributes.delete(:directory)
+      @file.copy_from_uri('source_uri')
+    end
+  end
+
+  def test_copy_from_uri_method_without_key_exception
+    assert_raises(ArgumentError) do
+      @file.attributes.delete(:key)
+      @file.copy_from_uri('source_uri')
+    end
+  end
+
+  def test_destroy_method_success
+    @service.stub :delete_blob, true do
+      assert @file.destroy
+      assert @file.attributes[:body].nil?
+    end
+  end
+
+  def test_destroy_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @file.attributes.delete(:directory)
+      @file.destroy
+    end
+  end
+
+  def test_destroy_method_without_key_exception
+    assert_raises(ArgumentError) do
+      @file.attributes.delete(:key)
+      @file.destroy
+    end
+  end
+
+  def test_public_method_success
+    @file.directory.acl = 'container'
+    assert @file.public?
+
+    @file.directory.acl = 'blob'
+    assert @file.public?
+
+    @file.directory.acl = nil
+    assert !@file.public?
+  end
+
+  def test_public_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @file.attributes.delete(:directory)
+      @file.public?
+    end
+  end
+
+  def test_public_url_method_success
+    @file.stub :public?, false do
+      assert @file.public_url.nil?
+    end
+
+    @file.stub :public?, true do
+      @service.stub :get_blob_url, @blob_https_url do
+        assert @file.public_url, @blob_https_url
+      end
+    end
+
+    http_url = @blob_https_url.gsub('https:', 'http:')
+    options = { scheme: 'https' }
+    @file.stub :public?, true do
+      @service.stub :get_blob_url, http_url do
+        assert @file.public_url(options), http_url
+      end
+    end
+  end
+
+  def test_public_url_method_without_directory_exception
+    assert_raises(ArgumentError) do
+      @file.attributes.delete(:directory)
+      @file.public_url
+    end
+  end
+
+  def test_public_url_method_without_key_exception
+    assert_raises(ArgumentError) do
+      @file.attributes.delete(:key)
+      @file.public_url
+    end
+  end
+
+  def test_url_method_success
+    @file.collection.stub :get_url, @blob_https_url do
+      assert @file.url(Time.now + 3600), @blob_https_url
+    end
+  end
+
+  def test_url_method_without_key_exception
+    assert_raises(ArgumentError) do
+      @file.attributes.delete(:key)
+      @file.url(Time.now + 3600)
+    end
+  end
+end

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