require 'rack/test'
require 'rack/geo_locale'
require 'geoip'

include Rack::Test::Methods

def app
  Rack::GeoLocale.new(proc {|env| [200, {}, 'hello']})
end

describe Rack::GeoLocale do
  describe "GeoIP" do
    before do
      geoip = double("geoip")
      geoip.stub(:country).with("10.0.0.1").and_return(GeoIP::Country.new("10.0.0.1", "10.0.0.1", 191, "SE", "SWE", "Sweden", "EU"))
      geoip.stub(:country).with("10.0.0.2").and_return(GeoIP::Country.new("10.0.0.2", "10.0.0.2", 225, "US", "USA", "USA", "NA"))
      geoip.stub(:country).with("10.0.0.3").and_return(GeoIP::Country.new("10.0.0.3", "10.0.0.3", 0, "--", "---", "N/A", "--"))

      GeoIP.stub :new => geoip
    end

    it "should handle an empty REMOTE_ADDR field" do
      get '/', {}, {"REMOTE_ADDR" => nil}
      last_request.env["locale.country"].should == nil
    end

    it "should take HTTP_X_FORWARDED_FOR before REMOTE_ADDR field" do
      get '/', {}, {"REMOTE_ADDR" => "10.0.0.1", "HTTP_X_FORWARDED_FOR" => "10.0.0.2"}
      last_request.env["locale.country"].should == "US"
    end

    it "should resolve 10.0.0.1 to SE" do
      get '/', {}, {"REMOTE_ADDR" => "10.0.0.1"}
      last_request.env["locale.country"].should == "SE"
    end

    it "should resolve 10.0.0.2 to US" do
      get '/', {}, {"REMOTE_ADDR" => "10.0.0.2"}
      last_request.env["locale.country"].should == "US"
    end

    it "should resolve 10.0.0.3 to nil" do
      get '/', {}, {"REMOTE_ADDR" => "10.0.0.3"}
      last_request.env["locale.country"].should == nil
    end
  end

  describe "parsing HTTP_ACCEPT_LANGUAGE" do
    it "should return an empty result if no HTTP_ACCEPT_LANGUAGE passed" do
      get '/', {}, {}
      last_request.env["locale.language"].should == nil
    end

    it "should parse HTTP_ACCEPT_LANGUAGE 'en'" do
      get '/', {}, {"HTTP_ACCEPT_LANGUAGE" => "en"}
      last_request.env["locale.language"].should == "en"
    end

    it "should parse HTTP_ACCEPT_LANGUAGE 'sv'" do
      get '/', {}, {"HTTP_ACCEPT_LANGUAGE" => "sv"}
      last_request.env["locale.language"].should == "sv"
    end

    it "should parse HTTP_ACCEPT_LANGUAGE 'sv;q=0.1, en'" do
      get '/', {}, {"HTTP_ACCEPT_LANGUAGE" => "sv;q=0.1, en"}
      last_request.env["locale.language"].should == "en"
    end

    it "should parse HTTP_ACCEPT_LANGUAGE 'sv, en'" do
      get '/', {}, {"HTTP_ACCEPT_LANGUAGE" => "sv, en"}
      last_request.env["locale.language"].should == "sv"
    end

    it "should parse HTTP_ACCEPT_LANGUAGE 'en;q=0.4, de;q=0.7'" do
      get '/', {}, {"HTTP_ACCEPT_LANGUAGE" => "en;q=0.4, de;q=0.7"}
      last_request.env["locale.language"].should == "de"
    end

    it "should parse HTTP_ACCEPT_LANGUAGE 'en-US;q=0.7'" do
      get '/', {}, {"HTTP_ACCEPT_LANGUAGE" => "en-US;q=0.7"}
      last_request.env["locale.language"].should == "en"
    end
  end

  describe "missing database" do
    it "should fallback on HTTP_ACCEPT_LANGUAGE for country" do
      get '/', {}, {"HTTP_ACCEPT_LANGUAGE" => "en-US", "REMOTE_ADDR" => "10.0.0.1"}
      last_request.env["locale.language"].should == "en"
      last_request.env["locale.country"].should == "US"
    end
  end
end