Skip to content

Commit 57ba57b

Browse files
committed
test: update ModerationClientTester to include tearDown and fix client type
1 parent cbe23ee commit 57ba57b

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

packages/stream_feeds_test/lib/src/testers/moderation_client_tester.dart

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -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.
9191
Future<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

Comments
 (0)