require "simplecov"
SimpleCov.start

require "test/unit"
require "rack/test"
require "rack/revision"
require "fileutils"

class TestRevision < Test::Unit::TestCase
  include Rack::Test::Methods

  def default_app
    lambda do |env|
      headers = {"Content-Type" => "text/html"}
      [200, headers, ["OK"]]
    end
  end

  def app
    @app ||= Rack::Revision.new(default_app)
  end

  def app_url
    "http://foobar.com/"
  end

  attr_writer :app

  def setup
    FileUtils.mkdir("./test/tmp")
  end

  def teardown
    FileUtils.rm_rf("./test/tmp")
  end

  def test_revision_request
    self.app.reset_revision
    get app_url

    assert last_response.ok?
  end

  def test_presence
    self.app.reset_revision
    get app_url

    assert_not_nil last_response.headers["X-REVISION"]
    assert_not_nil last_response.headers["X-Revision"]
  end

  def test_blank
    self.app = Rack::Revision.new(default_app, :header => false)
    self.app.reset_revision
    get app_url

    assert_nil last_response.headers["X-REVISION"]
    assert_nil last_response.headers["X-Revision"]
  end

  def test_default_value
    self.app.reset_revision
    get app_url

    assert_equal "UNDEFINED", last_response.headers["X-Revision"]
  end

  def test_custom_value
    self.app = Rack::Revision.new(default_app, :default => "FOOBAR")
    self.app.reset_revision

    get app_url
    assert_equal "FOOBAR", last_response.headers["X-Revision"]
  end

  def test_custom_header
    self.app = Rack::Revision.new(default_app, :header => "FOOBAR")
    self.app.reset_revision

    get app_url
    assert_not_nil last_response.headers["FOOBAR"]
  end

  def test_custom_filename
    File.open("./test/tmp/REVISION", "w") { |f| f.write("qwe123") }

    self.app = Rack::Revision.new(default_app, :filename => "./test/tmp/REVISION")
    self.app.reset_revision

    get app_url
    assert_equal "qwe123", last_response.headers["X-Revision"]
  end

  def test_custom_filename_starting_from_root
    File.open("./test/tmp/REVISION", "w") { |f| f.write("qwe123") }
    filename = File.expand_path("./test/tmp/REVISION")

    self.app = Rack::Revision.new(default_app, :filename => filename)
    self.app.reset_revision

    get app_url
    assert_equal "qwe123", last_response.headers["X-Revision"]
  end

  def test_dir_does_not_exist
    File.write("./test/tmp/REVISION", "example")

    Dir.chdir("./test/tmp") do
      self.app.reset_revision

      get app_url
      assert_equal "example", last_response.headers["X-Revision"]

      FileUtils.rm_rf("../tmp")
      self.app.reset_revision

      get app_url
      assert_equal "UNDEFINED", last_response.headers["X-Revision"]
    end
  end

  def test_env_is_present
    self.app.reset_revision
    get app_url

    assert_not_nil last_request.env["rack.app_revision"]
  end

  def test_custom_env
    self.app = Rack::Revision.new(default_app, :rack_env => "rack.custom_env")
    self.app.reset_revision
    get app_url

    assert_nil last_request.env["rack.app_revision"]
    assert_not_nil last_request.env["rack.custom_env"]
  end

  def test_disable_env
    self.app = Rack::Revision.new(default_app, :rack_env => false)
    self.app.reset_revision
    get app_url

    assert_nil last_request.env["rack.app_revision"]
  end

  def test_from_env_var
    self.app = Rack::Revision.new(default_app)
    self.app.reset_revision

    ENV["RACK_REVISION"] = "revision"
    get app_url
    assert_equal "revision", last_response.headers["X-Revision"]
  end

  def test_from_custom_env_var
    self.app = Rack::Revision.new(default_app, env_var: "SERVICE_REVISION")
    self.app.reset_revision

    ENV["RACK_REVISION"] = "bad"
    ENV["SERVICE_REVISION"] = "good"
    get app_url
    assert_equal "good", last_response.headers["X-Revision"]
  end
end