188 lines
7.8 KiB
Go
188 lines
7.8 KiB
Go
package httpapi
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"encoding/json"
|
|
"errors"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"git.pcux.cn/xuans/Scert/backend/internal/buildinfo"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/certificate"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/config"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/deployment"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/domain"
|
|
)
|
|
|
|
type stubDomainRepository struct {
|
|
items []domain.Domain
|
|
created domain.Domain
|
|
loaded domain.Domain
|
|
updated domain.Domain
|
|
listErr error
|
|
createErr error
|
|
getErr error
|
|
updateErr error
|
|
deleteErr error
|
|
stateErr error
|
|
}
|
|
|
|
func (r *stubDomainRepository) UpdateCertificateState(_ context.Context, id, status string, expiresAt *time.Time) error {
|
|
return r.stateErr
|
|
}
|
|
|
|
func (r *stubDomainRepository) Create(_ context.Context, item domain.Domain) error {
|
|
r.created = item
|
|
return r.createErr
|
|
}
|
|
|
|
func (r *stubDomainRepository) GetByID(_ context.Context, _ string) (domain.Domain, error) {
|
|
if r.getErr != nil {
|
|
return domain.Domain{}, r.getErr
|
|
}
|
|
if r.loaded.ID == "" {
|
|
return domain.Domain{}, sql.ErrNoRows
|
|
}
|
|
return r.loaded, nil
|
|
}
|
|
|
|
func (r *stubDomainRepository) List(_ context.Context) ([]domain.Domain, error) {
|
|
return r.items, r.listErr
|
|
}
|
|
|
|
func (r *stubDomainRepository) Delete(_ context.Context, _ string) error {
|
|
return r.deleteErr
|
|
}
|
|
|
|
func (r *stubDomainRepository) Update(_ context.Context, item domain.Domain) error {
|
|
r.updated = item
|
|
return r.updateErr
|
|
}
|
|
|
|
func TestStatusReturnsApplicationAndConfigVersions(t *testing.T) {
|
|
cfg := readyTestConfig(t.TempDir())
|
|
server := New(Dependencies{Config: cfg, Repository: &stubDomainRepository{}})
|
|
response := httptest.NewRecorder()
|
|
serveAdminHTTP(server, response, httptest.NewRequest(http.MethodGet, "/api/v1/status", nil))
|
|
if response.Code != http.StatusOK {
|
|
t.Fatalf("状态接口响应错误：%d", response.Code)
|
|
}
|
|
var payload struct {
|
|
Version string `json:"version"`
|
|
ConfigVersion int `json:"config_version"`
|
|
}
|
|
if err := json.Unmarshal(response.Body.Bytes(), &payload); err != nil {
|
|
t.Fatalf("解析状态响应失败：%v", err)
|
|
}
|
|
if payload.Version != buildinfo.Version || payload.ConfigVersion != config.CurrentVersion {
|
|
t.Fatalf("状态版本字段不正确：%+v", payload)
|
|
}
|
|
}
|
|
|
|
func TestDomainAPIValidatesSANAndHidesInternalErrors(t *testing.T) {
|
|
cfg := readyTestConfig(t.TempDir())
|
|
repo := &stubDomainRepository{}
|
|
server := New(Dependencies{Config: cfg, Repository: repo})
|
|
|
|
invalid := httptest.NewRequest(http.MethodPost, "/api/v1/domains", strings.NewReader(`{"name":"example.com","certificate_type":"san","sans":["bad_domain"]}`))
|
|
invalid.Header.Set("Content-Type", "application/json")
|
|
invalidResponse := httptest.NewRecorder()
|
|
serveAdminHTTP(server, invalidResponse, invalid)
|
|
if invalidResponse.Code != http.StatusBadRequest || strings.Contains(invalidResponse.Body.String(), "detail") {
|
|
t.Fatalf("非法 SAN 响应不正确：%d %s", invalidResponse.Code, invalidResponse.Body.String())
|
|
}
|
|
|
|
valid := httptest.NewRequest(http.MethodPost, "/api/v1/domains", strings.NewReader(`{"name":"example.com","certificate_type":"san","sans":["www.example.com"]}`))
|
|
valid.Header.Set("Content-Type", "application/json")
|
|
validResponse := httptest.NewRecorder()
|
|
serveAdminHTTP(server, validResponse, valid)
|
|
if validResponse.Code != http.StatusCreated || len(repo.created.SANs) != 2 || repo.created.SANs[0] != "example.com" {
|
|
t.Fatalf("合法 SAN 响应不正确：%d %s", validResponse.Code, validResponse.Body.String())
|
|
}
|
|
|
|
repo.listErr = errors.New("internal-dsn-secret")
|
|
listResponse := httptest.NewRecorder()
|
|
serveAdminHTTP(server, listResponse, httptest.NewRequest(http.MethodGet, "/api/v1/domains", nil))
|
|
if listResponse.Code != http.StatusInternalServerError {
|
|
t.Fatalf("列表错误状态码不正确：%d", listResponse.Code)
|
|
}
|
|
if strings.Contains(listResponse.Body.String(), "internal-dsn-secret") || strings.Contains(listResponse.Body.String(), "detail") {
|
|
t.Fatalf("接口泄露了底层错误：%s", listResponse.Body.String())
|
|
}
|
|
}
|
|
|
|
func TestDeleteMissingDomainReturnsNotFound(t *testing.T) {
|
|
cfg := readyTestConfig(t.TempDir())
|
|
repo := &stubDomainRepository{deleteErr: sql.ErrNoRows}
|
|
server := New(Dependencies{Config: cfg, Repository: repo})
|
|
response := httptest.NewRecorder()
|
|
serveAdminHTTP(server, response, httptest.NewRequest(http.MethodDelete, "/api/v1/domains/missing", nil))
|
|
if response.Code != http.StatusNotFound || strings.Contains(response.Body.String(), "detail") {
|
|
t.Fatalf("删除不存在域名响应不正确：%d %s", response.Code, response.Body.String())
|
|
}
|
|
}
|
|
|
|
func TestUpdateDomainAPI(t *testing.T) {
|
|
cfg := readyTestConfig(t.TempDir())
|
|
loaded := domain.Domain{ID: "domain-1", Name: "example.com", CertificateType: domain.CertificateSingle, SANs: []string{"example.com"}, Status: "active"}
|
|
repo := &stubDomainRepository{loaded: loaded}
|
|
server := New(Dependencies{Config: cfg, Repository: repo})
|
|
request := httptest.NewRequest(http.MethodPut, "/api/v1/domains/domain-1", strings.NewReader(`{"name":"example.com","certificate_type":"san","sans":["www.example.com"],"auto_renew":false,"acme_environment":"production","dns_provider":"cloudflare"}`))
|
|
request.Header.Set("Content-Type", "application/json")
|
|
response := httptest.NewRecorder()
|
|
serveAdminHTTP(server, response, request)
|
|
if response.Code != http.StatusOK {
|
|
t.Fatalf("更新域名响应错误：%d %s", response.Code, response.Body.String())
|
|
}
|
|
if repo.updated.ID != loaded.ID || repo.updated.Status != "pending" || repo.updated.AutoRenew || repo.updated.ACMEEnvironment != domain.ACMEEnvironmentProduction || len(repo.updated.SANs) != 2 {
|
|
t.Fatalf("更新域名参数不正确：%+v", repo.updated)
|
|
}
|
|
}
|
|
|
|
func TestCreateDomainConflictReturnsConflict(t *testing.T) {
|
|
cfg := readyTestConfig(t.TempDir())
|
|
repo := &stubDomainRepository{createErr: errors.New("UNIQUE constraint failed: domains.name")}
|
|
server := New(Dependencies{Config: cfg, Repository: repo})
|
|
request := httptest.NewRequest(http.MethodPost, "/api/v1/domains", strings.NewReader(`{"name":"example.com","certificate_type":"single"}`))
|
|
request.Header.Set("Content-Type", "application/json")
|
|
response := httptest.NewRecorder()
|
|
serveAdminHTTP(server, response, request)
|
|
if response.Code != http.StatusConflict {
|
|
t.Fatalf("重复域名响应错误：%d %s", response.Code, response.Body.String())
|
|
}
|
|
}
|
|
|
|
func TestUpdateDomainRejectsInvalidEnvironment(t *testing.T) {
|
|
cfg := readyTestConfig(t.TempDir())
|
|
loaded := domain.Domain{ID: "domain-1", Name: "example.com"}
|
|
server := New(Dependencies{Config: cfg, Repository: &stubDomainRepository{loaded: loaded}})
|
|
request := httptest.NewRequest(http.MethodPut, "/api/v1/domains/domain-1", strings.NewReader(`{"name":"example.com","certificate_type":"single","acme_environment":"invalid"}`))
|
|
request.Header.Set("Content-Type", "application/json")
|
|
response := httptest.NewRecorder()
|
|
serveAdminHTTP(server, response, request)
|
|
if response.Code != http.StatusBadRequest {
|
|
t.Fatalf("非法 ACME 环境响应错误：%d %s", response.Code, response.Body.String())
|
|
}
|
|
}
|
|
|
|
func TestIssueDomainUsesCertificateWorkflow(t *testing.T) {
|
|
cfg := readyTestConfig(t.TempDir())
|
|
calledID := ""
|
|
server := New(Dependencies{
|
|
Config: cfg, Repository: &stubDomainRepository{},
|
|
IssueCertificate: func(_ context.Context, id string) (certificate.IssueResult, deployment.Report, error) {
|
|
calledID = id
|
|
return certificate.IssueResult{DomainID: id}, deployment.Report{Status: "success"}, nil
|
|
},
|
|
})
|
|
response := httptest.NewRecorder()
|
|
serveAdminHTTP(server, response, httptest.NewRequest(http.MethodPost, "/api/v1/domains/domain-1/issue", nil))
|
|
if response.Code != http.StatusOK || calledID != "domain-1" || !strings.Contains(response.Body.String(), `"status":"success"`) {
|
|
t.Fatalf("证书申请接口响应错误：%d %s", response.Code, response.Body.String())
|
|
}
|
|
}
|