epusdt/src/model/service/task_tron_test.go

package service

import (
	"testing"

	"github.com/GMWalletApp/epusdt/internal/testutil"
	"github.com/GMWalletApp/epusdt/model/dao"
	"github.com/GMWalletApp/epusdt/model/mdb"
)

// TestResolveTronNode_NoRow verifies that resolveTronNode requires an enabled
// TRON row from rpc_nodes.
func TestResolveTronNode_NoRow(t *testing.T) {
	cleanup := testutil.SetupTestDatabases(t)
	defer cleanup()

	if gotURL, gotKey, err := resolveTronNode(); err == nil {
		t.Fatalf("resolveTronNode() = (%q, %q, nil), want error", gotURL, gotKey)
	}
}

// TestResolveTronNode_WithRow verifies that resolveTronNode uses the DB row
// when an enabled TRON HTTP node is present.
func TestResolveTronNode_WithRow(t *testing.T) {
	cleanup := testutil.SetupTestDatabases(t)
	defer cleanup()

	// Insert an enabled TRON http node.
	node := &mdb.RpcNode{
		Network: mdb.NetworkTron,
		Url:     "https://custom-tron.example.com",
		Type:    mdb.RpcNodeTypeHttp,
		ApiKey:  "db-api-key",
		Weight:  1,
		Enabled: true,
		Status:  mdb.RpcNodeStatusOk,
	}
	if err := dao.Mdb.Create(node).Error; err != nil {
		t.Fatalf("seed rpc_node: %v", err)
	}

	gotURL, gotKey, err := resolveTronNode()
	if err != nil {
		t.Fatalf("resolveTronNode(): %v", err)
	}
	if gotURL != "https://custom-tron.example.com" {
		t.Errorf("url = %q, want https://custom-tron.example.com", gotURL)
	}
	if gotKey != "db-api-key" {
		t.Errorf("apiKey = %q, want \"db-api-key\"", gotKey)
	}
}

func TestResolveTronNodeIgnoresManualVerifyOnly(t *testing.T) {
	cleanup := testutil.SetupTestDatabases(t)
	defer cleanup()

	if err := dao.Mdb.Create(&mdb.RpcNode{
		Network: mdb.NetworkTron,
		Url:     "https://paid-tron.example.com",
		Type:    mdb.RpcNodeTypeHttp,
		ApiKey:  "paid-key",
		Weight:  100,
		Enabled: true,
		Purpose: mdb.RpcNodePurposeManualVerify,
		Status:  mdb.RpcNodeStatusOk,
	}).Error; err != nil {
		t.Fatalf("seed manual rpc_node: %v", err)
	}

	if gotURL, gotKey, err := resolveTronNode(); err == nil {
		t.Fatalf("resolveTronNode() = (%q, %q, nil), want error", gotURL, gotKey)
	}
}

func TestResolveTronNodeUsesGeneralWhenManualVerifyExists(t *testing.T) {
	cleanup := testutil.SetupTestDatabases(t)
	defer cleanup()

	rows := []mdb.RpcNode{
		{Network: mdb.NetworkTron, Url: "https://paid-tron.example.com", Type: mdb.RpcNodeTypeHttp, ApiKey: "paid-key", Weight: 100, Enabled: true, Purpose: mdb.RpcNodePurposeManualVerify, Status: mdb.RpcNodeStatusOk},
		{Network: mdb.NetworkTron, Url: "https://general-tron.example.com", Type: mdb.RpcNodeTypeHttp, ApiKey: "general-key", Weight: 1, Enabled: true, Purpose: mdb.RpcNodePurposeGeneral, Status: mdb.RpcNodeStatusOk},
	}
	if err := dao.Mdb.Create(&rows).Error; err != nil {
		t.Fatalf("seed rpc_nodes: %v", err)
	}

	gotURL, gotKey, err := resolveTronNode()
	if err != nil {
		t.Fatalf("resolveTronNode(): %v", err)
	}
	if gotURL != "https://general-tron.example.com" || gotKey != "general-key" {
		t.Fatalf("resolveTronNode() = (%q, %q), want general node", gotURL, gotKey)
	}
}

// TestResolveTronNode_DisabledRow verifies that a disabled row is ignored.
func TestResolveTronNode_DisabledRow(t *testing.T) {
	cleanup := testutil.SetupTestDatabases(t)
	defer cleanup()

	node := &mdb.RpcNode{
		Network: mdb.NetworkTron,
		Url:     "https://disabled-tron.example.com",
		Type:    mdb.RpcNodeTypeHttp,
		ApiKey:  "disabled-key",
		Weight:  1,
		Enabled: true, // start enabled, then disable below
		Status:  mdb.RpcNodeStatusOk,
	}
	if err := dao.Mdb.Create(node).Error; err != nil {
		t.Fatalf("seed rpc_node: %v", err)
	}
	// Explicitly disable — GORM does not save bool zero-value on Create.
	if err := dao.Mdb.Model(node).Update("enabled", false).Error; err != nil {
		t.Fatalf("disable rpc_node: %v", err)
	}

	if gotURL, gotKey, err := resolveTronNode(); err == nil {
		t.Fatalf("resolveTronNode() = (%q, %q, nil), want error", gotURL, gotKey)
	}
}