shangguanxuan.sgx
99e5a3cceb
The macOS/Windows coverage gates only execute tests matching ^(TestAllShortcuts|TestCrossPlatformCoverage), so the 401 refresh-retry tests added for this change were invisible to them, leaving 12 changed statements uncovered (92.73% < 100%). Rename the 12 existing tests into the TestCrossPlatformCoverage prefix and add a fetchTicketAttempt edge test covering transport failures, retryable statuses, and missing endpoint/ticket payload fields.
48 lines
1.9 KiB
Go
48 lines
1.9 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/DingTalk-Real-AI/dingtalk-workspace-cli/internal/event/source"
|
|
)
|
|
|
|
// TestCrossPlatformCoverageNewEventSourceWiresForceRefreshRejectedToken asserts the portal ticket
|
|
// source receives a ForceRefreshToken callback that forwards the actual
|
|
// rejected token into the app-level compare-and-refresh chain.
|
|
func TestCrossPlatformCoverageNewEventSourceWiresForceRefreshRejectedToken(t *testing.T) {
|
|
oldNew, oldRefresh := eventNewDingtalkSource, eventForceRefreshRejected
|
|
t.Cleanup(func() { eventNewDingtalkSource, eventForceRefreshRejected = oldNew, oldRefresh })
|
|
|
|
var captured source.Config
|
|
eventNewDingtalkSource = func(cfg source.Config, _ ...source.SourceOption) (*source.DingtalkSource, error) {
|
|
captured = cfg
|
|
return &source.DingtalkSource{}, nil
|
|
}
|
|
var gotDir, gotRejected string
|
|
eventForceRefreshRejected = func(_ context.Context, configDir, rejectedToken string) (string, error) {
|
|
gotDir, gotRejected = configDir, rejectedToken
|
|
return "fresh", nil
|
|
}
|
|
if _, err := newEventSource(context.Background(), "config-dir", "client", "secret", eventStreamTicketOptions{Mode: "custom"}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if captured.PortalTicket == nil || captured.PortalTicket.ForceRefreshToken == nil {
|
|
t.Fatal("ForceRefreshToken not wired into portal ticket config")
|
|
}
|
|
tok, err := captured.PortalTicket.ForceRefreshToken(context.Background(), "rejected-token")
|
|
if err != nil || tok != "fresh" {
|
|
t.Fatalf("force refresh = %q, %v", tok, err)
|
|
}
|
|
if gotDir != "config-dir" || gotRejected != "rejected-token" {
|
|
t.Fatalf("wiring passed dir %q rejected %q", gotDir, gotRejected)
|
|
}
|
|
|
|
fail := errors.New("refresh failed")
|
|
eventForceRefreshRejected = func(context.Context, string, string) (string, error) { return "", fail }
|
|
if _, err := captured.PortalTicket.ForceRefreshToken(context.Background(), "x"); !errors.Is(err, fail) {
|
|
t.Fatalf("refresh error = %v", err)
|
|
}
|
|
}
|