refactor(llc): simplify WebSocket test setup and improve test utilities

This commit refactors the WebSocket testing utilities to simplify test setup and enhance maintainability.

The `ws_test_helpers.dart` file has been removed, and its functionality is replaced by a new, more streamlined `whenListenWebSocket` function in `web_socket_mocks.dart`. This function encapsulates the mock setup for `WebSocketChannel`, including handling authentication responses.

Key changes:
- **Replaced `WsTestConnection` with `whenListenWebSocket`**: Simplified the setup for mocking WebSocket channels in tests.
- **Improved test token generation**: Replaced a hardcoded JWT with a `generateTestUserToken` function for creating test tokens dynamically.
- **Relocated `api_mocker_mixin.dart`**: Moved the file for better organization within the test utilities.
- **Removed unused mocks**: Cleaned up the `mocks.dart` file by removing `MockFeedsRepository`, `MockFeedsClient`, `MockWebSocketClient`, `MockWebSocketSink`, and `FakeFeedsClient`.
- **Updated test implementation**: All tests now use the new `whenListenWebSocket` helper and dynamic token generation, reducing boilerplate and improving clarity.

Author: xsahil03x

packages/stream_feeds/test/client/feeds_client_impl_test.dart

@@ -6,10 +6,13 @@ import '../test_utils.dart';
 
 void main() {
   test('Create a feeds client', () {
+    const user = User(id: 'userId');
+    final token = generateTestUserToken(user.id);
+
     final client = StreamFeedsClient(
       apiKey: 'apiKey',
-      user: const User(id: 'userId'),
-      tokenProvider: TokenProvider.static(UserToken(testToken)),
+      user: user,
+      tokenProvider: TokenProvider.static(token),
     );
 
     expect(client, isA<StreamFeedsClientImpl>());

packages/stream_feeds/test/test_utils.dart

@@ -11,4 +11,4 @@ export 'test_utils/testers/feed_tester.dart';
 export 'test_utils/testers/follow_list_tester.dart';
 export 'test_utils/testers/poll_list_tester.dart';
 export 'test_utils/testers/poll_vote_list_tester.dart';
-export 'test_utils/ws_test_helpers.dart';
+export 'test_utils/web_socket_mocks.dart';

…test_utils/testers/api_mocker_mixin.dart …ds/test/test_utils/api_mocker_mixin.dartpackages/stream_feeds/test/test_utils/testers/api_mocker_mixin.dart renamed to packages/stream_feeds/test/test_utils/api_mocker_mixin.dart packages/stream_feeds/test/test_utils/testers/api_mocker_mixin.dart renamed to packages/stream_feeds/test/test_utils/api_mocker_mixin.dart

@@ -1,7 +1,7 @@
 import 'package:mocktail/mocktail.dart';
 import 'package:stream_feeds/stream_feeds.dart';
 
-import '../mocks.dart';
+import 'mocks.dart';
 
 /// Mixin for test utilities that need to mock and verify API calls.
 ///

packages/stream_feeds/test/test_utils/mocks.dart

@@ -1,38 +1,24 @@
+import 'dart:convert';
+
 import 'package:mocktail/mocktail.dart';
-import 'package:stream_feeds/src/client/feeds_client_impl.dart';
-import 'package:stream_feeds/src/repository/feeds_repository.dart';
-import 'package:stream_feeds/stream_feeds.dart';
 import 'package:stream_feeds/stream_feeds.dart' as api;
 import 'package:web_socket_channel/web_socket_channel.dart';
 
-class MockFeedsRepository extends Mock implements FeedsRepository {}
-
-class MockFeedsClient extends Mock implements StreamFeedsClient {}
-
-class MockWebSocketClient extends Mock implements StreamWebSocketClient {}
-
 class MockDefaultApi extends Mock implements api.DefaultApi {}
 
 class MockWebSocketChannel extends Mock implements WebSocketChannel {}
 
-class MockWebSocketSink extends Mock implements WebSocketSink {}
+api.UserToken generateTestUserToken(String userId) {
+  String b64UrlNoPad(Object jsonObj) {
+    final bytes = utf8.encode(jsonEncode(jsonObj));
+    return base64Url.encode(bytes).replaceAll('=', '');
+  }
 
-class FakeFeedsClient extends Fake implements StreamFeedsClientImpl {
-  FakeFeedsClient({
-    User? user,
-  }) : user = user ?? fakeUser;
+  final header = {'alg': 'none', 'typ': 'JWT'};
+  final payload = {'user_id': userId};
 
-  @override
-  final User user;
+  final jwt = '${b64UrlNoPad(header)}.${b64UrlNoPad(payload)}.';
+  // trailing dot = empty signature (valid for alg=none)
 
-  @override
-  final EventEmitter events = MutableEventEmitter();
+  return api.UserToken(jwt);
 }
-
-const fakeUser = User(
-  id: 'user_id',
-  name: 'user_name',
-);
-
-const testToken =
-    'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX2lkIjoibHVrZV9za3l3YWxrZXIifQ.hZ59SWtp_zLKVV9ShkqkTsCGi_jdPHly7XNCf5T_Ev0';

packages/stream_feeds/test/test_utils/testers/base_tester.dart

@@ -5,9 +5,9 @@ import 'package:meta/meta.dart';
 import 'package:stream_feeds/stream_feeds.dart';
 import 'package:test/test.dart' as test;
 
+import '../api_mocker_mixin.dart';
 import '../mocks.dart';
-import '../ws_test_helpers.dart';
-import 'api_mocker_mixin.dart';
+import '../web_socket_mocks.dart';
 
 /// Factory function signature for creating tester instances.
 ///
@@ -81,22 +81,12 @@ Future<T> createTester<T extends BaseTester<Object?>>({
   required MockWebSocketChannel webSocketChannel,
   required T Function(StreamController<Object>) create,
 }) async {
-  // Create WebSocket components
+  // Create WebSocket stream controller
   final wsStreamController = StreamController<Object>();
-  final webSocketSink = MockWebSocketSink();
+  test.addTearDown(wsStreamController.close); // Close controller after test
 
-  // Register automatic cleanup
-  test.addTearDown(() async {
-    await webSocketSink.close();
-    await wsStreamController.close();
-  });
-
-  // Setup WebSocket connection
-  WsTestConnection(
-    wsStreamController: wsStreamController,
-    webSocketSink: webSocketSink,
-    webSocketChannel: webSocketChannel,
-  ).setUp();
+  // Set up WebSocket channel mocks
+  whenListenWebSocket(webSocketChannel, wsStreamController);
 
   // Connect client
   await client.connect();
@@ -141,7 +131,7 @@ void testWithTester<S, T extends BaseTester<S>>(
     () async {
       await _runZonedGuarded(() async {
         const user = User(id: 'luke_skywalker');
-        final userToken = UserToken(testToken);
+        final userToken = generateTestUserToken(user.id);
 
         final feedsApi = MockDefaultApi();
         final webSocketChannel = MockWebSocketChannel();

packages/stream_feeds/test/test_utils/web_socket_mocks.dart

@@ -0,0 +1,65 @@
+import 'dart:async';
+import 'dart:convert';
+
+import 'package:mocktail/mocktail.dart';
+import 'package:stream_feeds/src/ws/events/events.dart';
+import 'package:web_socket_channel/web_socket_channel.dart';
+
+class _MockWebSocketSink extends Mock implements WebSocketSink {}
+
+/// Sets up WebSocket channel mocks for testing.
+///
+/// Configures [webSocketChannel] to use [wsStreamController] for its stream
+/// when the channel's stream is listened to. Also creates a mock sink for
+/// sending messages and handles authentication by automatically responding
+/// with a health check event when a token is sent.
+///
+/// Example:
+/// ```dart
+/// final wsStreamController = StreamController<Object>();
+/// whenListenWebSocket(webSocketChannel, wsStreamController);
+/// ```
+void whenListenWebSocket(
+  WebSocketChannel webSocketChannel,
+  StreamController<Object> wsStreamController,
+) {
+  final webSocketSink = _MockWebSocketSink();
+
+  when(
+    webSocketSink.close,
+  ).thenAnswer((_) => Future<void>.value());
+
+  when(
+    () => webSocketChannel.ready,
+  ).thenAnswer((_) => Future<void>.value());
+
+  when(
+    () => webSocketChannel.stream,
+  ).thenAnswer((_) => wsStreamController.stream);
+
+  when(
+    () => webSocketChannel.sink,
+  ).thenAnswer((_) => webSocketSink);
+
+  // Handle authentication: when a token is sent, respond with health check
+  when(
+    () => webSocketSink.add(any<Object>()),
+  ).thenAnswer((invocation) {
+    final event = jsonDecode(
+      invocation.positionalArguments.first as String,
+    ) as Map<String, dynamic>;
+
+    if (event['token'] != null) {
+      return wsStreamController.add(
+        jsonEncode(
+          HealthCheckEvent(
+            connectionId: 'connectionId',
+            createdAt: DateTime.now(),
+            custom: const {},
+            type: 'health.check',
+          ),
+        ),
+      );
+    }
+  });
+}