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