chore: import upstream snapshot with attribution
CI / license-header (push) Has been skipped
CI / e2e-dry-run (push) Has been skipped
CI / fast-gate (push) Failing after 0s
Test PR Label Logic / test-pr-labels (push) Failing after 1s
Skill Format Check / check-format (push) Failing after 2s
CI / security (push) Failing after 5s
CI / unit-test (push) Has been skipped
CI / lint (push) Has been skipped
CI / script-test (push) Has been skipped
CI / deterministic-gate (push) Has been skipped
CI / coverage (push) Has been skipped
CI / results (push) Has been cancelled
CI / deadcode (push) Has been cancelled
CI / e2e-live (push) Has been cancelled
CI / license-header (push) Has been skipped
CI / e2e-dry-run (push) Has been skipped
CI / fast-gate (push) Failing after 0s
Test PR Label Logic / test-pr-labels (push) Failing after 1s
Skill Format Check / check-format (push) Failing after 2s
CI / security (push) Failing after 5s
CI / unit-test (push) Has been skipped
CI / lint (push) Has been skipped
CI / script-test (push) Has been skipped
CI / deterministic-gate (push) Has been skipped
CI / coverage (push) Has been skipped
CI / results (push) Has been cancelled
CI / deadcode (push) Has been cancelled
CI / e2e-live (push) Has been cancelled
This commit is contained in:
@@ -0,0 +1,61 @@
|
||||
// Copyright (c) 2026 Lark Technologies Pte. Ltd.
|
||||
// SPDX-License-Identifier: MIT
|
||||
|
||||
package source
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/larksuite/cli/internal/event"
|
||||
)
|
||||
|
||||
type mockSource struct {
|
||||
name string
|
||||
events []*event.RawEvent
|
||||
}
|
||||
|
||||
func (s *mockSource) Name() string { return s.name }
|
||||
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 TestRegister(t *testing.T) {
|
||||
ResetForTest()
|
||||
|
||||
src := &mockSource{name: "test-source"}
|
||||
Register(src)
|
||||
|
||||
sources := All()
|
||||
if len(sources) != 1 || sources[0].Name() != "test-source" {
|
||||
t.Errorf("unexpected sources: %v", sources)
|
||||
}
|
||||
}
|
||||
|
||||
func TestMockSource_EmitsEvents(t *testing.T) {
|
||||
src := &mockSource{
|
||||
name: "test",
|
||||
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))
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user