require File.join(File.dirname(__FILE__), 'test_base_geocoder')

class GeonamesGeocoderTest < BaseGeocoderTest #:nodoc: all
  def geonames_escape(value)
    Geokit::Inflector.url_escape(value.to_s.gsub(/,/, ' '))
  end

  def setup
    super
    @response = MockSuccess.new
  end

  def test_geonames_general_geocoding
    @response.expects(:body).returns(fixture('geonames/postal_code_search'))
    url = "http://ws.geonames.org/postalCodeSearch?placename=#{geonames_escape(@address)}&maxRows=10"
    Geokit::Geocoders::GeonamesGeocoder.expects(:call_geocoder_service).with(url).returns(@response)
    res=Geokit::Geocoders::GeonamesGeocoder.geocode(@address)
    assert_equal "California", res.state
    assert_equal "San Francisco", res.city 
    assert_equal "37.781334,-122.416728", res.ll
    assert res.is_us?
    assert_equal "San Francisco, California, 94102, US", res.full_address
    assert_equal "geonames", res.provider
  end

  def test_geonames_search
    @response.expects(:body).returns(fixture('geonames/full_text_search'))
    url = "http://ws.geonames.org/search?q=#{geonames_escape(@address)}&style=FULL&maxRows=10"
    Geokit::Geocoders::GeonamesGeocoder.expects(:call_geocoder_service).with(url).returns(@response)
    res=Geokit::Geocoders::GeonamesGeocoder.search(@address)
    assert_equal "San Francisco", res.name
    assert_equal "CA", res.state
    assert_equal "San Francisco", res.city 
    assert_equal "37.7749295,-122.4194155", res.ll
    assert res.is_us?
    assert_equal "San Francisco, CA, US", res.full_address
    assert_equal "America/Los_Angeles", res.timezone
    assert_equal "geonames", res.provider
  end

  def test_geonames_search_city_is_not_set_if_location_is_not_a_city
    @response.expects(:body).returns(fixture('geonames/fts_not_a_city'))
    Geokit::Geocoders::GeonamesGeocoder.expects(:call_geocoder_service).returns(@response)
    res=Geokit::Geocoders::GeonamesGeocoder.search(@address)
    assert_equal "London King's Cross Railway Station", res.name
    assert_nil res.city
  end

  def test_geonames_search_with_continent_restriction
    @response.expects(:body).returns(fixture('geonames/full_text_search'))
    url = "http://ws.geonames.org/search?q=#{geonames_escape(@address)}&continentCode=NA&style=FULL&maxRows=10"
    Geokit::Geocoders::GeonamesGeocoder.expects(:call_geocoder_service).with(url).returns(@response)
    Geokit::Geocoders::GeonamesGeocoder.search(@address, :continent => 'NA')
  end

  def test_geonames_search_with_feature_class_restriction
    @response.stubs(:body).returns(fixture('geonames/full_text_search'))
    url = "http://ws.geonames.org/search?q=#{geonames_escape(@address)}&featureClass=P&style=FULL&maxRows=10"
    Geokit::Geocoders::GeonamesGeocoder.expects(:call_geocoder_service).with(url).returns(@response)
    Geokit::Geocoders::GeonamesGeocoder.search(@address, :feature_class => 'P')
    url = "http://ws.geonames.org/search?q=#{geonames_escape(@address)}&featureClass=P&featureClass=S&style=FULL&maxRows=10"
    Geokit::Geocoders::GeonamesGeocoder.expects(:call_geocoder_service).with(url).returns(@response)
    Geokit::Geocoders::GeonamesGeocoder.search(@address, :feature_class => %w(P S))
  end

  def test_geonames_search_with_feature_code_restriction
    @response.stubs(:body).returns(fixture('geonames/full_text_search'))
    url = "http://ws.geonames.org/search?q=#{geonames_escape(@address)}&featureCode=PPLC&style=FULL&maxRows=10"
    Geokit::Geocoders::GeonamesGeocoder.expects(:call_geocoder_service).with(url).returns(@response)
    Geokit::Geocoders::GeonamesGeocoder.search(@address, :feature_code => 'PPLC')
    url = "http://ws.geonames.org/search?q=#{geonames_escape(@address)}&featureCode=PPLC&featureCode=PPLX&style=FULL&maxRows=10"
    Geokit::Geocoders::GeonamesGeocoder.expects(:call_geocoder_service).with(url).returns(@response)
    Geokit::Geocoders::GeonamesGeocoder.search(@address, :feature_code => %w(PPLC PPLX))
  end

  def test_http_error
    error = MockFailure.new
    Geokit::Geocoders::GeonamesGeocoder.stubs(:call_geocoder_service).returns(error)
    assert_raise GeoKit::Geocoders::GeocodeError do
      Geokit::Geocoders::GeonamesGeocoder.search(@address)
    end
  end

  def test_geocoding_failure
    @response.stubs(:body).returns(fixture('geonames/nothing_found'))
    Geokit::Geocoders::GeonamesGeocoder.stubs(:call_geocoder_service).returns(@response)
    result = Geokit::Geocoders::GeonamesGeocoder.search(@address)
    assert !result.success?
  end
end