Add some more tests

1 files changed, 17 insertions, 0 deletions

diff --git a/test/connection/base_test.rb b/test/connection/base_test.rb
new file mode 100644
index 0000000000..66e303a804
--- /dev/null
+++ b/test/connection/base_test.rb
@@ -0,0 +1,17 @@
+require 'test_helper'
+require 'stubs/test_server'
+
+class ActionCable::Connection::BaseTest < ActiveSupport::TestCase
+  setup do
+    @server = TestServer.new
+
+    env = Rack::MockRequest.env_for "/test", 'HTTP_CONNECTION' => 'upgrade', 'HTTP_UPGRADE' => 'websocket'
+    @connection = ActionCable::Connection::Base.new(@server, env)
+  end
+
+  test "making a connection with invalid headers" do
+    connection = ActionCable::Connection::Base.new(@server, Rack::MockRequest.env_for("/test"))
+    response = connection.process
+    assert_equal 404, response[0]
+  end
+end