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,113 @@
|
||||
// Copyright (c) 2026 Lark Technologies Pte. Ltd.
|
||||
// SPDX-License-Identifier: MIT
|
||||
|
||||
package source
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/larksuite/cli/internal/event/protocol"
|
||||
)
|
||||
|
||||
// "disconnected to <url>" contains "connected to ws" — must use HasPrefix to avoid misclassifying as connect.
|
||||
func TestTryNotify_Classify(t *testing.T) {
|
||||
cases := []struct {
|
||||
name string
|
||||
msg string
|
||||
errDetail string
|
||||
wantState string
|
||||
wantDetail string
|
||||
wantCalled bool
|
||||
}{
|
||||
{
|
||||
name: "connected (SDK connect success)",
|
||||
msg: "connected to wss://example.com/gw [conn_id=abc]",
|
||||
wantState: protocol.SourceStateConnected,
|
||||
wantCalled: true,
|
||||
},
|
||||
{
|
||||
name: "disconnected must not be misclassified as connected",
|
||||
msg: "disconnected to wss://example.com/gw [conn_id=abc]",
|
||||
wantState: protocol.SourceStateDisconnected,
|
||||
wantCalled: true,
|
||||
},
|
||||
{
|
||||
name: "disconnected carries errDetail through",
|
||||
msg: "disconnected to wss://example.com/gw [conn_id=abc]",
|
||||
errDetail: "read tcp: broken pipe",
|
||||
wantState: protocol.SourceStateDisconnected,
|
||||
wantDetail: "read tcp: broken pipe",
|
||||
wantCalled: true,
|
||||
},
|
||||
{
|
||||
name: "reconnecting with attempt 1",
|
||||
msg: "trying to reconnect: 1 [conn_id=abc]",
|
||||
wantState: protocol.SourceStateReconnecting,
|
||||
wantDetail: "attempt 1",
|
||||
wantCalled: true,
|
||||
},
|
||||
{
|
||||
name: "reconnecting with attempt 12",
|
||||
msg: "trying to reconnect: 12",
|
||||
wantState: protocol.SourceStateReconnecting,
|
||||
wantDetail: "attempt 12",
|
||||
wantCalled: true,
|
||||
},
|
||||
{
|
||||
name: "case-insensitive connected",
|
||||
msg: "CONNECTED TO WSS://example.com",
|
||||
wantState: protocol.SourceStateConnected,
|
||||
wantCalled: true,
|
||||
},
|
||||
{
|
||||
name: "ignore generic connect-failed error",
|
||||
msg: "connect failed, err: dial tcp: i/o timeout",
|
||||
errDetail: "connect failed, err: dial tcp: i/o timeout",
|
||||
wantCalled: false,
|
||||
},
|
||||
{
|
||||
name: "ignore read-loop failure",
|
||||
msg: "receive message failed, err: websocket: close 1006",
|
||||
errDetail: "receive message failed, err: websocket: close 1006",
|
||||
wantCalled: false,
|
||||
},
|
||||
{
|
||||
name: "ignore heartbeat noise",
|
||||
msg: "receive pong",
|
||||
wantCalled: false,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tc := range cases {
|
||||
t.Run(tc.name, func(t *testing.T) {
|
||||
var gotState, gotDetail string
|
||||
called := false
|
||||
lg := &sdkLogger{notify: func(state, detail string) {
|
||||
called = true
|
||||
gotState = state
|
||||
gotDetail = detail
|
||||
}}
|
||||
lg.tryNotify(tc.msg, tc.errDetail)
|
||||
|
||||
if called != tc.wantCalled {
|
||||
t.Fatalf("called=%v, want %v (msg=%q)", called, tc.wantCalled, tc.msg)
|
||||
}
|
||||
if !called {
|
||||
return
|
||||
}
|
||||
if gotState != tc.wantState {
|
||||
t.Errorf("state = %q, want %q", gotState, tc.wantState)
|
||||
}
|
||||
if gotDetail != tc.wantDetail {
|
||||
t.Errorf("detail = %q, want %q", gotDetail, tc.wantDetail)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestTryNotify_NilNotifySafe(t *testing.T) {
|
||||
lg := &sdkLogger{notify: nil}
|
||||
lg.tryNotify("disconnected to wss://example.com", "")
|
||||
lg.tryNotify("connected to wss://example.com", "")
|
||||
lg.tryNotify("trying to reconnect: 1", "")
|
||||
}
|
||||
Reference in New Issue
Block a user