feat: add support for release channels in version upgrade and node agent updates

- Introduced ReleaseChannel type to manage stable and preview releases. - Updated DashboardTopbar to handle version upgrades based on selected release channel. - Enhanced node detail page to allow manual checks for agent updates on stable and preview channels. - Modified API endpoints to support fetching and upgrading based on release channels. - Updated UI components to reflect changes in version checking and upgrade processes. - Added tests for new functionality related to preview releases and agent updates.
2026-10-06 15:46:37 +08:00 · 2026-03-12 11:25:13 +08:00
parent 93e43fb3b0
commit f50eb9adee
25 changed files with 1448 additions and 239 deletions
@@ -0,0 +1,117 @@
+package service
+
+import (
+	"atsflare/common"
+	"atsflare/model"
+	"io"
+	"net/http"
+	"strings"
+	"testing"
+)
+
+type roundTripFunc func(req *http.Request) (*http.Response, error)
+
+func (f roundTripFunc) RoundTrip(req *http.Request) (*http.Response, error) {
+	return f(req)
+}
+
+func TestRequestNodeAgentPreviewUpdate(t *testing.T) {
+	setupServiceTestDB(t)
+
+	node, err := CreateNode(NodeInput{Name: "preview-edge-1"})
+	if err != nil {
+		t.Fatalf("failed to create node: %v", err)
+	}
+
+	originalClient := UpdateHTTPClientForTest()
+	SetUpdateHTTPClientForTest(&http.Client{
+		Transport: roundTripFunc(func(req *http.Request) (*http.Response, error) {
+			if req.URL.String() != "https://api.github.com/repos/"+common.AgentUpdateRepo+"/releases/tags/v0.5.0-rc.1" {
+				t.Fatalf("unexpected request url: %s", req.URL.String())
+			}
+			return &http.Response{
+				StatusCode: http.StatusOK,
+				Header:     make(http.Header),
+				Body:       io.NopCloser(strings.NewReader(`{"tag_name":"v0.5.0-rc.1","prerelease":true}`)),
+			}, nil
+		}),
+	})
+	t.Cleanup(func() {
+		SetUpdateHTTPClientForTest(originalClient)
+	})
+
+	updated, err := RequestNodeAgentUpdate(node.ID, NodeAgentUpdateInput{
+		Channel: "preview",
+		TagName: "v0.5.0-rc.1",
+	})
+	if err != nil {
+		t.Fatalf("expected preview update request to succeed: %v", err)
+	}
+	if !updated.UpdateRequested {
+		t.Fatal("expected update_requested to be true")
+	}
+	if updated.UpdateChannel != "preview" {
+		t.Fatalf("unexpected update channel: %s", updated.UpdateChannel)
+	}
+	if updated.UpdateTag != "v0.5.0-rc.1" {
+		t.Fatalf("unexpected update tag: %s", updated.UpdateTag)
+	}
+}
+
+func TestHeartbeatNodeReturnsPreviewUpdateSettings(t *testing.T) {
+	setupServiceTestDB(t)
+
+	node := &model.Node{
+		NodeID:            "node-preview-1",
+		Name:              "preview-edge-1",
+		IP:                "10.0.0.8",
+		AgentToken:        "agent-token",
+		AgentVersion:      "v0.4.0",
+		NginxVersion:      "1.27.1.2",
+		Status:            NodeStatusOnline,
+		UpdateRequested:   true,
+		UpdateChannel:     "preview",
+		UpdateTag:         "v0.5.0-rc.1",
+		AutoUpdateEnabled: false,
+	}
+	if err := node.Insert(); err != nil {
+		t.Fatalf("failed to seed node: %v", err)
+	}
+
+	resp, err := HeartbeatNode(node, AgentNodePayload{
+		NodeID:       node.NodeID,
+		Name:         node.Name,
+		IP:           node.IP,
+		AgentVersion: node.AgentVersion,
+		NginxVersion: node.NginxVersion,
+	})
+	if err != nil {
+		t.Fatalf("expected heartbeat to succeed: %v", err)
+	}
+	if resp.AgentSettings == nil {
+		t.Fatal("expected agent settings in heartbeat response")
+	}
+	if !resp.AgentSettings.UpdateNow {
+		t.Fatal("expected update_now to be true")
+	}
+	if resp.AgentSettings.UpdateChannel != "preview" {
+		t.Fatalf("unexpected update channel: %s", resp.AgentSettings.UpdateChannel)
+	}
+	if resp.AgentSettings.UpdateTag != "v0.5.0-rc.1" {
+		t.Fatalf("unexpected update tag: %s", resp.AgentSettings.UpdateTag)
+	}
+
+	storedNode, err := model.GetNodeByID(node.ID)
+	if err != nil {
+		t.Fatalf("failed to reload node: %v", err)
+	}
+	if storedNode.UpdateRequested {
+		t.Fatal("expected update_requested to be reset after heartbeat")
+	}
+	if storedNode.UpdateChannel != "stable" {
+		t.Fatalf("expected update channel to reset to stable, got %s", storedNode.UpdateChannel)
+	}
+	if storedNode.UpdateTag != "" {
+		t.Fatalf("expected update tag to be cleared, got %s", storedNode.UpdateTag)
+	}
+}