repository resource tests #69
create repository_resource_test
- Loading branch information
commit d8b0056998c621ab9be8678259a5da79d11b5c15
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,172 @@ | ||
| package github | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/base64" | ||
| "net/http" | ||
| "testing" | ||
|
|
||
| "github.com/google/go-github/v69/github" | ||
| "github.com/mark3labs/mcp-go/mcp" | ||
| "github.com/migueleliasweb/go-github-mock/src/mock" | ||
| "github.com/stretchr/testify/assert" | ||
| "github.com/stretchr/testify/require" | ||
| ) | ||
|
|
||
| func Test_RepoContentsResourceHandler(t *testing.T) { | ||
| mockDirContent := []*github.RepositoryContent{ | ||
| { | ||
| Type: github.Ptr("file"), | ||
| Name: github.Ptr("README.md"), | ||
| Path: github.Ptr("README.md"), | ||
| SHA: github.Ptr("abc123"), | ||
| Size: github.Ptr(42), | ||
| HTMLURL: github.Ptr("https://github.com/owner/repo/blob/main/README.md"), | ||
| }, | ||
| { | ||
| Type: github.Ptr("dir"), | ||
| Name: github.Ptr("src"), | ||
| Path: github.Ptr("src"), | ||
| SHA: github.Ptr("def456"), | ||
| HTMLURL: github.Ptr("https://github.com/owner/repo/tree/main/src"), | ||
| }, | ||
| } | ||
| expectedDirContent := []mcp.TextResourceContents{ | ||
| { | ||
| URI: "https://github.com/owner/repo/blob/main/README.md", | ||
| MIMEType: "", | ||
|
There was a problem hiding this comment. feels like this shouldn't be empty, but that's what the current code returns 🤷 There was a problem hiding this comment. Hmmm, at very least I suppose we could special case markdown. I used the stdlib mime type lib, but it's perhaps not for for purpose. |
||
| Text: "README.md", | ||
| }, | ||
| { | ||
| URI: "https://github.com/owner/repo/tree/main/src", | ||
| MIMEType: "text/directory", | ||
| Text: "src", | ||
| }, | ||
| } | ||
|
|
||
| mockFileContent := &github.RepositoryContent{ | ||
| Type: github.Ptr("file"), | ||
| Name: github.Ptr("README.md"), | ||
| Path: github.Ptr("README.md"), | ||
| Content: github.Ptr("IyBUZXN0IFJlcG9zaXRvcnkKClRoaXMgaXMgYSB0ZXN0IHJlcG9zaXRvcnku"), // Base64 encoded "# Test Repository\n\nThis is a test repository." | ||
| SHA: github.Ptr("abc123"), | ||
| Size: github.Ptr(42), | ||
| HTMLURL: github.Ptr("https://github.com/owner/repo/blob/main/README.md"), | ||
| DownloadURL: github.Ptr("https://raw.githubusercontent.com/owner/repo/main/README.md"), | ||
| } | ||
|
|
||
| expectedFileContent := []mcp.BlobResourceContents{ | ||
| { | ||
| Blob: base64.StdEncoding.EncodeToString([]byte("IyBUZXN0IFJlcG9zaXRvcnkKClRoaXMgaXMgYSB0ZXN0IHJlcG9zaXRvcnku")), // Base64 encoded "# Test Repository\n\nThis is a test repository." | ||
|
|
||
| }, | ||
| } | ||
|
|
||
| tests := []struct { | ||
| name string | ||
| mockedClient *http.Client | ||
| requestArgs map[string]any | ||
| expectError bool | ||
| expectedResult any | ||
| expectedErrMsg string | ||
| }{ | ||
| { | ||
| name: "successful file content fetch", | ||
| mockedClient: mock.NewMockedHTTPClient( | ||
| mock.WithRequestMatch( | ||
| mock.GetReposContentsByOwnerByRepoByPath, | ||
| mockFileContent, | ||
| ), | ||
| ), | ||
| requestArgs: map[string]any{ | ||
| "owner": []string{"owner"}, | ||
| "repo": []string{"repo"}, | ||
| "path": []string{"README.md"}, | ||
| "branch": []string{"main"}, | ||
| }, | ||
| expectError: false, | ||
| expectedResult: expectedFileContent, | ||
| }, | ||
| { | ||
| name: "successful directory content fetch", | ||
| mockedClient: mock.NewMockedHTTPClient( | ||
| mock.WithRequestMatch( | ||
| mock.GetReposContentsByOwnerByRepoByPath, | ||
| mockDirContent, | ||
| ), | ||
| ), | ||
| requestArgs: map[string]any{ | ||
| "owner": []string{"owner"}, | ||
| "repo": []string{"repo"}, | ||
| "path": []string{"src"}, | ||
| }, | ||
| expectError: false, | ||
| expectedResult: expectedDirContent, | ||
| }, | ||
| { | ||
| name: "empty content fetch", | ||
| mockedClient: mock.NewMockedHTTPClient( | ||
| mock.WithRequestMatch( | ||
| mock.GetReposContentsByOwnerByRepoByPath, | ||
| []*github.RepositoryContent{}, | ||
| ), | ||
| ), | ||
| requestArgs: map[string]any{ | ||
| "owner": []string{"owner"}, | ||
| "repo": []string{"repo"}, | ||
| "path": []string{"src"}, | ||
| }, | ||
| expectError: false, | ||
| expectedResult: nil, | ||
| }, | ||
| { | ||
| name: "content fetch fails", | ||
| mockedClient: mock.NewMockedHTTPClient( | ||
| mock.WithRequestMatchHandler( | ||
| mock.GetReposContentsByOwnerByRepoByPath, | ||
| http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) { | ||
| w.WriteHeader(http.StatusNotFound) | ||
| _, _ = w.Write([]byte(`{"message": "Not Found"}`)) | ||
| }), | ||
| ), | ||
| ), | ||
| requestArgs: map[string]any{ | ||
| "owner": []string{"owner"}, | ||
| "repo": []string{"repo"}, | ||
| "path": []string{"nonexistent.md"}, | ||
| "branch": []string{"main"}, | ||
| }, | ||
| expectError: true, | ||
| expectedErrMsg: "404 Not Found", | ||
| }, | ||
| } | ||
|
|
||
| for _, tc := range tests { | ||
| t.Run(tc.name, func(t *testing.T) { | ||
| // Setup client with mock | ||
| client := github.NewClient(tc.mockedClient) | ||
| handler := repoContentsResourceHandler(client) | ||
|
|
||
| // Create call request | ||
| request := mcp.ReadResourceRequest{ | ||
| Params: struct { | ||
| URI string `json:"uri"` | ||
| Arguments map[string]any `json:"arguments,omitempty"` | ||
| }{ | ||
| Arguments: tc.requestArgs, | ||
| }, | ||
| } | ||
|
|
||
| resp, err := handler(context.TODO(), request) | ||
|
|
||
| if tc.expectError { | ||
| require.Error(t, err) | ||
| assert.Contains(t, err.Error(), tc.expectedErrMsg) | ||
| return | ||
| } | ||
|
|
||
| require.NoError(t, err) | ||
| require.ElementsMatch(t, resp, tc.expectedResult) | ||
| }) | ||
| } | ||
| } | ||
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
largely copied from
github-mcp-server/pkg/github/repositories_test.go
Line 18 in d8b0056