fix: 端启停改为仅由 identity 执行

This commit is contained in:
Nixevol
2026-09-30 16:22:48 +08:00
parent 1a4bf6f185
commit ac08b6c071
3 changed files with 161 additions and 1 deletions
+6 -1
View File
@@ -350,8 +350,13 @@ func (h *Handler) handleEndpointPatch(w http.ResponseWriter, r *http.Request) {
hasMeta := req.Name != nil || req.Remark != nil || req.DefaultDelaySeconds != nil
var wasEnabled bool
var err error
// 注入 Identity 时启停只走 identity,避免先写 enabled 再级联失败造成半生效。
patchEnabled := req.Enabled
if h.identity != nil {
patchEnabled = nil
}
if hasMeta || (req.Enabled != nil && h.identity == nil) {
wasEnabled, err = h.patchEndpoint(r.Context(), id, req.Name, req.Remark, req.DefaultDelaySeconds, req.Enabled)
wasEnabled, err = h.patchEndpoint(r.Context(), id, req.Name, req.Remark, req.DefaultDelaySeconds, patchEnabled)
if err != nil {
if errors.Is(err, sql.ErrNoRows) {
h.audit(actorString(p), "endpoint_patch", id, "not_found", ip)
+146
View File
@@ -0,0 +1,146 @@
package admin_test
import (
"context"
"errors"
"net/http"
"sync"
"testing"
"git.asio.asia/nixevol/NixMsg/internal/admin"
"git.asio.asia/nixevol/NixMsg/internal/app/identity"
"git.asio.asia/nixevol/NixMsg/internal/auth"
"git.asio.asia/nixevol/NixMsg/internal/store"
"net/http/cookiejar"
"net/http/httptest"
"path/filepath"
)
type failDisableIdentity struct {
identity.Stub
err error
}
func (f *failDisableIdentity) Disable(context.Context, string) error { return f.err }
func (f *failDisableIdentity) Enable(context.Context, string) error { return nil }
type trackIdentity struct {
identity.Stub
mu sync.Mutex
disableN int
enableN int
}
func (t *trackIdentity) Disable(context.Context, string) error {
t.mu.Lock()
defer t.mu.Unlock()
t.disableN++
return nil
}
func (t *trackIdentity) Enable(context.Context, string) error {
t.mu.Lock()
defer t.mu.Unlock()
t.enableN++
return nil
}
func setupEndpointsWithIdentity(t *testing.T, ident identity.Service) (*store.DB, *httptest.Server, *http.Client) {
t.Helper()
dir := t.TempDir()
db, err := store.Open(filepath.Join(dir, "data"), "FULL")
if err != nil {
t.Fatal(err)
}
t.Cleanup(func() { _ = db.Close() })
hash := auth.NewStubHashPool()
if seedErr := admin.SeedAdminPassword(context.Background(), db, hash, testPassword); seedErr != nil {
t.Fatal(seedErr)
}
h := admin.New(admin.Deps{
DB: db,
Hash: hash,
Tokens: admin.NewRandomAPITokens(),
Locks: admin.NewMemoryLoginLocks(),
Identity: ident,
})
srv := httptest.NewServer(h)
t.Cleanup(srv.Close)
jar, err := cookiejar.New(nil)
if err != nil {
t.Fatal(err)
}
client := &http.Client{Jar: jar}
res := postJSON(t, client, srv.URL+"/api/admin/login",
`{"username":"admin","password":"`+testPassword+`"}`, nil)
env := decodeEnv(t, res)
if res.StatusCode != http.StatusOK || !env.OK {
t.Fatalf("login failed: %d %+v", res.StatusCode, env)
}
return db, srv, client
}
func TestPatchEnabledFalseKeepsEnabledWhenIdentityFails(t *testing.T) {
ident := &failDisableIdentity{err: errors.New("disable failed")}
db, srv, client := setupEndpointsWithIdentity(t, ident)
base := srv.URL
res := postJSON(t, client, base+"/api/admin/endpoints",
`{"id":"keep-on","name":"仍启用","login_password":"password1"}`,
csrfHeaders())
env := decodeEnv(t, res)
if res.StatusCode != 200 || !env.OK {
t.Fatalf("create: %d %+v", res.StatusCode, env)
}
res = doReq(t, client, http.MethodPatch, base+"/api/admin/endpoints/keep-on",
`{"enabled":false}`, csrfHeaders())
env = decodeEnv(t, res)
if res.StatusCode != http.StatusInternalServerError {
t.Fatalf("want 500 got %d %+v", res.StatusCode, env)
}
var enabled int
if err := db.Read.QueryRow(`SELECT enabled FROM endpoints WHERE id='keep-on'`).Scan(&enabled); err != nil {
t.Fatal(err)
}
if enabled != 1 {
t.Fatalf("want still enabled, got %d", enabled)
}
}
func TestPatchNameOnlyDoesNotTouchEnabled(t *testing.T) {
ident := &trackIdentity{}
db, srv, client := setupEndpointsWithIdentity(t, ident)
base := srv.URL
res := postJSON(t, client, base+"/api/admin/endpoints",
`{"id":"name-only","name":"旧名","login_password":"password1"}`,
csrfHeaders())
env := decodeEnv(t, res)
if res.StatusCode != 200 || !env.OK {
t.Fatalf("create: %d %+v", res.StatusCode, env)
}
res = doReq(t, client, http.MethodPatch, base+"/api/admin/endpoints/name-only",
`{"name":"新名"}`, csrfHeaders())
env = decodeEnv(t, res)
if res.StatusCode != 200 || !env.OK {
t.Fatalf("patch: %d %+v", res.StatusCode, env)
}
ident.mu.Lock()
d, e := ident.disableN, ident.enableN
ident.mu.Unlock()
if d != 0 || e != 0 {
t.Fatalf("identity enable/disable should not run, disable=%d enable=%d", d, e)
}
var enabled int
var name string
if err := db.Read.QueryRow(`SELECT enabled, name FROM endpoints WHERE id='name-only'`).Scan(&enabled, &name); err != nil {
t.Fatal(err)
}
if enabled != 1 || name != "新名" {
t.Fatalf("enabled=%d name=%q", enabled, name)
}
}