inspect ws close code

AlexStocks · Nov 22, 2023 · fdaf534 · fdaf534
1 parent 1b5ea31
commit fdaf534
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/transport/client_test.go b/transport/client_test.go
@@ -379,8 +379,7 @@ func TestNewWSClient(t *testing.T) {
 	serverSession.Close()
 	select {
 	case code := <-done:
-		// refer to websocket.isValidReceivedCloseCode
-		assert.True(t, (code >= 1000 && code <= 1003) || (code >= 1007 && code <= 1013) || (code >= 3000 && code <= 4999))
+		assert.True(t, code == websocket.CloseNormalClosure)
 	case <-time.After(5e9):
 		assert.True(t, false)
 	}