|
7 | 7 | "encoding/base64" |
8 | 8 | "encoding/json" |
9 | 9 | "fmt" |
| 10 | + "net/http" |
10 | 11 | "os" |
11 | 12 | "os/exec" |
12 | 13 | "slices" |
@@ -61,7 +62,8 @@ func getRESTClient(t *testing.T) *gogithub.Client { |
61 | 62 |
|
62 | 63 | // Create a new GitHub client with the token |
63 | 64 | ghClient := gogithub.NewClient(nil).WithAuthToken(token) |
64 | | - if host := getE2EHost(); host != "https://github.com" { |
| 65 | + |
| 66 | + if host := getE2EHost(); host != "" && host != "https://github.com" { |
65 | 67 | var err error |
66 | 68 | // Currently this works for GHEC because the API is exposed at the api subdomain and the path prefix |
67 | 69 | // but it would be preferable to extract the host parsing from the main server logic, and use it here. |
@@ -210,7 +212,6 @@ func TestGetMe(t *testing.T) { |
210 | 212 | t.Parallel() |
211 | 213 |
|
212 | 214 | mcpClient := setupMCPClient(t) |
213 | | - |
214 | 215 | ctx := context.Background() |
215 | 216 |
|
216 | 217 | // When we call the "get_me" tool |
@@ -795,14 +796,13 @@ func TestDirectoryDeletion(t *testing.T) { |
795 | 796 | } |
796 | 797 |
|
797 | 798 | func TestRequestCopilotReview(t *testing.T) { |
| 799 | + t.Parallel() |
| 800 | + |
798 | 801 | if getE2EHost() != "" && getE2EHost() != "https://github.com" { |
799 | 802 | t.Skip("Skipping test because the host does not support copilot reviews") |
800 | 803 | } |
801 | 804 |
|
802 | | - t.Parallel() |
803 | | - |
804 | 805 | mcpClient := setupMCPClient(t) |
805 | | - |
806 | 806 | ctx := context.Background() |
807 | 807 |
|
808 | 808 | // First, who am I |
@@ -943,6 +943,112 @@ func TestRequestCopilotReview(t *testing.T) { |
943 | 943 | require.Equal(t, "Bot", *reviewRequests.Users[0].Type, "expected review request to be for Bot") |
944 | 944 | } |
945 | 945 |
|
| 946 | +func TestAssignCopilotToIssue(t *testing.T) { |
| 947 | + t.Parallel() |
| 948 | + |
| 949 | + if getE2EHost() != "" && getE2EHost() != "https://github.com" { |
| 950 | + t.Skip("Skipping test because the host does not support copilot being assigned to issues") |
| 951 | + } |
| 952 | + |
| 953 | + mcpClient := setupMCPClient(t) |
| 954 | + ctx := context.Background() |
| 955 | + |
| 956 | + // First, who am I |
| 957 | + getMeRequest := mcp.CallToolRequest{} |
| 958 | + getMeRequest.Params.Name = "get_me" |
| 959 | + |
| 960 | + t.Log("Getting current user...") |
| 961 | + resp, err := mcpClient.CallTool(ctx, getMeRequest) |
| 962 | + require.NoError(t, err, "expected to call 'get_me' tool successfully") |
| 963 | + require.False(t, resp.IsError, fmt.Sprintf("expected result not to be an error: %+v", resp)) |
| 964 | + |
| 965 | + require.False(t, resp.IsError, "expected result not to be an error") |
| 966 | + require.Len(t, resp.Content, 1, "expected content to have one item") |
| 967 | + |
| 968 | + textContent, ok := resp.Content[0].(mcp.TextContent) |
| 969 | + require.True(t, ok, "expected content to be of type TextContent") |
| 970 | + |
| 971 | + var trimmedGetMeText struct { |
| 972 | + Login string `json:"login"` |
| 973 | + } |
| 974 | + err = json.Unmarshal([]byte(textContent.Text), &trimmedGetMeText) |
| 975 | + require.NoError(t, err, "expected to unmarshal text content successfully") |
| 976 | + |
| 977 | + currentOwner := trimmedGetMeText.Login |
| 978 | + |
| 979 | + // Then create a repository with a README (via autoInit) |
| 980 | + repoName := fmt.Sprintf("github-mcp-server-e2e-%s-%d", t.Name(), time.Now().UnixMilli()) |
| 981 | + createRepoRequest := mcp.CallToolRequest{} |
| 982 | + createRepoRequest.Params.Name = "create_repository" |
| 983 | + createRepoRequest.Params.Arguments = map[string]any{ |
| 984 | + "name": repoName, |
| 985 | + "private": true, |
| 986 | + "autoInit": true, |
| 987 | + } |
| 988 | + |
| 989 | + t.Logf("Creating repository %s/%s...", currentOwner, repoName) |
| 990 | + _, err = mcpClient.CallTool(ctx, createRepoRequest) |
| 991 | + require.NoError(t, err, "expected to call 'create_repository' tool successfully") |
| 992 | + require.False(t, resp.IsError, fmt.Sprintf("expected result not to be an error: %+v", resp)) |
| 993 | + |
| 994 | + // Cleanup the repository after the test |
| 995 | + t.Cleanup(func() { |
| 996 | + // MCP Server doesn't support deletions, but we can use the GitHub Client |
| 997 | + ghClient := getRESTClient(t) |
| 998 | + t.Logf("Deleting repository %s/%s...", currentOwner, repoName) |
| 999 | + _, err := ghClient.Repositories.Delete(context.Background(), currentOwner, repoName) |
| 1000 | + require.NoError(t, err, "expected to delete repository successfully") |
| 1001 | + }) |
| 1002 | + |
| 1003 | + // Create an issue |
| 1004 | + createIssueRequest := mcp.CallToolRequest{} |
| 1005 | + createIssueRequest.Params.Name = "create_issue" |
| 1006 | + createIssueRequest.Params.Arguments = map[string]any{ |
| 1007 | + "owner": currentOwner, |
| 1008 | + "repo": repoName, |
| 1009 | + "title": "Test issue to assign copilot to", |
| 1010 | + } |
| 1011 | + |
| 1012 | + t.Logf("Creating issue in %s/%s...", currentOwner, repoName) |
| 1013 | + resp, err = mcpClient.CallTool(ctx, createIssueRequest) |
| 1014 | + require.NoError(t, err, "expected to call 'create_issue' tool successfully") |
| 1015 | + require.False(t, resp.IsError, fmt.Sprintf("expected result not to be an error: %+v", resp)) |
| 1016 | + |
| 1017 | + // Assign copilot to the issue |
| 1018 | + assignCopilotRequest := mcp.CallToolRequest{} |
| 1019 | + assignCopilotRequest.Params.Name = "assign_copilot_to_issue" |
| 1020 | + assignCopilotRequest.Params.Arguments = map[string]any{ |
| 1021 | + "owner": currentOwner, |
| 1022 | + "repo": repoName, |
| 1023 | + "issueNumber": 1, |
| 1024 | + } |
| 1025 | + |
| 1026 | + t.Logf("Assigning copilot to issue in %s/%s...", currentOwner, repoName) |
| 1027 | + resp, err = mcpClient.CallTool(ctx, assignCopilotRequest) |
| 1028 | + require.NoError(t, err, "expected to call 'assign_copilot_to_issue' tool successfully") |
| 1029 | + |
| 1030 | + textContent, ok = resp.Content[0].(mcp.TextContent) |
| 1031 | + require.True(t, ok, "expected content to be of type TextContent") |
| 1032 | + |
| 1033 | + possibleExpectedFailure := "copilot isn't available as an assignee for this issue. Please inform the user to visit https://docs.github.com/en/copilot/using-github-copilot/using-copilot-coding-agent-to-work-on-tasks/about-assigning-tasks-to-copilot for more information." |
| 1034 | + if resp.IsError && textContent.Text == possibleExpectedFailure { |
| 1035 | + t.Skip("skipping because copilot wasn't available as an assignee on this issue, it's likely that the owner doesn't have copilot enabled in their settings") |
| 1036 | + } |
| 1037 | + |
| 1038 | + require.False(t, resp.IsError, fmt.Sprintf("expected result not to be an error: %+v", resp)) |
| 1039 | + |
| 1040 | + require.Equal(t, "successfully assigned copilot to issue", textContent.Text) |
| 1041 | + |
| 1042 | + // Check that copilot is assigned to the issue |
| 1043 | + // MCP Server doesn't support getting assignees yet |
| 1044 | + ghClient := getRESTClient(t) |
| 1045 | + assignees, response, err := ghClient.Issues.Get(context.Background(), currentOwner, repoName, 1) |
| 1046 | + require.NoError(t, err, "expected to get issue successfully") |
| 1047 | + require.Equal(t, http.StatusOK, response.StatusCode, "expected to get issue successfully") |
| 1048 | + require.Len(t, assignees.Assignees, 1, "expected to find one assignee") |
| 1049 | + require.Equal(t, "Copilot", *assignees.Assignees[0].Login, "expected copilot to be assigned to the issue") |
| 1050 | +} |
| 1051 | + |
946 | 1052 | func TestPullRequestAtomicCreateAndSubmit(t *testing.T) { |
947 | 1053 | t.Parallel() |
948 | 1054 |
|
@@ -1145,7 +1251,7 @@ func TestPullRequestReviewCommentSubmit(t *testing.T) { |
1145 | 1251 |
|
1146 | 1252 | t.Logf("Creating repository %s/%s...", currentOwner, repoName) |
1147 | 1253 | _, err = mcpClient.CallTool(ctx, createRepoRequest) |
1148 | | - require.NoError(t, err, "expected to call 'get_me' tool successfully") |
| 1254 | + require.NoError(t, err, "expected to call 'create_repository' tool successfully") |
1149 | 1255 | require.False(t, resp.IsError, fmt.Sprintf("expected result not to be an error: %+v", resp)) |
1150 | 1256 |
|
1151 | 1257 | // Cleanup the repository after the test |
|
0 commit comments