fix(llc): handle channel mute expiration.
Signed-off-by: xsahil03x <[email protected]>
This commit is contained in:
@@ -1447,15 +1447,32 @@ class Channel {
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Timer to keep track of mute expiration. This is used to update the channel
|
||||||
|
// state when the mute expires.
|
||||||
|
Timer? _muteExpirationTimer;
|
||||||
|
|
||||||
/// Mutes the channel.
|
/// Mutes the channel.
|
||||||
Future<EmptyResponse> mute({Duration? expiration}) {
|
Future<EmptyResponse> mute({Duration? expiration}) {
|
||||||
_checkInitialized();
|
_checkInitialized();
|
||||||
|
|
||||||
|
// If there is a expiration set, we will set a timer to automatically unmute
|
||||||
|
// the channel when the mute expires.
|
||||||
|
if (expiration != null) {
|
||||||
|
_muteExpirationTimer?.cancel();
|
||||||
|
_muteExpirationTimer = Timer(expiration, unmute);
|
||||||
|
}
|
||||||
|
|
||||||
return _client.muteChannel(cid!, expiration: expiration);
|
return _client.muteChannel(cid!, expiration: expiration);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Unmute the channel.
|
/// Unmute the channel.
|
||||||
Future<EmptyResponse> unmute() {
|
Future<EmptyResponse> unmute() {
|
||||||
_checkInitialized();
|
_checkInitialized();
|
||||||
|
|
||||||
|
// Cancel the mute expiration timer if it is set.
|
||||||
|
_muteExpirationTimer?.cancel();
|
||||||
|
_muteExpirationTimer = null;
|
||||||
|
|
||||||
return _client.unmuteChannel(cid!);
|
return _client.unmuteChannel(cid!);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -2481,6 +2481,31 @@ void main() {
|
|||||||
)).called(1);
|
)).called(1);
|
||||||
});
|
});
|
||||||
|
|
||||||
|
test('`.mute with expiration`', () async {
|
||||||
|
const expiration = Duration(seconds: 3);
|
||||||
|
|
||||||
|
when(() => client.muteChannel(
|
||||||
|
channelCid,
|
||||||
|
expiration: expiration,
|
||||||
|
)).thenAnswer((_) async => EmptyResponse());
|
||||||
|
|
||||||
|
when(() => client.unmuteChannel(channelCid))
|
||||||
|
.thenAnswer((_) async => EmptyResponse());
|
||||||
|
|
||||||
|
final res = await channel.mute(expiration: expiration);
|
||||||
|
|
||||||
|
expect(res, isNotNull);
|
||||||
|
|
||||||
|
verify(() => client.muteChannel(
|
||||||
|
channelCid,
|
||||||
|
expiration: expiration,
|
||||||
|
)).called(1);
|
||||||
|
|
||||||
|
// wait for expiration
|
||||||
|
await Future.delayed(expiration);
|
||||||
|
verify(() => client.unmuteChannel(channelCid)).called(1);
|
||||||
|
});
|
||||||
|
|
||||||
test('`.unmute`', () async {
|
test('`.unmute`', () async {
|
||||||
when(
|
when(
|
||||||
() => client.unmuteChannel(channelCid),
|
() => client.unmuteChannel(channelCid),
|
||||||
|
|||||||
Reference in New Issue
Block a user