adding tests

tonytrg · SamMorrowDrums · commit 6fa8eafff5f7 · 2025-06-23T14:46:46.000+02:00
diff --git a/pkg/github/__toolsnaps__/get_file_contents.snap b/pkg/github/__toolsnaps__/get_file_contents.snap
@@ -6,10 +6,6 @@
   "description": "Get the contents of a file or directory from a GitHub repository",
   "inputSchema": {
     "properties": {
-      "branch": {
-        "description": "Branch to get contents from",
-        "type": "string"
-      },
       "owner": {
         "description": "Repository owner (username or organization)",
         "type": "string"
@@ -18,9 +14,17 @@
         "description": "Path to file/directory (directories must end with a slash '/')",
         "type": "string"
       },
+      "ref": {
+        "description": "Accepts optional git refs such as `refs/tags/\u003ctag\u003e`, `refs/heads/\u003cbranch\u003e` or `refs/pull/\u003cpr_number\u003e/head`",
+        "type": "string"
+      },
       "repo": {
         "description": "Repository name",
         "type": "string"
+      },
+      "sha": {
+        "description": "Accepts optional git sha, if sha is specified it will be used instead of ref",
+        "type": "string"
       }
     },
     "required": [
diff --git a/pkg/github/repositories_test.go b/pkg/github/repositories_test.go
@@ -31,7 +31,8 @@ func Test_GetFileContents(t *testing.T) {
 	assert.Contains(t, tool.InputSchema.Properties, "owner")
 	assert.Contains(t, tool.InputSchema.Properties, "repo")
 	assert.Contains(t, tool.InputSchema.Properties, "path")
-	assert.Contains(t, tool.InputSchema.Properties, "branch")
+	assert.Contains(t, tool.InputSchema.Properties, "ref")
+	assert.Contains(t, tool.InputSchema.Properties, "sha")
 	assert.ElementsMatch(t, tool.InputSchema.Required, []string{"owner", "repo", "path"})
 
 	// Mock response for raw content
@@ -77,10 +78,10 @@ func Test_GetFileContents(t *testing.T) {
 				),
 			),
 			requestArgs: map[string]interface{}{
-				"owner":  "owner",
-				"repo":   "repo",
-				"path":   "README.md",
-				"branch": "main",
+				"owner": "owner",
+				"repo":  "repo",
+				"path":  "README.md",
+				"ref":   "refs/heads/main",
 			},
 			expectError: false,
 			expectedResult: mcp.TextResourceContents{
@@ -101,10 +102,10 @@ func Test_GetFileContents(t *testing.T) {
 				),
 			),
 			requestArgs: map[string]interface{}{
-				"owner":  "owner",
-				"repo":   "repo",
-				"path":   "test.png",
-				"branch": "main",
+				"owner": "owner",
+				"repo":  "repo",
+				"path":  "test.png",
+				"ref":   "refs/heads/main",
 			},
 			expectError: false,
 			expectedResult: mcp.BlobResourceContents{
@@ -158,10 +159,10 @@ func Test_GetFileContents(t *testing.T) {
 				),
 			),
 			requestArgs: map[string]interface{}{
-				"owner":  "owner",
-				"repo":   "repo",
-				"path":   "nonexistent.md",
-				"branch": "main",
+				"owner": "owner",
+				"repo":  "repo",
+				"path":  "nonexistent.md",
+				"ref":   "refs/heads/main",
 			},
 			expectError:    false,
 			expectedResult: mcp.NewToolResultError("Failed to get file contents. The path does not point to a file or directory, or the file does not exist in the repository."),
