mirror of
https://github.com/larksuite/cli.git
synced 2026-09-14 18:42:53 +08:00
leave330
47 lines
982 B
Go
47 lines
982 B
Go
// Copyright (c) 2026 Lark Technologies Pte. Ltd.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package websocket
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/larksuite/cli/internal/event"
|
|
)
|
|
|
|
type mockSource struct {
|
|
events []*event.RawEvent
|
|
}
|
|
|
|
func (s *mockSource) Start(ctx context.Context, _ []string, emit func(*event.RawEvent), _ StatusNotifier) error {
|
|
for _, e := range s.events {
|
|
emit(e)
|
|
}
|
|
<-ctx.Done()
|
|
return nil
|
|
}
|
|
|
|
func TestMockSource_EmitsEvents(t *testing.T) {
|
|
src := &mockSource{
|
|
events: []*event.RawEvent{
|
|
{EventID: "1", EventType: "im.message.receive_v1"},
|
|
{EventID: "2", EventType: "im.message.receive_v1"},
|
|
},
|
|
}
|
|
|
|
received := make(chan *event.RawEvent, 10)
|
|
ctx, cancel := context.WithTimeout(context.Background(), 100*time.Millisecond)
|
|
defer cancel()
|
|
|
|
go src.Start(ctx, nil, func(e *event.RawEvent) {
|
|
received <- e
|
|
}, nil)
|
|
|
|
time.Sleep(50 * time.Millisecond)
|
|
if len(received) != 2 {
|
|
t.Errorf("expected 2 events, got %d", len(received))
|
|
}
|
|
}
|