Add more tests for close behaviour. (#51)

ioquatix · ioquatix · commit 978e839e9d24 · 2023-03-08T16:16:22.000+13:00
* Expect the connection to be closed after sending and then receiving a close frame.
diff --git a/gems.rb b/gems.rb
@@ -19,4 +19,4 @@
 	gem "bake-test-external"
 end
 
-gem "rack", "~> 3.0.0"
+# gem "protocol-websocket", path: "../protocol-websocket"
diff --git a/test/async/websocket/client.rb b/test/async/websocket/client.rb
@@ -20,6 +20,27 @@
 		end
 	end
 	
+	with '#send_close' do
+		it "can read incoming messages and then close" do
+			connection = Async::WebSocket::Client.connect(client_endpoint)
+			3.times do
+				connection.send_text("Hello World!")
+			end
+			
+			# This informs the server we are done echoing messages:
+			connection.send_close
+			
+			# Collect all the echoed messages:
+			messages = []
+			while message = connection.read
+				messages << message
+			end
+			
+			expect(messages.size).to be == 3
+			expect(connection).to be(:closed?)
+		end
+	end
+	
 	with '#close' do
 		it "can connect to a websocket server and close underlying client" do
 			Async do |task|
@@ -46,6 +67,26 @@
 		end
 	end
 	
+	with "#close(1001)" do
+		let(:app) do
+			Protocol::HTTP::Middleware.for do |request|
+				Async::WebSocket::Adapters::HTTP.open(request) do |connection|
+					connection.send_text("Hello World!")
+					connection.close(1001)
+				end
+			end
+		end
+
+		it 'closes with custom error' do
+			connection = Async::WebSocket::Client.connect(client_endpoint)
+			message = connection.read
+			
+			expect do
+				connection.read
+			end.to raise_exception(Protocol::WebSocket::Error).and(have_attributes(code: be == 1001))
+		end
+	end
+	
 	with 'missing support for websockets' do
 		let(:app) do
 			Protocol::HTTP::Middleware.for do |request|
