Initial commit

script/geocoder_test

@@ -0,0 +1,52 @@
+#!/usr/bin/env ruby
+require 'yaml'
+require 'tmpdir'
+require_relative '../services/geocoder/lib/hires_geocoder_factory'
+
+TIMEOUT = 300 # seconds before raising a timeout error
+
+def usage
+  abort "usage: script/geocoder_test environment <non-batch>"
+end
+
+environment = ARGV[0]
+non_batch = (ARGV[1] == 'non-batch')
+usage unless environment
+
+def load_config(environment)
+  config_file_hash = YAML.load_file(File.expand_path('../../config/app_config.yml',  __FILE__))
+  config_file_hash[environment]["geocoder"]
+rescue => e
+  raise "Missing or inaccessible config for environment #{environment} in config/app_config.yml: #{e.message}"
+end
+
+config = load_config(environment)
+CartoDB::GeocoderConfig.instance.set config.merge('force_batch' => !non_batch)
+input_file = File.expand_path('../../services/geocoder/spec/fixtures/nokia_input.csv',  __FILE__)
+working_dir = Dir.mktmpdir
+geocoder = CartoDB::HiresGeocoderFactory.get(input_file, working_dir)
+
+puts "Runing #{(non_batch ? 'non batch' : 'batch')} geocoder..."
+start = Time.now
+geocoder.run
+finish = Time.now
+until geocoder.status == 'completed' do
+  finish = Time.now
+  raise "Geocoder FAILURE: Timeout" if (finish - start) > TIMEOUT
+  geocoder.update_status
+  sleep(2)
+end
+raise "Geocoder FAILURE" unless geocoder.status == 'completed'
+
+Dir.chdir geocoder.dir
+`unp #{geocoder.result} 2>&1`
+file_path = (non_batch ? geocoder.result : Dir[File.join(geocoder.dir, '*_out.txt')][0])
+response = File.read(file_path).split(',')
+
+expected_position = [45.96, -66.60]
+position = [response[4].to_f, response[5].to_f]
+distance = Math.sqrt(expected_position.zip(position).map { |x| (x[1] - x[0])**2 }.reduce(:+))
+raise "Geocoder FAILURE: wrong geocoded position #{position}, expected #{expected_position}, distance: #{distance}" if distance > 0.04
+
+FileUtils.rm_f working_dir
+puts "\e[32mGeocoder OK #{(finish - start)} secs\e[0m"