Set appropriate origin and host in the tests

shunyama · Oct 12, 2015 · d621ae4 · d621ae4
1 parent 8275cfb
commit d621ae4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/connection/base_test.rb b/test/connection/base_test.rb
@@ -16,9 +16,10 @@ def disconnect
 
   setup do
     @server = TestServer.new
-    @server.config.disable_request_forgery_protection = true
 
-    env = Rack::MockRequest.env_for "/test", 'HTTP_CONNECTION' => 'upgrade', 'HTTP_UPGRADE' => 'websocket'
+    env = Rack::MockRequest.env_for "/test", 'HTTP_CONNECTION' => 'upgrade', 'HTTP_UPGRADE' => 'websocket',
+      'SERVER_NAME' => 'rubyonrails.com', 'HTTP_ORIGIN' => 'http://rubyonrails.com'
+
     @connection = Connection.new(@server, env)
     @response = @connection.process
   end