@@ -16,6 +16,7 @@ import 'base_tester.dart';
1616/// [setUp] is optional and runs before [body] for setting up mocks and test state.
1717/// [body] is the test callback that receives a [ModerationClientTester] for interactions.
1818/// [verify] is optional and runs after [body] for verifying API calls and interactions.
19+ /// [tearDown] is optional and runs after [verify] for custom cleanup.
1920/// [skip] is optional, skip this test.
2021/// [tags] is optional, tags for test filtering. Defaults to ['moderation-client'] .
2122/// [timeout] is optional, custom timeout for this test.
@@ -47,18 +48,20 @@ void moderationClientTest(
4748 FutureOr <void > Function (ModerationClientTester tester)? setUp,
4849 required FutureOr <void > Function (ModerationClientTester tester) body,
4950 FutureOr <void > Function (ModerationClientTester tester)? verify,
51+ FutureOr <void > Function (ModerationClientTester tester)? tearDown,
5052 bool skip = false ,
5153 Iterable <String > tags = const ['moderation-client' ],
5254 test.Timeout ? timeout,
5355}) {
54- return testWithTester <StreamFeedsClient , ModerationClientTester >(
56+ return testWithTester <ModerationClient , ModerationClientTester >(
5557 description,
5658 user: user,
57- build: (client) => client,
59+ build: (client) => client.moderation ,
5860 createTesterFn: _createModerationClientTester,
5961 setUp: setUp,
6062 body: body,
6163 verify: verify,
64+ tearDown: tearDown,
6265 skip: skip,
6366 tags: tags,
6467 timeout: timeout,
@@ -70,26 +73,23 @@ void moderationClientTest(
7073/// Provides helper methods for mocking and verifying moderation API calls.
7174///
7275/// Resources are automatically cleaned up after the test completes.
73- final class ModerationClientTester extends BaseTester <StreamFeedsClient > {
76+ final class ModerationClientTester extends BaseTester <ModerationClient > {
7477 const ModerationClientTester ._({
7578 required super .subject,
7679 required super .wsStreamController,
7780 required super .feedsApi,
7881 });
7982
80- /// The Stream Feeds client being tested.
81- StreamFeedsClient get client => subject;
82-
83- /// The moderation client for performing moderation operations.
84- ModerationClient get moderation => subject.moderation;
83+ /// The Moderation client being tested.
84+ ModerationClient get moderation => subject;
8585}
8686
8787// Creates a ModerationClientTester for testing moderation client operations.
8888//
8989// Automatically sets up WebSocket connection and registers cleanup handlers.
9090// This function is for internal use by moderationClientTest only.
9191Future <ModerationClientTester > _createModerationClientTester ({
92- required StreamFeedsClient subject,
92+ required ModerationClient subject,
9393 required StreamFeedsClient client,
9494 required MockDefaultApi feedsApi,
9595 required MockWebSocketChannel webSocketChannel,
0 commit comments