fixed tests

fizx · Jun 23, 2010 · 9482eff · 9482eff
1 parent 8ab1b10
commit 9482eff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_robots.rb b/test/test_robots.rb
@@ -14,7 +14,7 @@ def status
 
 class TestRobots < Test::Unit::TestCase
   def setup
-    def Robots.get_robots_txt(uri)
+    def Robots.get_robots_txt(uri, user_agent)
       fixture_file = File.dirname(__FILE__) + "/fixtures/" + uri.host.split(".")[-2] + ".txt"
       File.open(fixture_file).extend(FakeHttp)
     end
@@ -23,7 +23,7 @@ def Robots.get_robots_txt(uri)
   end
 
   def test_allowed_if_no_robots
-    def Robots.get_robots_txt(uri)
+    def Robots.get_robots_txt(uri, user_agent)
       return nil
     end