fix(im): gate optional resource downloads at runtime

This commit is contained in:
wxianfeng committed 2026-07-29 15:33:01 +08:00
1 parent 75bb01bb64
commit 4567dd1cd6
18 files changed
+521 -354

No files matched your search

+34
View File
@@ -40,6 +40,40 @@ func TestCrossPlatformCoverageRiskDefaultsToRead(t *testing.T) {
}
}
func TestCrossPlatformCoverageEffectiveRiskUsesParsedFlagsAndFailsClosed(t *testing.T) {
cmd := &cobra.Command{}
cmd.Flags().Bool("download-resources", false, "")
rt := &RuntimeContext{cmd: cmd}
s := Shortcut{
Risk: RiskRead,
RiskWhen: func(rt *RuntimeContext) Risk {
if rt.Bool("download-resources") {
return RiskWrite
}
return RiskRead
},
}
if got := s.effectiveRisk(rt); got != RiskRead {
t.Fatalf("default effectiveRisk() = %q, want read", got)
}
if err := cmd.Flags().Set("download-resources", "true"); err != nil {
t.Fatal(err)
}
if got := s.effectiveRisk(rt); got != RiskWrite {
t.Fatalf("download effectiveRisk() = %q, want write", got)
}
s.Risk = RiskHighWrite
s.RiskWhen = func(*RuntimeContext) Risk { return Risk("unknown") }
if got := s.effectiveRisk(rt); got != RiskHighWrite {
t.Fatalf("invalid callback effectiveRisk() = %q, want high-risk-write", got)
}
s.RiskWhen = func(*RuntimeContext) Risk { return RiskRead }
if got := s.effectiveRisk(rt); got != RiskHighWrite {
t.Fatalf("downgrade effectiveRisk() = %q, want high-risk-write", got)
}
}
func TestCrossPlatformCoverageMountRegistersFlagsAndUse(t *testing.T) {
s := Shortcut{
Service: "contact",