3x-ui/internal/database/api_token_timestamp_test.go

package database

import (
	"testing"

	"github.com/mhsanaei/3x-ui/v3/internal/database/model"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
)

func TestNormalizeApiTokenCreatedAtSeconds(t *testing.T) {
	originalDB := db
	t.Cleanup(func() { db = originalDB })

	var err error
	db, err = gorm.Open(sqlite.Open(":memory:"), &gorm.Config{Logger: logger.Discard})
	if err != nil {
		t.Fatalf("open sqlite: %v", err)
	}
	if err := db.AutoMigrate(&model.ApiToken{}); err != nil {
		t.Fatalf("migrate api_tokens: %v", err)
	}

	rows := []model.ApiToken{
		{Name: "seconds", Token: "a", CreatedAt: 1_782_485_394},
		{Name: "milliseconds", Token: "b", CreatedAt: 1_782_485_394_270},
	}
	if err := db.Create(&rows).Error; err != nil {
		t.Fatalf("seed api tokens: %v", err)
	}

	if err := normalizeApiTokenCreatedAtSeconds(); err != nil {
		t.Fatalf("normalize timestamps: %v", err)
	}
	if err := normalizeApiTokenCreatedAtSeconds(); err != nil {
		t.Fatalf("normalize timestamps again: %v", err)
	}

	var got []model.ApiToken
	if err := db.Order("id asc").Find(&got).Error; err != nil {
		t.Fatalf("read api tokens: %v", err)
	}
	for _, row := range got {
		if row.CreatedAt != 1_782_485_394 {
			t.Fatalf("%s created_at = %d, want seconds", row.Name, row.CreatedAt)
		}
	}
}