3x-ui/internal/web/service/client_group_reset_test.go

package service

import (
	"testing"

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

func groupByName(t *testing.T, svc *ClientService, name string) GroupSummary {
	t.Helper()
	rows, err := svc.ListGroups()
	if err != nil {
		t.Fatalf("ListGroups: %v", err)
	}
	for _, g := range rows {
		if g.Name == name {
			return g
		}
	}
	t.Fatalf("group %q not found in %v", name, rows)
	return GroupSummary{}
}

func seedGroupedClient(t *testing.T, email, group string, up, down int64) {
	t.Helper()
	if err := database.GetDB().Create(&model.ClientRecord{Email: email, Enable: true, Group: group}).Error; err != nil {
		t.Fatalf("seed client record %q: %v", email, err)
	}
	seedClientRow(t, email, 1, up, down, 0)
}

func TestResetGroupTraffic_ZeroesGroupButKeepsClients(t *testing.T) {
	initTrafficTestDB(t)
	svc := &ClientService{}

	seedGroupedClient(t, "alice", "vip", 100, 200)
	seedGroupedClient(t, "bob", "vip", 50, 50)

	before := groupByName(t, svc, "vip")
	if before.Up != 150 || before.Down != 250 || before.TrafficUsed != 400 || before.ClientCount != 2 {
		t.Fatalf("before reset: got %+v, want up=150 down=250 used=400 count=2", before)
	}

	if err := svc.ResetGroupTraffic("vip"); err != nil {
		t.Fatalf("ResetGroupTraffic: %v", err)
	}

	after := groupByName(t, svc, "vip")
	if after.Up != 0 || after.Down != 0 || after.TrafficUsed != 0 {
		t.Fatalf("after reset: got %+v, want up=0 down=0 used=0", after)
	}
	if after.ClientCount != 2 {
		t.Fatalf("after reset: client count changed to %d, want 2", after.ClientCount)
	}

	var alice xray.ClientTraffic
	if err := database.GetDB().Where("email = ?", "alice").First(&alice).Error; err != nil {
		t.Fatalf("load alice traffic: %v", err)
	}
	if alice.Up != 100 || alice.Down != 200 {
		t.Fatalf("client counter modified by group reset: alice up=%d down=%d, want 100/200", alice.Up, alice.Down)
	}
}

func TestResetGroupTraffic_NewTrafficAccumulatesAboveBaseline(t *testing.T) {
	initTrafficTestDB(t)
	svc := &ClientService{}

	seedGroupedClient(t, "carol", "team", 100, 100)
	if err := svc.ResetGroupTraffic("team"); err != nil {
		t.Fatalf("ResetGroupTraffic: %v", err)
	}
	if g := groupByName(t, svc, "team"); g.Up != 0 || g.Down != 0 {
		t.Fatalf("after reset: got %+v, want up=0 down=0", g)
	}

	if err := database.GetDB().Table("client_traffics").
		Where("email = ?", "carol").
		Updates(map[string]any{"up": 130, "down": 100}).Error; err != nil {
		t.Fatalf("bump carol traffic: %v", err)
	}

	g := groupByName(t, svc, "team")
	if g.Up != 30 || g.Down != 0 || g.TrafficUsed != 30 {
		t.Fatalf("post-bump: got %+v, want up=30 down=0 used=30", g)
	}
}

func TestResetGroupTraffic_CreatesRowForDerivedGroup(t *testing.T) {
	initTrafficTestDB(t)
	svc := &ClientService{}

	seedGroupedClient(t, "dave", "adhoc", 70, 30)

	var rows int64
	if err := database.GetDB().Model(&model.ClientGroup{}).Where("name = ?", "adhoc").Count(&rows).Error; err != nil {
		t.Fatalf("count client_groups: %v", err)
	}
	if rows != 0 {
		t.Fatalf("precondition: derived group should have no client_groups row, got %d", rows)
	}

	if err := svc.ResetGroupTraffic("adhoc"); err != nil {
		t.Fatalf("ResetGroupTraffic: %v", err)
	}

	var stored model.ClientGroup
	if err := database.GetDB().Where("name = ?", "adhoc").First(&stored).Error; err != nil {
		t.Fatalf("client_groups row not created: %v", err)
	}
	if stored.ResetUp != 70 || stored.ResetDown != 30 {
		t.Fatalf("baseline not snapshotted: got up=%d down=%d, want 70/30", stored.ResetUp, stored.ResetDown)
	}
	if g := groupByName(t, svc, "adhoc"); g.Up != 0 || g.Down != 0 {
		t.Fatalf("after reset: got %+v, want up=0 down=0", g)
	}
}

func TestResetGroupTraffic_EmptyNameRejected(t *testing.T) {
	initTrafficTestDB(t)
	svc := &ClientService{}
	if err := svc.ResetGroupTraffic("   "); err == nil {
		t.Fatal("ResetGroupTraffic(blank) = nil, want error")
	}
}

func TestGroupTotalsSurviveSingleClientReset(t *testing.T) {
	initTrafficTestDB(t)
	csvc := &ClientService{}
	isvc := &InboundService{}

	seedGroupedClient(t, "erin", "gold", 100, 200)
	seedGroupedClient(t, "frank", "gold", 40, 60)

	if err := isvc.ResetClientTrafficByEmail("erin"); err != nil {
		t.Fatalf("ResetClientTrafficByEmail: %v", err)
	}

	g := groupByName(t, csvc, "gold")
	if g.Up != 140 || g.Down != 260 || g.TrafficUsed != 400 {
		t.Fatalf("group totals changed by client reset: got %+v, want up=140 down=260 used=400", g)
	}

	var erin xray.ClientTraffic
	if err := database.GetDB().Where("email = ?", "erin").First(&erin).Error; err != nil {
		t.Fatalf("load erin traffic: %v", err)
	}
	if erin.Up != 0 || erin.Down != 0 {
		t.Fatalf("client traffic not reset: up=%d down=%d", erin.Up, erin.Down)
	}
}

func TestGroupTotalsSurviveBulkClientReset(t *testing.T) {
	initTrafficTestDB(t)
	csvc := &ClientService{}
	isvc := &InboundService{}

	seedGroupedClient(t, "gina", "silver", 10, 20)
	seedGroupedClient(t, "hank", "silver", 30, 40)

	affected, err := csvc.BulkResetTraffic(isvc, []string{"gina", "hank"})
	if err != nil {
		t.Fatalf("BulkResetTraffic: %v", err)
	}
	if affected != 2 {
		t.Fatalf("BulkResetTraffic affected = %d, want 2", affected)
	}

	g := groupByName(t, csvc, "silver")
	if g.Up != 40 || g.Down != 60 || g.TrafficUsed != 100 {
		t.Fatalf("group totals changed by bulk reset: got %+v, want up=40 down=60 used=100", g)
	}
}

func TestGroupTotalsSurviveClientDelete(t *testing.T) {
	initTrafficTestDB(t)
	csvc := &ClientService{}
	isvc := &InboundService{}

	seedGroupedClient(t, "iris", "bronze", 70, 30)
	seedGroupedClient(t, "jack", "bronze", 5, 5)

	var rec model.ClientRecord
	if err := database.GetDB().Where("email = ?", "iris").First(&rec).Error; err != nil {
		t.Fatalf("load iris record: %v", err)
	}
	if _, err := csvc.Delete(isvc, rec.Id, false); err != nil {
		t.Fatalf("Delete: %v", err)
	}

	g := groupByName(t, csvc, "bronze")
	if g.Up != 75 || g.Down != 35 || g.TrafficUsed != 110 {
		t.Fatalf("group totals changed by client delete: got %+v, want up=75 down=35 used=110", g)
	}
	if g.ClientCount != 1 {
		t.Fatalf("client count = %d, want 1", g.ClientCount)
	}

	var trafficRows int64
	if err := database.GetDB().Model(&xray.ClientTraffic{}).Where("email = ?", "iris").Count(&trafficRows).Error; err != nil {
		t.Fatalf("count iris traffic rows: %v", err)
	}
	if trafficRows != 0 {
		t.Fatalf("iris traffic row survived delete")
	}
}

func TestGroupResetStillZeroesAfterBaselineAdjustments(t *testing.T) {
	initTrafficTestDB(t)
	csvc := &ClientService{}
	isvc := &InboundService{}

	seedGroupedClient(t, "kate", "iron", 100, 100)
	if err := isvc.ResetClientTrafficByEmail("kate"); err != nil {
		t.Fatalf("ResetClientTrafficByEmail: %v", err)
	}
	if g := groupByName(t, csvc, "iron"); g.TrafficUsed != 200 {
		t.Fatalf("pre group-reset: got %+v, want used=200", g)
	}

	if err := csvc.ResetGroupTraffic("iron"); err != nil {
		t.Fatalf("ResetGroupTraffic: %v", err)
	}
	if g := groupByName(t, csvc, "iron"); g.Up != 0 || g.Down != 0 || g.TrafficUsed != 0 {
		t.Fatalf("group reset did not zero adjusted baselines: got %+v", g)
	}
}

// Counters are stored up to TrafficMax (a node can report that much); adding two in SQL
// overflowed and failed every group listing, reset and baseline shift for the panel.
func TestGroupTrafficSurvivesCountersAtTheCap(t *testing.T) {
	initTrafficTestDB(t)
	svc := &ClientService{}
	seedGroupedClient(t, "huge-a", "vip", database.TrafficMax, database.TrafficMax)
	seedGroupedClient(t, "huge-b", "vip", database.TrafficMax, database.TrafficMax)
	if g := groupByName(t, svc, "vip"); g.Up != database.TrafficMax || g.ClientCount != 2 {
		t.Fatalf("group = %+v, want its counters held at TrafficMax", g)
	}
	if err := svc.ResetGroupTraffic("vip"); err != nil {
		t.Fatalf("ResetGroupTraffic: %v", err)
	}
	if err := adjustGroupBaselinesForRemovedTraffic(database.GetDB(), []string{"huge-a", "huge-b"}); err != nil {
		t.Fatalf("shift baselines: %v", err)
	}
}