1// SPDX-FileCopyrightText: Amolith <amolith@secluded.site>
2//
3// SPDX-License-Identifier: LicenseRef-MutuaL-1.2
4
5package mcp
6
7import (
8 "context"
9 "reflect"
10 "slices"
11 "strings"
12 "testing"
13)
14
15// tools.READ_TOOL.1 tools.SCHEMA.4 tools.RESPONSES.6
16func TestCallToolACIDToolsReadTool1RejectsUnsupportedTargetBeforeCallingCooked(t *testing.T) {
17 backend := &fakeBackend{}
18 server := NewServer(backend, "test")
19
20 _, err := server.CallReadTool(context.Background(), ReadArguments{Target: "meal_plan"})
21 if err == nil {
22 t.Fatal("CallReadTool() error = nil, want unsupported target error")
23 }
24 if err.Error() != `unsupported read target "meal_plan" (supported: shopping_list, recipes, recipe, example_recipes)` {
25 t.Fatalf("CallReadTool() error = %q", err.Error())
26 }
27 if backend.readShoppingListCalls != 0 || backend.metadataCalls != 0 || backend.contentCalls != 0 {
28 t.Fatalf(
29 "backend calls = shopping list %d metadata %d content %d, want none",
30 backend.readShoppingListCalls,
31 backend.metadataCalls,
32 backend.contentCalls,
33 )
34 }
35}
36
37// recipes.SAVE.10 tools.SCHEMA.4 tools.RESPONSES.6
38func TestCallToolACIDRecipesSave10RejectsUnsupportedSourceBeforeCallingCooked(t *testing.T) {
39 backend := &fakeBackend{}
40 server := NewServer(backend, "test")
41
42 _, _, err := server.callSaveRecipeTool(context.Background(), nil, SaveRecipeArguments{Source: "unsupported"})
43 if err == nil {
44 t.Fatal("callSaveRecipeTool() error = nil, want unsupported source error")
45 }
46 if err.Error() != `unsupported save_recipe source "unsupported" (supported: raw_text, prepared, url, draft, existing)` {
47 t.Fatalf("callSaveRecipeTool() error = %q", err.Error())
48 }
49 if backend.saveCalls != 0 || backend.draftSaveCalls != 0 {
50 t.Fatalf("save calls = prepared %d draft %d, want none", backend.saveCalls, backend.draftSaveCalls)
51 }
52}
53
54// tools.SCHEMA.3 tools.SCHEMA.5 tools.SAVE_RECIPE_TOOL.1 tools.CHANGE_SHOPPING_LIST_TOOL.1
55func TestToolDescriptionsACIDToolsSchema3And5DescribeCurrentLimits(t *testing.T) {
56 read := readTool()
57 for _, want := range []string{
58 "shopping_list",
59 "recipes",
60 "recipe with recipe_id",
61 "example_recipes",
62 "Recipe results omit images and thumbnails",
63 } {
64 if !strings.Contains(read.Description, want) {
65 t.Fatalf("read description missing %q: %q", want, read.Description)
66 }
67 }
68
69 save := saveRecipeTool()
70 for _, want := range []string{
71 "raw text",
72 "prepared markdown",
73 "URL import",
74 "reviewed URL-import draft",
75 "existing recipe update",
76 "draft_id",
77 } {
78 if !strings.Contains(save.Description, want) {
79 t.Fatalf("save_recipe description missing %q: %q", want, save.Description)
80 }
81 }
82
83 shopping := changeShoppingListTool()
84 for _, want := range []string{
85 "add",
86 "update_item",
87 "replace_selection",
88 "add_selection",
89 "remove_selection",
90 "remove",
91 "clear",
92 "Remove and clear are destructive",
93 "put quantities before names",
94 "milk — 1",
95 } {
96 if !strings.Contains(shopping.Description, want) {
97 t.Fatalf("change_shopping_list description missing %q: %q", want, shopping.Description)
98 }
99 }
100}
101
102// tools.SCHEMA.6
103func TestChangeShoppingListToolACIDToolsSchema6ExposesProductGroupIDsAsOptionalNonNullableStringArray(t *testing.T) {
104 schema := changeShoppingListInputSchema()
105 property, ok := schema.Properties["product_group_ids"]
106 if !ok {
107 t.Fatal("product_group_ids schema missing")
108 }
109 if slices.Contains(schema.Required, "product_group_ids") {
110 t.Fatalf("required fields = %#v, want product_group_ids optional", schema.Required)
111 }
112 if property.Type != "array" {
113 t.Fatalf("product_group_ids type = %q, want array", property.Type)
114 }
115 if len(property.Types) != 0 {
116 t.Fatalf("product_group_ids types = %#v, want none", property.Types)
117 }
118 if len(property.AnyOf) != 0 {
119 t.Fatalf("product_group_ids anyOf = %#v, want none", property.AnyOf)
120 }
121 if property.Items == nil || property.Items.Type != "string" {
122 t.Fatalf("product_group_ids items = %#v, want string items", property.Items)
123 }
124
125 resolved, err := schema.Resolve(nil)
126 if err != nil {
127 t.Fatalf("Resolve() error = %v", err)
128 }
129 if err := resolved.Validate(map[string]any{"action": "clear"}); err != nil {
130 t.Fatalf("Validate() with omitted product_group_ids error = %v", err)
131 }
132 if err := resolved.Validate(map[string]any{"action": "clear", "product_group_ids": nil}); err == nil {
133 t.Fatal("Validate() with null product_group_ids error = nil, want rejection")
134 }
135}
136
137// tools.READ_TOOL.1 tools.SAVE_RECIPE_TOOL.3 tools.SAVE_RECIPE_TOOL.5 tools.SAVE_RECIPE_TOOL.6
138func TestArgumentDescriptionsACIDToolsReadTool1AndSaveRecipeTool3And5And6PointToExampleRecipes(t *testing.T) {
139 readTarget, ok := reflect.TypeFor[ReadArguments]().FieldByName("Target")
140 if !ok {
141 t.Fatal("ReadArguments.Target field missing")
142 }
143 if !strings.Contains(string(readTarget.Tag), "example_recipes") {
144 t.Fatalf("ReadArguments.Target tag = %q, want example_recipes", readTarget.Tag)
145 }
146
147 saveMarkdown, ok := reflect.TypeFor[SaveRecipeArguments]().FieldByName("Markdown")
148 if !ok {
149 t.Fatal("SaveRecipeArguments.Markdown field missing")
150 }
151 for _, want := range []string{"example_recipes", "Cooked's markdown dialect"} {
152 if !strings.Contains(string(saveMarkdown.Tag), want) {
153 t.Fatalf("SaveRecipeArguments.Markdown tag missing %q: %q", want, saveMarkdown.Tag)
154 }
155 }
156}