refactor: always wait for hello message in withSocket test helper

flowr-analysis · Aug 30, 2023 · 4179305 · 4179305
1 parent cff0b88
commit 4179305
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/test/flowr/server.spec.ts b/test/flowr/server.spec.ts
@@ -6,7 +6,6 @@ import { assert } from 'chai'
 
 describe('FlowR Server', withShell(shell => {
 	it('Correct Hello Message', withSocket(shell,async socket => {
-		await socket.waitForMessage('hello')
 		const messages = socket.getMessages()
 		assert.strictEqual(messages.length, 1, 'Expected exactly one message to hello the client')
 

diff --git a/test/helper/net.ts b/test/helper/net.ts
@@ -90,8 +90,9 @@ export function withSocket(shell: RShell, fn: (socket: FakeSocket, server: FakeS
 		const net = new FakeServer()
 		const server = new FlowRServer(shell, await defaultTokenMap(), net)
 		await server.start(42)
-		const exampleSocket = new FakeSocket()
-		net.connectClient(exampleSocket)
-		await fn(exampleSocket, net)
+		const socket = new FakeSocket()
+		net.connectClient(socket)
+		await socket.waitForMessage('hello')
+		await fn(socket, net)
 	}
 }