Fix after connect

PlugFox · Feb 24, 2024 · 9ab3519 · 9ab3519
1 parent debbe55
commit 9ab3519
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 1 deletion.
diff --git a/lib/src/client/ws_client.dart b/lib/src/client/ws_client.dart
@@ -126,7 +126,7 @@ final class WebSocketClient implements IWebSocketClient {
       }
       try {
         // Send first messages after connection is established:
-        await _options.afterConnect?.call(this);
+        await _options.afterConnect?.call(_client);
       } on Object {
         _client.disconnect(1006, 'AFTER_CONNECT_ERROR');
       }

diff --git a/test/client/client_test.dart b/test/client/client_test.dart
@@ -383,6 +383,32 @@ void main() {
     });
   });
 
+  group('First message after connect', () {
+    const url = 'wss://echo.plugfox.dev:443/connect';
+
+    late IWebSocketClient client;
+
+    setUpAll(() {
+      client = WebSocketClient(WebSocketOptions.common(
+        afterConnect: (c) => c.add('auth.abc.123'),
+      ));
+    });
+
+    tearDownAll(() {
+      client.close();
+    });
+
+    test('Can send first message from options', () async {
+      await client.connect(url);
+      var received =
+          await client.stream.first.timeout(const Duration(seconds: 5));
+      expect(received, equals('auth.abc.123'));
+      client.add('message');
+      received = await client.stream.first.timeout(const Duration(seconds: 5));
+      expect(received, equals('message'));
+    });
+  });
+
   group(
     'Binary data as a Blob',
     () {