dd4b2939e3## Motivation and Context Replace shared temp paths with `t.TempDir()`` so each test gets its own cleanup-scoped directory. - This avoids leftover files and cross-test interference. No production code changed. ## How Has This Been Tested? GitHub CI passed. - https://github.com/yowainwright/registry/pull/1 ## Breaking Changes No breaking changes; just tests. ## Types of changes - [x] chore ## Checklist <!-- Go over all the following points, and put an `x` in all the boxes that apply. --> - [x] I have read the [MCP Documentation](https://modelcontextprotocol.io) - [x] My code follows the repository's style guidelines - [x] New and existing tests pass locally - [x] I have added appropriate error handling - [x] I have added or updated documentation as needed ## Additional context Small test-only cleanup. No production code paths are changed. Co-authored-by: Radoslav Dimitrov <radoslav@stacklok.com>
116 lines
3 KiB
Go
116 lines
3 KiB
Go
package commands_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/modelcontextprotocol/registry/internal/validators"
|
|
apiv0 "github.com/modelcontextprotocol/registry/pkg/api/v0"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// SetupMockRegistryServer creates an httptest.Server that mocks the registry API
|
|
func SetupMockRegistryServer(t *testing.T, publishHandler func(w http.ResponseWriter, r *http.Request), validateHandler func(w http.ResponseWriter, r *http.Request)) *httptest.Server {
|
|
t.Helper()
|
|
|
|
mux := http.NewServeMux()
|
|
|
|
// Default handlers if not provided
|
|
if publishHandler == nil {
|
|
publishHandler = func(w http.ResponseWriter, _ *http.Request) {
|
|
w.WriteHeader(http.StatusCreated)
|
|
response := apiv0.ServerResponse{
|
|
Server: apiv0.ServerJSON{
|
|
Name: "com.example/test",
|
|
Version: "1.0.0",
|
|
},
|
|
}
|
|
_ = json.NewEncoder(w).Encode(response)
|
|
}
|
|
}
|
|
|
|
if validateHandler == nil {
|
|
validateHandler = func(w http.ResponseWriter, _ *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
result := validators.ValidationResult{Valid: true}
|
|
_ = json.NewEncoder(w).Encode(result)
|
|
}
|
|
}
|
|
|
|
mux.HandleFunc("/v0/publish", func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodPost {
|
|
w.WriteHeader(http.StatusMethodNotAllowed)
|
|
return
|
|
}
|
|
publishHandler(w, r)
|
|
})
|
|
|
|
mux.HandleFunc("/v0/validate", func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodPost {
|
|
w.WriteHeader(http.StatusMethodNotAllowed)
|
|
return
|
|
}
|
|
validateHandler(w, r)
|
|
})
|
|
|
|
server := httptest.NewServer(mux)
|
|
t.Cleanup(server.Close)
|
|
|
|
return server
|
|
}
|
|
|
|
// SetupTestToken creates a token file pointing to the test server.
|
|
// It overrides $HOME to a temp directory so tests don't touch real credentials.
|
|
func SetupTestToken(t *testing.T, registryURL, token string) string {
|
|
t.Helper()
|
|
|
|
// Override the home directory so tokenFilePath() resolves to a temp directory.
|
|
tempHome := t.TempDir()
|
|
t.Setenv("HOME", tempHome)
|
|
t.Setenv("USERPROFILE", tempHome)
|
|
|
|
dir := filepath.Join(tempHome, ".config", "mcp-publisher")
|
|
require.NoError(t, os.MkdirAll(dir, 0700))
|
|
|
|
tokenPath := filepath.Join(dir, "token.json")
|
|
tokenData := map[string]string{
|
|
"token": token,
|
|
"registry": registryURL,
|
|
}
|
|
|
|
data, err := json.Marshal(tokenData)
|
|
require.NoError(t, err)
|
|
|
|
err = os.WriteFile(tokenPath, data, 0600)
|
|
require.NoError(t, err)
|
|
|
|
return tokenPath
|
|
}
|
|
|
|
// CreateTestServerJSON creates a server.json file in a temp directory and changes to it
|
|
func CreateTestServerJSON(t *testing.T, serverJSON apiv0.ServerJSON) (string, string) {
|
|
t.Helper()
|
|
|
|
tempDir := t.TempDir()
|
|
|
|
jsonData, err := json.MarshalIndent(serverJSON, "", " ")
|
|
require.NoError(t, err)
|
|
|
|
serverFile := filepath.Join(tempDir, "server.json")
|
|
err = os.WriteFile(serverFile, jsonData, 0600)
|
|
require.NoError(t, err)
|
|
|
|
// Change to temp directory
|
|
originalDir, err := os.Getwd()
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() { _ = os.Chdir(originalDir) })
|
|
|
|
err = os.Chdir(tempDir)
|
|
require.NoError(t, err)
|
|
|
|
return tempDir, serverFile
|
|
}
|