3x-ui/internal/web/service/tgbot/tgbot_client_expiry_test.go

package tgbot

import (
	"encoding/json"
	"path/filepath"
	"strings"
	"testing"

	"github.com/mhsanaei/3x-ui/v3/internal/database"
	"github.com/mhsanaei/3x-ui/v3/internal/web/locale"
	"github.com/mhsanaei/3x-ui/v3/internal/xray"

	"github.com/nicksnyder/go-i18n/v2/i18n"
	"golang.org/x/text/language"
)

// clientInfoLocalizer renders the lines clientInfoMsg prints with the templates
// the translation files carry; without it I18n returns the bare keys.
func clientInfoLocalizer(t *testing.T) {
	t.Helper()
	bundle := i18n.NewBundle(language.MustParse("en-US"))
	bundle.RegisterUnmarshalFunc("json", json.Unmarshal)
	_ = bundle.AddMessages(language.MustParse("en-US"),
		&i18n.Message{ID: "tgbot.messages.email", Other: "Email: {{ .Email }}\r\n"},
		&i18n.Message{ID: "tgbot.days", Other: "Days"},
		&i18n.Message{ID: "tgbot.messages.expireIn", Other: "Expire In: {{ .Time }}\r\n"},
		&i18n.Message{ID: "tgbot.messages.expire", Other: "Expire Date: {{ .Time }}\r\n"},
		&i18n.Message{ID: "tgbot.wentWrong", Other: "went wrong"},
	)
	orig := locale.LocalizerBot
	t.Cleanup(func() { locale.LocalizerBot = orig })
	locale.LocalizerBot = i18n.NewLocalizer(bundle, "en-US")
}

// Regression test: a start-after-first-use client is stored as a negative duration,
// and a disabled one rendered it as a 1969 date.
func TestClientInfoShowsStartAfterFirstUseWhenDisabled(t *testing.T) {
	if err := database.InitDB(filepath.Join(t.TempDir(), "x-ui.db")); err != nil {
		t.Fatalf("InitDB: %v", err)
	}
	t.Cleanup(func() { _ = database.CloseDB() })
	clientInfoLocalizer(t)

	traffic := &xray.ClientTraffic{
		Email:      "[email protected]",
		Enable:     false,
		ExpiryTime: -30 * 24 * 60 * 60000,
	}

	out := (&Tgbot{}).clientInfoMsg(traffic, false, false, false, true, false, false)

	if strings.Contains(out, "1969") {
		t.Errorf("client info = %q, want the days left, not a 1969 date", out)
	}
	if !strings.Contains(out, "Expire In: 30 Days") {
		t.Errorf("client info = %q, want it to contain %q", out, "Expire In: 30 Days")
	}
}