Initial commit: DSH Web Manager

2026-10-13 09:34:47 +08:00
commit 85332bbc56
135 changed files with 19712 additions and 0 deletions
@@ -0,0 +1,158 @@
+package api
+
+import (
+	"context"
+	"git.misaka.ren/m1saka/dsh-web/internal/store"
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"strings"
+	"testing"
+)
+
+func TestIntentsReserveCancel(t *testing.T) {
+	a, sid := runtimeTestAPI(t, func(w http.ResponseWriter, r *http.Request) { t.Error("reservation contacted Docker") })
+	if err := a.initAuth(); err != nil {
+		t.Fatal(err)
+	}
+	for n := 0; n < 3; n++ {
+		w := httptest.NewRecorder()
+		a.intents(w, runtimeRequest("POST", "/api/intents", `{"name":"one","version":"1.2.3"}`, sid))
+		want := 202
+		if n == 2 {
+			want = 409
+		}
+		if w.Code != want {
+			t.Fatalf("%d: %d %s", n, w.Code, w.Body.String())
+		}
+	}
+	list, err := a.s.Intents("owner")
+	if err != nil || len(list) != 2 {
+		t.Fatalf("%v %v", list, err)
+	}
+	w := httptest.NewRecorder()
+	a.intent(w, runtimeRequest("DELETE", "/api/intents/"+list[0].ID, "", sid))
+	if w.Code != 200 {
+		t.Fatal(w.Code)
+	}
+	if err := a.s.ReserveIntent(store.Intent{ID: store.ID(), UID: "owner"}); err != nil {
+		t.Fatal(err)
+	}
+}
+
+func TestIntentBuildSuccessUsesRealRuntime(t *testing.T) {
+	if os.Geteuid() != 0 {
+		t.Skip("runtime filesystem ownership requires root")
+	}
+	image := "sha256:" + strings.Repeat("a", 64)
+	creates := 0
+	a, _ := runtimeTestAPI(t, func(w http.ResponseWriter, r *http.Request) {
+		switch {
+		case strings.Contains(r.URL.Path, "/images/"):
+			writeJSON(w, map[string]any{"Id": image, "Config": map[string]any{"Labels": map[string]string{"dsh-mgr.role": "runtime", "dsh-mgr.version": "1.2.3"}}})
+		case strings.HasSuffix(r.URL.Path, "/containers/create"):
+			creates++
+			writeJSON(w, map[string]string{"Id": "runtime-created"})
+		default:
+			w.WriteHeader(204)
+		}
+	})
+	seedVerifiedImage(t, a, image, "1.2.3")
+	if err := a.s.EnsureIntents(); err != nil {
+		t.Fatal(err)
+	}
+	var build string
+	if err := a.s.DB.QueryRow("SELECT id FROM image_builds WHERE version='1.2.3'").Scan(&build); err != nil {
+		t.Fatal(err)
+	}
+	in := store.Intent{ID: store.ID(), UID: "owner", Name: "one", Version: "1.2.3", Host: "panel.test"}
+	if err := a.s.ReserveIntent(in); err != nil {
+		t.Fatal(err)
+	}
+	if _, err := a.s.DB.Exec("UPDATE creation_intents SET build_id=? WHERE id=?", build, in.ID); err != nil {
+		t.Fatal(err)
+	}
+	a.processIntents(context.Background())
+	a.processIntents(context.Background())
+	list, err := a.s.Intents("owner")
+	if err != nil || len(list) != 1 || list[0].Status != "success" || creates != 1 {
+		t.Fatalf("%+v err=%v creates=%d", list, err, creates)
+	}
+	instances, err := a.s.Instances("owner")
+	if err != nil || len(instances) != 1 || instances[0].ImageID != image || instances[0].Version != "1.2.3" {
+		t.Fatalf("%+v %v", instances, err)
+	}
+}
+
+func TestIntentFailedBuildReleasesAndCancelPreservesBuild(t *testing.T) {
+	a, _ := runtimeTestAPI(t, func(w http.ResponseWriter, r *http.Request) { t.Error("failed intent contacted Docker") })
+	seedVerifiedImage(t, a, "sha256:"+strings.Repeat("b", 64), "1.2.3")
+	if err := a.s.EnsureIntents(); err != nil {
+		t.Fatal(err)
+	}
+	var build string
+	a.s.DB.QueryRow("SELECT id FROM image_builds LIMIT 1").Scan(&build)
+	for _, id := range []string{"cancel", "fail"} {
+		if err := a.s.ReserveIntent(store.Intent{ID: id, UID: "owner", Version: "1.2.3"}); err != nil {
+			t.Fatal(err)
+		}
+		a.s.DB.Exec("UPDATE creation_intents SET build_id=? WHERE id=?", build, id)
+	}
+	if err := a.s.CancelIntent("cancel", "owner"); err != nil {
+		t.Fatal(err)
+	}
+	var status string
+	a.s.DB.QueryRow("SELECT status FROM image_builds WHERE id=?", build).Scan(&status)
+	if status != "success" {
+		t.Fatal(status)
+	}
+	a.s.DB.Exec("UPDATE image_builds SET status='failed' WHERE id=?", build)
+	a.processIntents(context.Background())
+	for n := 0; n < 2; n++ {
+		if err := a.s.ReserveIntent(store.Intent{ID: store.ID(), UID: "owner"}); err != nil {
+			t.Fatal(err)
+		}
+	}
+}
+
+func TestIntentRestartCleansInterruptedCreationAndRetainsWaiting(t *testing.T) {
+	deletes := 0
+	a, _ := runtimeTestAPI(t, func(w http.ResponseWriter, r *http.Request) {
+		if r.Method == "DELETE" {
+			deletes++
+		}
+		w.WriteHeader(204)
+	})
+	if err := a.s.EnsureIntents(); err != nil {
+		t.Fatal(err)
+	}
+	for _, id := range []string{"waiting", "creating"} {
+		if err := a.s.ReserveIntent(store.Intent{ID: id, UID: "owner", Version: "1.2.3"}); err != nil {
+			t.Fatal(err)
+		}
+	}
+	if err := a.s.ClaimIntent("creating", store.Instance{ID: "reserved", UID: "owner", Version: "1.2.3", Status: "creating"}); err != nil {
+		t.Fatal(err)
+	}
+	if err := a.recoverIntents(context.Background()); err != nil {
+		t.Fatal(err)
+	}
+	if deletes != 2 {
+		t.Fatalf("cleanup calls=%d", deletes)
+	}
+	list, err := a.s.Intents("owner")
+	if err != nil {
+		t.Fatal(err)
+	}
+	for _, in := range list {
+		if in.ID == "waiting" && in.Status != "waiting" {
+			t.Fatal(in)
+		}
+		if in.ID == "creating" && in.Status != "failed" {
+			t.Fatal(in)
+		}
+	}
+	if err := a.s.ReserveIntent(store.Intent{ID: store.ID(), UID: "owner"}); err != nil {
+		t.Fatal(err)
+	}
+}