101 lines
2.9 KiB
Go
101 lines
2.9 KiB
Go
package unit_test
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/DingTalk-Real-AI/dingtalk-workspace-cli/internal/cache"
|
|
"github.com/DingTalk-Real-AI/dingtalk-workspace-cli/internal/discovery"
|
|
"github.com/DingTalk-Real-AI/dingtalk-workspace-cli/internal/market"
|
|
"github.com/DingTalk-Real-AI/dingtalk-workspace-cli/internal/transport"
|
|
)
|
|
|
|
func TestDiscoverDetailFallsBackToCachedSnapshot(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
store := cache.NewStore(t.TempDir())
|
|
store.Now = func() time.Time { return time.Date(2026, 3, 21, 0, 0, 0, 0, time.UTC) }
|
|
payload := market.DetailResponse{
|
|
Success: true,
|
|
Result: market.DetailResult{
|
|
MCPID: 9629,
|
|
Name: "钉钉文档",
|
|
Tools: []market.DetailTool{{ToolName: "create_document", ToolTitle: "创建文档"}},
|
|
},
|
|
}
|
|
raw, err := json.Marshal(payload)
|
|
if err != nil {
|
|
t.Fatalf("Marshal() error = %v", err)
|
|
}
|
|
if err := store.SaveDetail("default/default", "doc", cache.DetailSnapshot{
|
|
MCPID: 9629,
|
|
Payload: raw,
|
|
}); err != nil {
|
|
t.Fatalf("SaveDetail() error = %v", err)
|
|
}
|
|
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
http.Error(w, "boom", http.StatusInternalServerError)
|
|
}))
|
|
defer server.Close()
|
|
|
|
service := discovery.NewService(market.NewClient(server.URL, server.Client()), transport.NewClient(server.Client()), store)
|
|
detail, err := service.DiscoverDetail(context.Background(), market.ServerDescriptor{
|
|
Key: "doc",
|
|
DetailLocator: market.DetailLocator{
|
|
MCPID: 9629,
|
|
},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("DiscoverDetail() error = %v", err)
|
|
}
|
|
if !detail.Success || detail.Result.MCPID != 9629 {
|
|
t.Fatalf("DiscoverDetail() detail = %#v", detail)
|
|
}
|
|
}
|
|
|
|
func TestDiscoverDetailRejectsCachedSnapshotForDifferentMCPID(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
store := cache.NewStore(t.TempDir())
|
|
store.Now = func() time.Time { return time.Date(2026, 3, 21, 0, 0, 0, 0, time.UTC) }
|
|
payload := market.DetailResponse{
|
|
Success: true,
|
|
Result: market.DetailResult{
|
|
MCPID: 1047,
|
|
Name: "钉钉文档（旧）",
|
|
Tools: []market.DetailTool{{ToolName: "search_documents", ToolTitle: "搜索文档"}},
|
|
},
|
|
}
|
|
raw, err := json.Marshal(payload)
|
|
if err != nil {
|
|
t.Fatalf("Marshal() error = %v", err)
|
|
}
|
|
if err := store.SaveDetail("default/default", "doc", cache.DetailSnapshot{
|
|
MCPID: 1047,
|
|
Payload: raw,
|
|
}); err != nil {
|
|
t.Fatalf("SaveDetail() error = %v", err)
|
|
}
|
|
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
http.Error(w, "boom", http.StatusInternalServerError)
|
|
}))
|
|
defer server.Close()
|
|
|
|
service := discovery.NewService(market.NewClient(server.URL, server.Client()), transport.NewClient(server.Client()), store)
|
|
_, err = service.DiscoverDetail(context.Background(), market.ServerDescriptor{
|
|
Key: "doc",
|
|
DetailLocator: market.DetailLocator{
|
|
MCPID: 9629,
|
|
},
|
|
})
|
|
if err == nil {
|
|
t.Fatal("DiscoverDetail() error = nil, want live fetch failure")
|
|
}
|
|
}
|