Skip to content

Commit e0a96b4

Browse files
committed
Add context
1 parent fbcc586 commit e0a96b4

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

pkg/github/repository_resource_completions_test.go

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@ func TestRepositoryResourceCompletionHandler(t *testing.T) {
5050
Ref: &mcp.CompleteReference{
5151
Type: "ref/resource",
5252
},
53+
Context: &mcp.CompleteContext{},
5354
Argument: mcp.CompleteParamsArgument{
5455
Name: "unknown_arg",
5556
Value: "test",
@@ -91,6 +92,11 @@ func TestRepositoryResourceCompletionHandler_GetClientError(t *testing.T) {
9192
Ref: &mcp.CompleteReference{
9293
Type: "ref/resource",
9394
},
95+
Context: &mcp.CompleteContext{
96+
Arguments: map[string]string{
97+
"owner": "test",
98+
},
99+
},
94100
Argument: mcp.CompleteParamsArgument{
95101
Name: "owner",
96102
Value: "test",
@@ -262,6 +268,11 @@ func TestRepositoryResourceCompletionHandler_MaxResults(t *testing.T) {
262268
Ref: &mcp.CompleteReference{
263269
Type: "ref/resource",
264270
},
271+
Context: &mcp.CompleteContext{
272+
Arguments: map[string]string{
273+
"owner": "test",
274+
},
275+
},
265276
Argument: mcp.CompleteParamsArgument{
266277
Name: "owner",
267278
Value: "test",
@@ -346,6 +357,9 @@ func TestRepositoryResourceCompletionHandler_NilContext(t *testing.T) {
346357
Value: "test",
347358
},
348359
// Context is not set, so it should default to empty map
360+
Context: &mcp.CompleteContext{
361+
Arguments: map[string]string{},
362+
},
349363
},
350364
}
351365

0 commit comments

Comments
 (0)