Add some tests.

fixtures/test_channel.rb

@@ -0,0 +1,4 @@
+require "action_cable"
+
+class TestChannel < ActionCable::Channel::Base
+end

gems.rb

@@ -3,7 +3,7 @@
 
 source "https://rubygems.org"
 
-gem "rails"
+gemspec
 
 group :maintenance, optional: true do
 	gem "bake-gem"
@@ -15,4 +15,9 @@
 	gem "covered"
 	gem "decode"
 	gem "rubocop"
+
+	gem "sus-fixtures-async-http"
+	gem "sus-fixtures-console"
+
+	gem "async-websocket"
 end

lib/async/cable/middleware.rb

@@ -1,9 +1,10 @@
 # Released under the MIT License.
 # Copyright, 2024, by Samuel Williams.
 
-require 'async/websocket/adapters/rack'
+require "async/websocket/adapters/rack"
+require "action_cable"
 
-require_relative 'socket'
+require_relative "socket"
 
 module Async
 	module Cable
@@ -15,6 +16,8 @@ def initialize(app, server: ActionCable.server)
 				@protocols = ::ActionCable::INTERNAL[:protocols]
 			end
 
+			attr :server
+
 			def call(env)
 				if Async::WebSocket::Adapters::Rack.websocket?(env) and allow_request_origin?(env)
 					Async::WebSocket::Adapters::Rack.open(env, protocols: @protocols) do |websocket|
@@ -37,7 +40,7 @@ def handle_incoming_websocket(env, websocket)
 				@server.setup_heartbeat_timer
 
 				while message = websocket.read
-					Console.debug(self, "Received cable data:", message)
+					Console.debug(self, "Received cable data:", message.buffer)
 					connection.handle_incoming(@coder.decode(message.buffer))
 				end
 			rescue Protocol::WebSocket::ClosedError, EOFError

lib/async/cable/socket.rb

@@ -3,11 +3,6 @@
 
 module Async::Cable
 	class Socket
-		#== Action Cable socket interface ==
-		attr_reader :env, :logger, :protocol
-
-		delegate :worker_pool, :logger, to: :@server
-
 		def initialize(env, websocket, server, coder: ActiveSupport::JSON)
 			@env = env
 			@websocket = websocket
@@ -17,8 +12,13 @@ def initialize(env, websocket, server, coder: ActiveSupport::JSON)
 			@write_guard = Mutex.new
 		end
 
+		attr :env
 		attr :output
 
+		def logger
+			Console
+		end
+
 		def request
 			# Copied from ActionCable::Server::Socket#request
 			@request ||= begin

test/async/cable/middleware.rb

@@ -0,0 +1,60 @@
+require "async/cable/middleware"
+
+require "protocol/rack/adapter"
+require "async/websocket/client"
+require "sus/fixtures/async/http/server_context"
+
+require "test_channel"
+
+describe Async::Cable::Middleware do
+	include Sus::Fixtures::Async::HTTP::ServerContext
+
+	let(:cable_server) {::ActionCable::Server::Base.new}
+
+	before do
+		cable_server.config.disable_request_forgery_protection = true
+	end
+
+	after do
+		@cable_server&.restart
+	end
+
+	let(:app) do
+		Protocol::Rack::Adapter.new(subject.new(nil, server: cable_server))
+	end
+
+	let(:connection) {Async::WebSocket::Client.connect(client_endpoint)}
+
+	it "can connect and receive welcome messages" do
+		welcome_message = connection.read.parse
+
+		expect(welcome_message).to have_keys(
+			type: be == "welcome"
+		)
+	ensure
+		connection.close
+	end
+
+	it "can connect and send broadcast messages" do
+		subscribe_message = ::Protocol::WebSocket::TextMessage.generate({
+			command: "subscribe",
+			identifier: JSON.dump({"channel" => "TestChannel"}),
+		})
+
+		subscribe_message.send(connection)
+
+		while message = connection.read
+			confirmation = message.parse
+
+			if confirmation[:type] == "confirm_subscription"
+				break
+			end
+		end
+
+		expect(confirmation).to have_keys(
+			identifier: be == JSON.dump({"channel" => "TestChannel"})
+		)
+	ensure
+		connection.close
+	end
+end