Silverfish/proton-bridge
1
0
Fork 1
mirror of https://github.com/ProtonMail/proton-bridge.git synced 2025-12-24 02:46:44 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(GODT-2800): ChanneledSubcriber types

Browse Source
This commit is contained in:
Leander Beernaert
2023-07-21 14:53:20 +02:00
parent 82efa16d65
commit 11f6f84dd6
3 changed files with 32 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
internal/services/userevents/subscriber_test.go
View File

@ -39,13 +39,13 @@ func TestChanneledSubscriber_CtxTimeoutDoesNotBlockFutureEvents(t *testing.T) {
defer wg.Done()
// Send one event, that succeeds.
require.NoError(t, subscriber.handle(context.Background(), []int{30}))
require.NoError(t, subscriber.handle(context.Background(), 30))
// Add an impossible deadline that fails immediately to simulate on event taking too long to send.
ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(time.Microsecond))
defer cancel()
err := subscriber.handle(ctx, []int{20})
err := subscriber.handle(ctx, 20)
require.Error(t, err)
require.True(t, errors.Is(err, context.DeadlineExceeded))
}()
@ -53,8 +53,8 @@ func TestChanneledSubscriber_CtxTimeoutDoesNotBlockFutureEvents(t *testing.T) {
// Receive first event. Notify success.
event, ok := <-subscriber.OnEventCh()
require.True(t, ok)
event.Consume(func(event []int) error {
require.Equal(t, []int{30}, event)
event.Consume(func(event int) error {
require.Equal(t, 30, event)
return nil
})
wg.Wait()
@ -63,13 +63,13 @@ func TestChanneledSubscriber_CtxTimeoutDoesNotBlockFutureEvents(t *testing.T) {
wg.Add(1)
go func() {
defer wg.Done()
require.NoError(t, subscriber.handle(context.Background(), []int{40}))
require.NoError(t, subscriber.handle(context.Background(), 40))
}()
event, ok = <-subscriber.OnEventCh()
require.True(t, ok)
event.Consume(func(event []int) error {
require.Equal(t, []int{40}, event)
event.Consume(func(event int) error {
require.Equal(t, 40, event)
return nil
})
@ -88,7 +88,7 @@ func TestChanneledSubscriber_ErrorReported(t *testing.T) {
defer wg.Done()
// Send one event, that succeeds.
err := subscriber.handle(context.Background(), []int{30})
err := subscriber.handle(context.Background(), 30)
require.Error(t, err)
require.Equal(t, reportedErr, err)
}()
@ -96,8 +96,8 @@ func TestChanneledSubscriber_ErrorReported(t *testing.T) {
// Receive first event. Notify success.
event, ok := <-subscriber.OnEventCh()
require.True(t, ok)
event.Consume(func(event []int) error {
require.Equal(t, []int{30}, event)
event.Consume(func(event int) error {
require.Equal(t, 30, event)
return reportedErr
})