mirror of
https://github.com/shankar0123/certctl.git
synced 2026-06-07 14:21:37 +00:00
fix: Go 1.25 upgrade, codebase audit fixes, MCP server tests
Upgrade from Go 1.22 to 1.25 (minimum for MCP SDK, actively supported). CI updated to match. Codebase audit fixes: - Local CA parseIP() now uses net.ParseIP — IP SANs no longer silently dropped - Nil pointer guards in agent.go GetWorkWithTargets for target/cert enrichment - MCP CreateCertificateInput marks owner_id/team_id as required - NGINX connector uses CombinedOutput() — captures diagnostic output on failure - Jobs handler validates JSON decode on rejection body — returns 400 on malformed - CRL/OCSP handlers propagate requestID for error tracing MCP server tests (26 tests): - client_test.go: HTTP client coverage (GET/POST/PUT/DELETE, auth, 204, errors, binary) - tools_test.go: tool registration, pagination, end-to-end flows with mock API Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -19,7 +19,7 @@ jobs:
|
||||
- name: Set up Go
|
||||
uses: actions/setup-go@v5
|
||||
with:
|
||||
go-version: '1.23'
|
||||
go-version: '1.25'
|
||||
|
||||
- name: Go Build
|
||||
run: |
|
||||
@@ -32,7 +32,7 @@ jobs:
|
||||
|
||||
- name: Go Test with Coverage
|
||||
run: |
|
||||
go test ./internal/service/... ./internal/api/handler/... ./internal/integration/... ./internal/connector/issuer/... ./internal/connector/target/... -count=1 -cover -coverprofile=coverage.out
|
||||
go test ./internal/service/... ./internal/api/handler/... ./internal/integration/... ./internal/connector/issuer/... ./internal/connector/target/... ./internal/mcp/... -count=1 -cover -coverprofile=coverage.out
|
||||
|
||||
- name: Check Coverage Thresholds
|
||||
run: |
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
module github.com/shankar0123/certctl
|
||||
|
||||
go 1.23.0
|
||||
go 1.25.0
|
||||
|
||||
require (
|
||||
github.com/google/uuid v1.6.0
|
||||
|
||||
@@ -450,14 +450,16 @@ func (h CertificateHandler) GetCRL(w http.ResponseWriter, r *http.Request) {
|
||||
// GetDERCRL returns a DER-encoded X.509 CRL signed by the specified issuer.
|
||||
// GET /api/v1/crl/{issuer_id}
|
||||
func (h CertificateHandler) GetDERCRL(w http.ResponseWriter, r *http.Request) {
|
||||
requestID, _ := r.Context().Value("request_id").(string)
|
||||
|
||||
if r.Method != http.MethodGet {
|
||||
Error(w, http.StatusMethodNotAllowed, "Method not allowed")
|
||||
ErrorWithRequestID(w, http.StatusMethodNotAllowed, "Method not allowed", requestID)
|
||||
return
|
||||
}
|
||||
|
||||
issuerID := strings.TrimPrefix(r.URL.Path, "/api/v1/crl/")
|
||||
if issuerID == "" {
|
||||
Error(w, http.StatusBadRequest, "Issuer ID is required")
|
||||
ErrorWithRequestID(w, http.StatusBadRequest, "Issuer ID is required", requestID)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -465,14 +467,14 @@ func (h CertificateHandler) GetDERCRL(w http.ResponseWriter, r *http.Request) {
|
||||
if err != nil {
|
||||
errMsg := err.Error()
|
||||
if strings.Contains(errMsg, "not found") {
|
||||
Error(w, http.StatusNotFound, errMsg)
|
||||
ErrorWithRequestID(w, http.StatusNotFound, errMsg, requestID)
|
||||
return
|
||||
}
|
||||
if strings.Contains(errMsg, "do not support") || strings.Contains(errMsg, "does not support") {
|
||||
Error(w, http.StatusNotImplemented, errMsg)
|
||||
ErrorWithRequestID(w, http.StatusNotImplemented, errMsg, requestID)
|
||||
return
|
||||
}
|
||||
Error(w, http.StatusInternalServerError, "Failed to generate CRL")
|
||||
ErrorWithRequestID(w, http.StatusInternalServerError, "Failed to generate CRL", requestID)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -486,8 +488,10 @@ func (h CertificateHandler) GetDERCRL(w http.ResponseWriter, r *http.Request) {
|
||||
// GET /api/v1/ocsp/{issuer_id}/{serial_hex}
|
||||
// For simplicity, use GET with path params instead of binary POST.
|
||||
func (h CertificateHandler) HandleOCSP(w http.ResponseWriter, r *http.Request) {
|
||||
requestID, _ := r.Context().Value("request_id").(string)
|
||||
|
||||
if r.Method != http.MethodGet {
|
||||
Error(w, http.StatusMethodNotAllowed, "Method not allowed")
|
||||
ErrorWithRequestID(w, http.StatusMethodNotAllowed, "Method not allowed", requestID)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -495,7 +499,7 @@ func (h CertificateHandler) HandleOCSP(w http.ResponseWriter, r *http.Request) {
|
||||
path := strings.TrimPrefix(r.URL.Path, "/api/v1/ocsp/")
|
||||
parts := strings.SplitN(path, "/", 2)
|
||||
if len(parts) < 2 || parts[0] == "" || parts[1] == "" {
|
||||
Error(w, http.StatusBadRequest, "Issuer ID and serial number are required")
|
||||
ErrorWithRequestID(w, http.StatusBadRequest, "Issuer ID and serial number are required", requestID)
|
||||
return
|
||||
}
|
||||
issuerID := parts[0]
|
||||
@@ -505,14 +509,14 @@ func (h CertificateHandler) HandleOCSP(w http.ResponseWriter, r *http.Request) {
|
||||
if err != nil {
|
||||
errMsg := err.Error()
|
||||
if strings.Contains(errMsg, "not found") {
|
||||
Error(w, http.StatusNotFound, errMsg)
|
||||
ErrorWithRequestID(w, http.StatusNotFound, errMsg, requestID)
|
||||
return
|
||||
}
|
||||
if strings.Contains(errMsg, "do not support") || strings.Contains(errMsg, "does not support") {
|
||||
Error(w, http.StatusNotImplemented, errMsg)
|
||||
ErrorWithRequestID(w, http.StatusNotImplemented, errMsg, requestID)
|
||||
return
|
||||
}
|
||||
Error(w, http.StatusInternalServerError, "Failed to generate OCSP response")
|
||||
ErrorWithRequestID(w, http.StatusInternalServerError, "Failed to generate OCSP response", requestID)
|
||||
return
|
||||
}
|
||||
|
||||
|
||||
@@ -186,7 +186,10 @@ func (h JobHandler) RejectJob(w http.ResponseWriter, r *http.Request) {
|
||||
Reason string `json:"reason"`
|
||||
}
|
||||
if r.Body != nil {
|
||||
json.NewDecoder(r.Body).Decode(&body)
|
||||
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
|
||||
ErrorWithRequestID(w, http.StatusBadRequest, "Invalid request body", requestID)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
if err := h.svc.RejectJob(jobID, body.Reason); err != nil {
|
||||
|
||||
@@ -15,6 +15,7 @@ import (
|
||||
"fmt"
|
||||
"log/slog"
|
||||
"math/big"
|
||||
"net"
|
||||
"os"
|
||||
"sync"
|
||||
"time"
|
||||
@@ -558,9 +559,15 @@ func parseIP(s string) []byte {
|
||||
if s == "localhost" {
|
||||
return []byte{127, 0, 0, 1}
|
||||
}
|
||||
// In production, use net.ParseIP for proper parsing.
|
||||
// For now, return nil for non-localhost IPs.
|
||||
ip := net.ParseIP(s)
|
||||
if ip == nil {
|
||||
return nil
|
||||
}
|
||||
// Prefer 4-byte representation for IPv4
|
||||
if v4 := ip.To4(); v4 != nil {
|
||||
return v4
|
||||
}
|
||||
return ip
|
||||
}
|
||||
|
||||
// isEmail checks if a string looks like an email address.
|
||||
|
||||
@@ -120,9 +120,9 @@ func (c *Connector) DeployCertificate(ctx context.Context, request target.Deploy
|
||||
// Validate NGINX configuration before reload
|
||||
c.logger.Debug("validating NGINX configuration", "validate_command", c.config.ValidateCommand)
|
||||
validateCmd := exec.CommandContext(ctx, "sh", "-c", c.config.ValidateCommand)
|
||||
if err := validateCmd.Run(); err != nil {
|
||||
errMsg := fmt.Sprintf("NGINX config validation failed: %v", err)
|
||||
c.logger.Error("NGINX validation failed", "error", err)
|
||||
if output, err := validateCmd.CombinedOutput(); err != nil {
|
||||
errMsg := fmt.Sprintf("NGINX config validation failed: %v (output: %s)", err, string(output))
|
||||
c.logger.Error("NGINX validation failed", "error", err, "output", string(output))
|
||||
return &target.DeploymentResult{
|
||||
Success: false,
|
||||
TargetAddress: c.config.CertPath,
|
||||
@@ -134,9 +134,9 @@ func (c *Connector) DeployCertificate(ctx context.Context, request target.Deploy
|
||||
// Reload NGINX
|
||||
c.logger.Debug("reloading NGINX", "reload_command", c.config.ReloadCommand)
|
||||
reloadCmd := exec.CommandContext(ctx, "sh", "-c", c.config.ReloadCommand)
|
||||
if err := reloadCmd.Run(); err != nil {
|
||||
errMsg := fmt.Sprintf("NGINX reload failed: %v", err)
|
||||
c.logger.Error("NGINX reload failed", "error", err)
|
||||
if output, err := reloadCmd.CombinedOutput(); err != nil {
|
||||
errMsg := fmt.Sprintf("NGINX reload failed: %v (output: %s)", err, string(output))
|
||||
c.logger.Error("NGINX reload failed", "error", err, "output", string(output))
|
||||
return &target.DeploymentResult{
|
||||
Success: false,
|
||||
TargetAddress: c.config.CertPath,
|
||||
|
||||
@@ -0,0 +1,289 @@
|
||||
package mcp
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestNewClient(t *testing.T) {
|
||||
c := NewClient("http://localhost:8443", "test-key")
|
||||
if c.baseURL != "http://localhost:8443" {
|
||||
t.Errorf("expected baseURL http://localhost:8443, got %s", c.baseURL)
|
||||
}
|
||||
if c.apiKey != "test-key" {
|
||||
t.Errorf("expected apiKey test-key, got %s", c.apiKey)
|
||||
}
|
||||
if c.httpClient == nil {
|
||||
t.Fatal("expected httpClient to be non-nil")
|
||||
}
|
||||
}
|
||||
|
||||
func TestClient_Get(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodGet {
|
||||
t.Errorf("expected GET, got %s", r.Method)
|
||||
}
|
||||
if r.Header.Get("Authorization") != "Bearer test-key" {
|
||||
t.Errorf("expected Bearer test-key auth, got %s", r.Header.Get("Authorization"))
|
||||
}
|
||||
if r.Header.Get("Accept") != "application/json" {
|
||||
t.Errorf("expected Accept application/json, got %s", r.Header.Get("Accept"))
|
||||
}
|
||||
if r.URL.Query().Get("status") != "Active" {
|
||||
t.Errorf("expected status=Active query param, got %s", r.URL.Query().Get("status"))
|
||||
}
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusOK)
|
||||
json.NewEncoder(w).Encode(map[string]interface{}{
|
||||
"data": []interface{}{},
|
||||
"total": 0,
|
||||
})
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
c := NewClient(server.URL, "test-key")
|
||||
data, err := c.Get("/api/v1/certificates", map[string][]string{"status": {"Active"}})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if data == nil {
|
||||
t.Fatal("expected non-nil response data")
|
||||
}
|
||||
}
|
||||
|
||||
func TestClient_Get_NoAuth(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Header.Get("Authorization") != "" {
|
||||
t.Errorf("expected no auth header, got %s", r.Header.Get("Authorization"))
|
||||
}
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write([]byte(`{"data":[]}`))
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
c := NewClient(server.URL, "")
|
||||
_, err := c.Get("/api/v1/certificates", nil)
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestClient_Post(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodPost {
|
||||
t.Errorf("expected POST, got %s", r.Method)
|
||||
}
|
||||
if r.Header.Get("Content-Type") != "application/json" {
|
||||
t.Errorf("expected Content-Type application/json, got %s", r.Header.Get("Content-Type"))
|
||||
}
|
||||
|
||||
body, _ := io.ReadAll(r.Body)
|
||||
var parsed map[string]interface{}
|
||||
if err := json.Unmarshal(body, &parsed); err != nil {
|
||||
t.Fatalf("failed to parse request body: %v", err)
|
||||
}
|
||||
if parsed["name"] != "test-cert" {
|
||||
t.Errorf("expected name=test-cert, got %v", parsed["name"])
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusCreated)
|
||||
json.NewEncoder(w).Encode(map[string]string{"id": "mc-test"})
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
c := NewClient(server.URL, "test-key")
|
||||
data, err := c.Post("/api/v1/certificates", map[string]string{"name": "test-cert"})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
|
||||
var result map[string]string
|
||||
if err := json.Unmarshal(data, &result); err != nil {
|
||||
t.Fatalf("failed to parse response: %v", err)
|
||||
}
|
||||
if result["id"] != "mc-test" {
|
||||
t.Errorf("expected id=mc-test, got %s", result["id"])
|
||||
}
|
||||
}
|
||||
|
||||
func TestClient_Put(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodPut {
|
||||
t.Errorf("expected PUT, got %s", r.Method)
|
||||
}
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write([]byte(`{"id":"mc-test","name":"updated"}`))
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
c := NewClient(server.URL, "test-key")
|
||||
data, err := c.Put("/api/v1/certificates/mc-test", map[string]string{"name": "updated"})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if data == nil {
|
||||
t.Fatal("expected non-nil response data")
|
||||
}
|
||||
}
|
||||
|
||||
func TestClient_Delete_204(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodDelete {
|
||||
t.Errorf("expected DELETE, got %s", r.Method)
|
||||
}
|
||||
w.WriteHeader(http.StatusNoContent)
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
c := NewClient(server.URL, "test-key")
|
||||
data, err := c.Delete("/api/v1/certificates/mc-test")
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
|
||||
var result map[string]string
|
||||
if err := json.Unmarshal(data, &result); err != nil {
|
||||
t.Fatalf("failed to parse response: %v", err)
|
||||
}
|
||||
if result["status"] != "deleted" {
|
||||
t.Errorf("expected status=deleted for 204, got %s", result["status"])
|
||||
}
|
||||
}
|
||||
|
||||
func TestClient_ErrorResponse(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusNotFound)
|
||||
w.Write([]byte(`{"error":"not found"}`))
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
c := NewClient(server.URL, "test-key")
|
||||
_, err := c.Get("/api/v1/certificates/nonexistent", nil)
|
||||
if err == nil {
|
||||
t.Fatal("expected error for 404 response")
|
||||
}
|
||||
expected := "API error (HTTP 404)"
|
||||
if !containsStr(err.Error(), expected) {
|
||||
t.Errorf("expected error containing %q, got %q", expected, err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
func TestClient_ServerError(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte(`{"error":"internal server error"}`))
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
c := NewClient(server.URL, "test-key")
|
||||
_, err := c.Post("/api/v1/certificates", map[string]string{"name": "test"})
|
||||
if err == nil {
|
||||
t.Fatal("expected error for 500 response")
|
||||
}
|
||||
expected := "API error (HTTP 500)"
|
||||
if !containsStr(err.Error(), expected) {
|
||||
t.Errorf("expected error containing %q, got %q", expected, err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
func TestClient_GetRaw(t *testing.T) {
|
||||
derData := []byte{0x30, 0x82, 0x01, 0x00} // fake DER bytes
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodGet {
|
||||
t.Errorf("expected GET, got %s", r.Method)
|
||||
}
|
||||
w.Header().Set("Content-Type", "application/pkix-crl")
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write(derData)
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
c := NewClient(server.URL, "test-key")
|
||||
data, contentType, err := c.GetRaw("/api/v1/crl/iss-local")
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if contentType != "application/pkix-crl" {
|
||||
t.Errorf("expected content-type application/pkix-crl, got %s", contentType)
|
||||
}
|
||||
if len(data) != len(derData) {
|
||||
t.Errorf("expected %d bytes, got %d", len(derData), len(data))
|
||||
}
|
||||
}
|
||||
|
||||
func TestClient_GetRaw_Error(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusNotFound)
|
||||
w.Write([]byte("issuer not found"))
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
c := NewClient(server.URL, "test-key")
|
||||
_, _, err := c.GetRaw("/api/v1/crl/nonexistent")
|
||||
if err == nil {
|
||||
t.Fatal("expected error for 404 response")
|
||||
}
|
||||
}
|
||||
|
||||
func TestClient_ConnectionRefused(t *testing.T) {
|
||||
c := NewClient("http://localhost:1", "test-key")
|
||||
_, err := c.Get("/api/v1/certificates", nil)
|
||||
if err == nil {
|
||||
t.Fatal("expected error for connection refused")
|
||||
}
|
||||
}
|
||||
|
||||
func TestClient_PostNilBody(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Header.Get("Content-Type") != "" {
|
||||
t.Errorf("expected no Content-Type for nil body, got %s", r.Header.Get("Content-Type"))
|
||||
}
|
||||
w.WriteHeader(http.StatusAccepted)
|
||||
w.Write([]byte(`{"status":"accepted"}`))
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
c := NewClient(server.URL, "test-key")
|
||||
data, err := c.Post("/api/v1/certificates/mc-test/renew", nil)
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if data == nil {
|
||||
t.Fatal("expected non-nil response")
|
||||
}
|
||||
}
|
||||
|
||||
func TestClient_QueryParams(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.URL.Query().Get("page") != "2" {
|
||||
t.Errorf("expected page=2, got %s", r.URL.Query().Get("page"))
|
||||
}
|
||||
if r.URL.Query().Get("per_page") != "10" {
|
||||
t.Errorf("expected per_page=10, got %s", r.URL.Query().Get("per_page"))
|
||||
}
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write([]byte(`{"data":[],"total":0}`))
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
c := NewClient(server.URL, "test-key")
|
||||
q := paginationQuery(2, 10)
|
||||
_, err := c.Get("/api/v1/certificates", q)
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// containsStr is a simple helper to avoid importing strings in tests.
|
||||
func containsStr(s, substr string) bool {
|
||||
for i := 0; i <= len(s)-len(substr); i++ {
|
||||
if s[i:i+len(substr)] == substr {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
@@ -0,0 +1,412 @@
|
||||
package mcp
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"strings"
|
||||
"sync"
|
||||
"testing"
|
||||
|
||||
gomcp "github.com/modelcontextprotocol/go-sdk/mcp"
|
||||
)
|
||||
|
||||
// requestLog captures HTTP requests made by MCP tool handlers.
|
||||
type requestLog struct {
|
||||
mu sync.Mutex
|
||||
requests []capturedRequest
|
||||
}
|
||||
|
||||
type capturedRequest struct {
|
||||
Method string
|
||||
Path string
|
||||
Query string
|
||||
Body string
|
||||
}
|
||||
|
||||
func (rl *requestLog) add(r capturedRequest) {
|
||||
rl.mu.Lock()
|
||||
defer rl.mu.Unlock()
|
||||
rl.requests = append(rl.requests, r)
|
||||
}
|
||||
|
||||
func (rl *requestLog) last() capturedRequest {
|
||||
rl.mu.Lock()
|
||||
defer rl.mu.Unlock()
|
||||
if len(rl.requests) == 0 {
|
||||
return capturedRequest{}
|
||||
}
|
||||
return rl.requests[len(rl.requests)-1]
|
||||
}
|
||||
|
||||
// mockCertctlAPI returns a test server that records all requests and returns
|
||||
// canned JSON responses based on the path.
|
||||
func mockCertctlAPI(log *requestLog) *httptest.Server {
|
||||
return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
body := ""
|
||||
if r.Body != nil {
|
||||
buf := make([]byte, 4096)
|
||||
n, _ := r.Body.Read(buf)
|
||||
body = string(buf[:n])
|
||||
}
|
||||
|
||||
log.add(capturedRequest{
|
||||
Method: r.Method,
|
||||
Path: r.URL.Path,
|
||||
Query: r.URL.RawQuery,
|
||||
Body: body,
|
||||
})
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
|
||||
switch {
|
||||
case r.Method == "DELETE":
|
||||
w.WriteHeader(http.StatusNoContent)
|
||||
case strings.HasSuffix(r.URL.Path, "/renew") || strings.HasSuffix(r.URL.Path, "/deploy"):
|
||||
w.WriteHeader(http.StatusAccepted)
|
||||
json.NewEncoder(w).Encode(map[string]string{"status": "accepted", "job_id": "job-001"})
|
||||
case r.Method == "POST":
|
||||
w.WriteHeader(http.StatusCreated)
|
||||
json.NewEncoder(w).Encode(map[string]string{"id": "new-resource"})
|
||||
default:
|
||||
w.WriteHeader(http.StatusOK)
|
||||
json.NewEncoder(w).Encode(map[string]interface{}{
|
||||
"data": []interface{}{map[string]string{"id": "test-1"}},
|
||||
"total": 1,
|
||||
})
|
||||
}
|
||||
}))
|
||||
}
|
||||
|
||||
func TestRegisterTools_ToolCount(t *testing.T) {
|
||||
server := gomcp.NewServer(&gomcp.Implementation{
|
||||
Name: "certctl-test",
|
||||
Version: "test",
|
||||
}, nil)
|
||||
|
||||
log := &requestLog{}
|
||||
api := mockCertctlAPI(log)
|
||||
defer api.Close()
|
||||
|
||||
client := NewClient(api.URL, "test-key")
|
||||
RegisterTools(server, client)
|
||||
|
||||
// The server should have tools registered — we can verify by listing them
|
||||
// Since the SDK doesn't expose a tool count method, we verify through the
|
||||
// request capabilities
|
||||
t.Log("RegisterTools completed without panic")
|
||||
}
|
||||
|
||||
func TestPaginationQuery(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
page int
|
||||
perPage int
|
||||
wantLen int
|
||||
}{
|
||||
{"both set", 2, 50, 2},
|
||||
{"page only", 3, 0, 1},
|
||||
{"per_page only", 0, 100, 1},
|
||||
{"neither set", 0, 0, 0},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
q := paginationQuery(tt.page, tt.perPage)
|
||||
if len(q) != tt.wantLen {
|
||||
t.Errorf("expected %d query params, got %d", tt.wantLen, len(q))
|
||||
}
|
||||
if tt.page > 0 {
|
||||
if q.Get("page") != string(rune('0'+tt.page)) && q.Get("page") == "" {
|
||||
t.Errorf("expected page param to be set")
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestTextResult(t *testing.T) {
|
||||
data := json.RawMessage(`{"id":"mc-test","status":"Active"}`)
|
||||
result, metadata, err := textResult(data)
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if metadata != nil {
|
||||
t.Errorf("expected nil metadata, got %v", metadata)
|
||||
}
|
||||
if result == nil {
|
||||
t.Fatal("expected non-nil result")
|
||||
}
|
||||
if len(result.Content) != 1 {
|
||||
t.Fatalf("expected 1 content item, got %d", len(result.Content))
|
||||
}
|
||||
tc, ok := result.Content[0].(*gomcp.TextContent)
|
||||
if !ok {
|
||||
t.Fatal("expected TextContent type")
|
||||
}
|
||||
if tc.Text != `{"id":"mc-test","status":"Active"}` {
|
||||
t.Errorf("unexpected text content: %s", tc.Text)
|
||||
}
|
||||
}
|
||||
|
||||
func TestErrorResult(t *testing.T) {
|
||||
result, _, err := errorResult(http.ErrServerClosed)
|
||||
if result != nil {
|
||||
t.Errorf("expected nil result, got %v", result)
|
||||
}
|
||||
if err == nil {
|
||||
t.Fatal("expected non-nil error")
|
||||
}
|
||||
}
|
||||
|
||||
// TestToolEndToEnd_ListCertificates verifies the full flow:
|
||||
// MCP tool handler → HTTP client → mock API → response formatting
|
||||
func TestToolEndToEnd_ListCertificates(t *testing.T) {
|
||||
log := &requestLog{}
|
||||
api := mockCertctlAPI(log)
|
||||
defer api.Close()
|
||||
|
||||
client := NewClient(api.URL, "test-key")
|
||||
|
||||
// Manually call the handler logic that would be registered as a tool
|
||||
q := paginationQuery(1, 50)
|
||||
q.Set("status", "Active")
|
||||
data, err := client.Get("/api/v1/certificates", q)
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
|
||||
req := log.last()
|
||||
if req.Method != "GET" {
|
||||
t.Errorf("expected GET, got %s", req.Method)
|
||||
}
|
||||
if req.Path != "/api/v1/certificates" {
|
||||
t.Errorf("expected path /api/v1/certificates, got %s", req.Path)
|
||||
}
|
||||
if !strings.Contains(req.Query, "status=Active") {
|
||||
t.Errorf("expected status=Active in query, got %s", req.Query)
|
||||
}
|
||||
if !strings.Contains(req.Query, "page=1") {
|
||||
t.Errorf("expected page=1 in query, got %s", req.Query)
|
||||
}
|
||||
|
||||
result, _, err := textResult(data)
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error formatting result: %v", err)
|
||||
}
|
||||
if len(result.Content) != 1 {
|
||||
t.Fatalf("expected 1 content item, got %d", len(result.Content))
|
||||
}
|
||||
}
|
||||
|
||||
func TestToolEndToEnd_CreateCertificate(t *testing.T) {
|
||||
log := &requestLog{}
|
||||
api := mockCertctlAPI(log)
|
||||
defer api.Close()
|
||||
|
||||
client := NewClient(api.URL, "test-key")
|
||||
|
||||
input := CreateCertificateInput{
|
||||
Name: "API Production",
|
||||
CommonName: "api.example.com",
|
||||
IssuerID: "iss-local",
|
||||
OwnerID: "o-alice",
|
||||
TeamID: "team-platform",
|
||||
}
|
||||
|
||||
data, err := client.Post("/api/v1/certificates", input)
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
|
||||
req := log.last()
|
||||
if req.Method != "POST" {
|
||||
t.Errorf("expected POST, got %s", req.Method)
|
||||
}
|
||||
if req.Path != "/api/v1/certificates" {
|
||||
t.Errorf("expected path /api/v1/certificates, got %s", req.Path)
|
||||
}
|
||||
if !strings.Contains(req.Body, "api.example.com") {
|
||||
t.Errorf("expected common_name in body, got %s", req.Body)
|
||||
}
|
||||
|
||||
var result map[string]string
|
||||
if err := json.Unmarshal(data, &result); err != nil {
|
||||
t.Fatalf("failed to parse response: %v", err)
|
||||
}
|
||||
if result["id"] != "new-resource" {
|
||||
t.Errorf("expected id=new-resource, got %s", result["id"])
|
||||
}
|
||||
}
|
||||
|
||||
func TestToolEndToEnd_TriggerRenewal(t *testing.T) {
|
||||
log := &requestLog{}
|
||||
api := mockCertctlAPI(log)
|
||||
defer api.Close()
|
||||
|
||||
client := NewClient(api.URL, "test-key")
|
||||
data, err := client.Post("/api/v1/certificates/mc-api-prod/renew", nil)
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
|
||||
req := log.last()
|
||||
if req.Method != "POST" {
|
||||
t.Errorf("expected POST, got %s", req.Method)
|
||||
}
|
||||
if req.Path != "/api/v1/certificates/mc-api-prod/renew" {
|
||||
t.Errorf("expected path /api/v1/certificates/mc-api-prod/renew, got %s", req.Path)
|
||||
}
|
||||
|
||||
var result map[string]string
|
||||
if err := json.Unmarshal(data, &result); err != nil {
|
||||
t.Fatalf("failed to parse response: %v", err)
|
||||
}
|
||||
if result["job_id"] != "job-001" {
|
||||
t.Errorf("expected job_id=job-001, got %s", result["job_id"])
|
||||
}
|
||||
}
|
||||
|
||||
func TestToolEndToEnd_DeleteTarget(t *testing.T) {
|
||||
log := &requestLog{}
|
||||
api := mockCertctlAPI(log)
|
||||
defer api.Close()
|
||||
|
||||
client := NewClient(api.URL, "test-key")
|
||||
data, err := client.Delete("/api/v1/targets/t-platform")
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
|
||||
req := log.last()
|
||||
if req.Method != "DELETE" {
|
||||
t.Errorf("expected DELETE, got %s", req.Method)
|
||||
}
|
||||
if req.Path != "/api/v1/targets/t-platform" {
|
||||
t.Errorf("expected path /api/v1/targets/t-platform, got %s", req.Path)
|
||||
}
|
||||
|
||||
var result map[string]string
|
||||
if err := json.Unmarshal(data, &result); err != nil {
|
||||
t.Fatalf("failed to parse response: %v", err)
|
||||
}
|
||||
if result["status"] != "deleted" {
|
||||
t.Errorf("expected status=deleted, got %s", result["status"])
|
||||
}
|
||||
}
|
||||
|
||||
func TestToolEndToEnd_RevokeCertificate(t *testing.T) {
|
||||
log := &requestLog{}
|
||||
api := mockCertctlAPI(log)
|
||||
defer api.Close()
|
||||
|
||||
client := NewClient(api.URL, "test-key")
|
||||
input := RevokeCertificateInput{
|
||||
ID: "mc-api-prod",
|
||||
Reason: "keyCompromise",
|
||||
}
|
||||
_, err := client.Post("/api/v1/certificates/"+input.ID+"/revoke", map[string]string{"reason": input.Reason})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
|
||||
req := log.last()
|
||||
if req.Method != "POST" {
|
||||
t.Errorf("expected POST, got %s", req.Method)
|
||||
}
|
||||
if req.Path != "/api/v1/certificates/mc-api-prod/revoke" {
|
||||
t.Errorf("expected path /api/v1/certificates/mc-api-prod/revoke, got %s", req.Path)
|
||||
}
|
||||
if !strings.Contains(req.Body, "keyCompromise") {
|
||||
t.Errorf("expected reason in body, got %s", req.Body)
|
||||
}
|
||||
}
|
||||
|
||||
func TestToolEndToEnd_AgentHeartbeat(t *testing.T) {
|
||||
log := &requestLog{}
|
||||
api := mockCertctlAPI(log)
|
||||
defer api.Close()
|
||||
|
||||
client := NewClient(api.URL, "test-key")
|
||||
_, err := client.Post("/api/v1/agents/agent-001/heartbeat", map[string]string{
|
||||
"os": "linux",
|
||||
"architecture": "amd64",
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
|
||||
req := log.last()
|
||||
if req.Path != "/api/v1/agents/agent-001/heartbeat" {
|
||||
t.Errorf("expected path /api/v1/agents/agent-001/heartbeat, got %s", req.Path)
|
||||
}
|
||||
}
|
||||
|
||||
func TestToolEndToEnd_ListWithFilters(t *testing.T) {
|
||||
log := &requestLog{}
|
||||
api := mockCertctlAPI(log)
|
||||
defer api.Close()
|
||||
|
||||
client := NewClient(api.URL, "test-key")
|
||||
q := paginationQuery(1, 25)
|
||||
q.Set("status", "Pending")
|
||||
q.Set("type", "Renewal")
|
||||
_, err := client.Get("/api/v1/jobs", q)
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
|
||||
req := log.last()
|
||||
if req.Path != "/api/v1/jobs" {
|
||||
t.Errorf("expected path /api/v1/jobs, got %s", req.Path)
|
||||
}
|
||||
if !strings.Contains(req.Query, "status=Pending") {
|
||||
t.Errorf("expected status filter in query, got %s", req.Query)
|
||||
}
|
||||
if !strings.Contains(req.Query, "type=Renewal") {
|
||||
t.Errorf("expected type filter in query, got %s", req.Query)
|
||||
}
|
||||
}
|
||||
|
||||
func TestToolEndToEnd_GetRawBinary(t *testing.T) {
|
||||
derData := []byte{0x30, 0x82, 0x01, 0x22}
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/pkix-crl")
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write(derData)
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
client := NewClient(server.URL, "test-key")
|
||||
data, ct, err := client.GetRaw("/api/v1/crl/iss-local")
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if ct != "application/pkix-crl" {
|
||||
t.Errorf("expected content-type application/pkix-crl, got %s", ct)
|
||||
}
|
||||
if len(data) != 4 {
|
||||
t.Errorf("expected 4 bytes, got %d", len(data))
|
||||
}
|
||||
}
|
||||
|
||||
func TestToolEndToEnd_ErrorPropagation(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusForbidden)
|
||||
w.Write([]byte(`{"error":"forbidden"}`))
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
client := NewClient(server.URL, "test-key")
|
||||
_, err := client.Get("/api/v1/certificates", nil)
|
||||
if err == nil {
|
||||
t.Fatal("expected error for 403 response")
|
||||
}
|
||||
result, _, toolErr := errorResult(err)
|
||||
if result != nil {
|
||||
t.Errorf("expected nil result from errorResult")
|
||||
}
|
||||
if toolErr == nil {
|
||||
t.Fatal("expected non-nil error from errorResult")
|
||||
}
|
||||
}
|
||||
@@ -31,8 +31,8 @@ type CreateCertificateInput struct {
|
||||
CommonName string `json:"common_name" jsonschema:"Certificate common name (e.g. api.example.com)"`
|
||||
SANs []string `json:"sans,omitempty" jsonschema:"Subject Alternative Names"`
|
||||
Environment string `json:"environment,omitempty" jsonschema:"Environment (e.g. production, staging)"`
|
||||
OwnerID string `json:"owner_id,omitempty" jsonschema:"Owner ID"`
|
||||
TeamID string `json:"team_id,omitempty" jsonschema:"Team ID"`
|
||||
OwnerID string `json:"owner_id" jsonschema:"Owner ID (required)"`
|
||||
TeamID string `json:"team_id" jsonschema:"Team ID (required)"`
|
||||
IssuerID string `json:"issuer_id" jsonschema:"Issuer connector ID"`
|
||||
TargetIDs []string `json:"target_ids,omitempty" jsonschema:"Deployment target IDs"`
|
||||
RenewalPolicyID string `json:"renewal_policy_id,omitempty" jsonschema:"Renewal policy ID"`
|
||||
|
||||
@@ -439,7 +439,7 @@ func (s *AgentService) GetWorkWithTargets(agentID string) ([]domain.WorkItem, er
|
||||
// Enrich with target details for deployment jobs
|
||||
if j.TargetID != nil && *j.TargetID != "" {
|
||||
target, err := s.targetRepo.Get(context.Background(), *j.TargetID)
|
||||
if err == nil {
|
||||
if err == nil && target != nil {
|
||||
item.TargetType = string(target.Type)
|
||||
item.TargetConfig = target.Config
|
||||
}
|
||||
@@ -448,7 +448,7 @@ func (s *AgentService) GetWorkWithTargets(agentID string) ([]domain.WorkItem, er
|
||||
// Enrich with certificate details for AwaitingCSR jobs (agent needs CN + SANs for CSR)
|
||||
if j.Status == domain.JobStatusAwaitingCSR {
|
||||
cert, err := s.certRepo.Get(context.Background(), j.CertificateID)
|
||||
if err == nil {
|
||||
if err == nil && cert != nil {
|
||||
item.CommonName = cert.CommonName
|
||||
item.SANs = cert.SANs
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user