package service

import (
	"testing"

	"github.com/mhsanaei/3x-ui/v3/internal/database"
	"github.com/mhsanaei/3x-ui/v3/internal/database/model"
)

// V2.1.0: the node form's probeOnly switch persists onto the bound
// agent rows via UpdateFromRequest (nil leaves the flag untouched),
// and the list view overlays it from the binding.
func TestNodeProbeOnlyUpdateAndView(t *testing.T) {
	setupConflictDB(t)
	db := database.GetDB()

	node := &model.Node{Id: 7601, Name: "probe-svc", Enable: true, Status: "online"}
	if err := db.Create(node).Error; err != nil {
		t.Fatalf("create node: %v", err)
	}
	ag := &model.AgentNode{TokenHash: "hash-probe-svc", Name: "probe-svc", NodeID: &node.Id}
	if err := db.Create(ag).Error; err != nil {
		t.Fatalf("create agent: %v", err)
	}

	svc := &NodeService{}
	truth := true
	lies := false
	base := func() *NodeMutationRequest {
		return &NodeMutationRequest{
			Name: "probe-svc", NodeType: "agent", Enable: true,
			Scheme: "https", BasePath: "/",
		}
	}

	// Nil: flag untouched (still false).
	if err := svc.UpdateFromRequest(node.Id, base()); err != nil {
		t.Fatalf("update (nil): %v", err)
	}
	var row model.AgentNode
	if err := db.First(&row, ag.Id).Error; err != nil {
		t.Fatal(err)
	}
	if row.ProbeOnly {
		t.Fatal("nil probeOnly must not set the flag")
	}

	// True: flag set on the agent row, and the annotated view shows it.
	req := base()
	req.ProbeOnly = &truth
	if err := svc.UpdateFromRequest(node.Id, req); err != nil {
		t.Fatalf("update (true): %v", err)
	}
	if err := db.First(&row, ag.Id).Error; err != nil {
		t.Fatal(err)
	}
	if !row.ProbeOnly {
		t.Fatal("probeOnly=true not persisted on the agent row")
	}
	view, err := svc.GetViewById(node.Id)
	if err != nil {
		t.Fatalf("GetViewById: %v", err)
	}
	if !view.ProbeOnly {
		t.Fatal("view probeOnly = false, want true from the binding")
	}

	// False: explicit off wins over the stored value.
	req = base()
	req.ProbeOnly = &lies
	if err := svc.UpdateFromRequest(node.Id, req); err != nil {
		t.Fatalf("update (false): %v", err)
	}
	if err := db.First(&row, ag.Id).Error; err != nil {
		t.Fatal(err)
	}
	if row.ProbeOnly {
		t.Fatal("probeOnly=false not persisted on the agent row")
	}
}