lib/geocoder/lookups/test.rb in geocoder-1.1.3 vs lib/geocoder/lookups/test.rb in geocoder-1.1.4

@@ -3,16 +3,18 @@
 
 module Geocoder
   module Lookup
     class Test < Base
 
-      def self.add_stub(query, results)
-        stubs[query] = results
+      def self.add_stub(query_text, results)
+        stubs[query_text] = results
       end
 
-      def self.read_stub(query)
-        stubs.fetch(query) { raise ArgumentError, "unknown stub request #{query}" }
+      def self.read_stub(query_text)
+        stubs.fetch(query_text) {
+          raise ArgumentError, "unknown stub request #{query_text}"
+        }
       end
 
       def self.stubs
         @stubs ||= {}
       end
@@ -21,11 +23,11 @@
         @stubs = {}
       end
 
       private
 
-      def results(query, reverse = false)
-        Geocoder::Lookup::Test.read_stub(query)
+      def results(query)
+        Geocoder::Lookup::Test.read_stub(query.text)
       end
 
     end
   end
 end