require 'test/test_helper'
require 'fileutils'
require 'right_aws'
require 'time'

class RightAWSCommandsTest < Test::Unit::TestCase

  def setup
    @s3 = RightAws::S3Interface.new('1E3GDYEOGFJPIT7XXXXXX','hgTHt68JY07JKUY08ftHYtERkjgtfERn57XXXXXX',
                                    {:multi_thread => false, :server => 'localhost',
                                      :port => 10453, :protocol => 'http', :logger => Logger.new("/dev/null"),
                                      :no_subdomains => true })
  end

  def teardown
  end

  def test_create_bucket
    bucket = @s3.create_bucket("s3media")
    assert_not_nil bucket
  end

  def test_store
    @s3.put("s3media", "helloworld", "Hello World Man!")
    obj = @s3.get("s3media", "helloworld")
    assert_equal "Hello World Man!", obj[:object]
  end

  # TODO - get Chinese to work
  #def test_store_chinese
  #  ni_hao = "你好"
  #  great_wall = "中国的长城"
#
  #  @s3.put("s3media", ni_hao, great_wall)
  #  obj = @s3.get("s3media", ni_hao)
  #  assert_equal(great_wall, obj[:object])
  #end

  def test_store_not_found
    begin
      obj = @s3.get("s3media", "helloworldnotexist")
    rescue RightAws::AwsError
      assert $!.message.include?('NoSuchKey')
    rescue
      fail 'Should have caught NoSuchKey Exception'
    end
  end

  def test_large_store
    @s3.put("s3media", "helloworld", "Hello World Man!")
    buffer = ""
    500000.times do
      buffer << "#{(rand * 100).to_i}"
    end

    buf_len = buffer.length
    time_before = Time.now
    @s3.put("s3media", "big", buffer)

    output = ""
    @s3.get("s3media","big") do |chunk|
      output << chunk
    end
    time_after = Time.now

    assert(time_after - time_before < 2) # Should run in under 2 seconds on normal machines
    assert_equal(buf_len, output.size)
  end

  # Test that GET requests with a delimiter return a list of
  def test_list_by_delimiter
    @s3.create_bucket("s3media")

    @s3.put("s3media", "delimited/item", "item")

    expected_prefixes = []
    (1..50).each do |i|
      key_prefix = "delimited/%02d/" % i
      @s3.put("s3media", key_prefix + "foo", "foo")
      @s3.put("s3media", key_prefix + "fie", "fie")
      expected_prefixes << key_prefix
    end

    key_names = []
    common_prefixes = []
    @s3.incrementally_list_bucket("s3media", {:prefix => "delimited", :delimiter => '/'}) do |currentResponse|
      common_prefixes += currentResponse[:common_prefixes]
    end
    assert_equal ["delimited/"], common_prefixes

    common_prefixes = []
    @s3.incrementally_list_bucket("s3media", {:prefix => "delimited/", :delimiter => '/', "max-keys" => 100}) do |currentResponse|
      key_names += currentResponse[:contents].map do |key|
        key[:key]
      end
      common_prefixes += currentResponse[:common_prefixes]
    end
    assert_equal expected_prefixes, common_prefixes
    assert_equal ["delimited/item"], key_names
  end

  def test_multi_directory
    @s3.put("s3media", "dir/right/123.txt", "recursive")
    output = ""
    obj = @s3.get("s3media", "dir/right/123.txt") do |chunk|
      output << chunk
    end
    assert_equal "recursive", output
  end

  def test_intra_bucket_copy
    @s3.put("s3media", "original.txt", "Hello World")
    @s3.copy("s3media", "original.txt", "s3media", "copy.txt")
    obj = @s3.get("s3media", "copy.txt")
    assert_equal "Hello World", obj[:object]
  end

  def test_copy_in_place
    @s3.put("s3media", "copy-in-place", "Hello World")
    @s3.copy("s3media", "copy-in-place", "s3media","copy-in-place")
    obj = @s3.get("s3media", "copy-in-place")
    assert_equal "Hello World", obj[:object]
  end

  def test_content_encoding
    foo_compressed = Zlib::Deflate.deflate("foo")
    @s3.put("s3media", "foo", foo_compressed, {"content-encoding" => "gzip"})
    obj = @s3.get("s3media", "foo")
    # assert_equal "gzip", obj[:headers]["content-encoding"] # TODO why doesn't checking content-encoding work?
    assert_equal "gzip", obj[:headers]["x-content-encoding"] # TODO why doesn't checking content-encoding work?
  end

  # def test_content_encoding_data
  #   foo_compressed = Zlib::Deflate.deflate("foo-two")
  #   @s3.put("s3media", "foo-two", foo_compressed, {"content-encoding" => "gzip"})
  #   obj = @s3.get("s3media", "foo-two")
  #   puts "*** GOT HERE 1 #{ obj[:object] }"
  #   assert_equal "foo-two", Zlib::Inflate::inflate(obj[:object])
  # end

  def test_copy_replace_metadata
    @s3.put("s3media", "copy_replace", "Hello World", {"content-type" => "application/octet-stream"})
    obj = @s3.get("s3media", "copy_replace")
    assert_equal "Hello World", obj[:object]
    assert_equal "application/octet-stream", obj[:headers]["content-type"]
    @s3.copy("s3media", "copy_replace", "s3media", "copy_replace", :replace, {"content-type"=>"text/plain"})
    obj = @s3.get("s3media", "copy_replace")
    assert_equal "Hello World", obj[:object]
    assert_equal "text/plain", obj[:headers]["content-type"]
  end

  def test_larger_lists
    @s3.create_bucket('right_aws_many')
    (0..50).each do |i|
      ('a'..'z').each do |letter|
        name = "#{letter}#{i}"
        @s3.put('right_aws_many', name, 'asdf')
      end
    end

    keys = @s3.list_bucket('right_aws_many')
    assert_equal(1000, keys.size)
    assert_equal('a0', keys.first[:key])
  end

  def test_destroy_bucket
    @s3.create_bucket('deletebucket')
    @s3.delete_bucket('deletebucket')

    begin
      bucket = @s3.list_bucket('deletebucket')
      fail("Shouldn't succeed here")
    rescue RightAws::AwsError
      assert $!.message.include?('NoSuchBucket')
    rescue
      fail 'Should have caught NoSuchBucket Exception'
    end
  end

  def test_if_none_match
    @s3.put("s3media", "if_none_match_test", "Hello World 1!")
    obj = @s3.get("s3media", "if_none_match_test")
    tag = obj[:headers]["etag"]
    begin
      @s3.get("s3media", "if_none_match_test", {"If-None-Match" => tag})
    rescue URI::InvalidURIError
      # expected error for 304
    else
      fail 'Should have encountered an error due to the server not returning a response due to caching'
    end
    @s3.put("s3media", "if_none_match_test", "Hello World 2!")
    obj = @s3.get("s3media", "if_none_match_test", {"If-None-Match" => tag})
    assert_equal "Hello World 2!", obj[:object]
  end

  def test_if_modified_since
    @s3.put("s3media", "if_modified_since_test", "Hello World 1!")
    obj = @s3.get("s3media", "if_modified_since_test")
    modified = obj[:headers]["last-modified"]
    begin
      @s3.get("s3media", "if_modified_since_test", {"If-Modified-Since" => modified})
    rescue URI::InvalidURIError
      # expected error for 304
    else
      fail 'Should have encountered an error due to the server not returning a response due to caching'
    end
    # Granularity of an HTTP Date is 1 second which isn't enough for the test
    # so manually rewind the clock by a second
    time_in_the_past = Time.httpdate(modified) - 1
    begin
      obj = @s3.get("s3media", "if_modified_since_test", {"If-Modified-Since" => time_in_the_past.httpdate})
    rescue
      fail 'Should have been downloaded since the date is in the past now'
    else
      #expected scenario
    end
  end

end