mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2026-07-17 01:56:06 +00:00
fix(settings): normalize API token timestamps (#5599)
* fix(settings): normalize API token timestamps * refactor(api-token): share timestamp threshold --------- Co-authored-by: Tomilla <5007859+Tomilla@users.noreply.github.com>
This commit is contained in:
@@ -0,0 +1,49 @@
|
||||
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)
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user