32 lines
684 B
Go
32 lines
684 B
Go
package userser
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"91porn-server/models/v/usermod"
|
|
"91porn-server/models/v/walletmod"
|
|
)
|
|
|
|
func TestToUserRecordListIncludesDramaExpire(t *testing.T) {
|
|
expire := time.Date(2026, 9, 26, 12, 0, 0, 0, time.UTC)
|
|
users := []usermod.User{{UID: 100001, DramaExpire: expire}}
|
|
|
|
records := toUserRecordList(
|
|
users,
|
|
map[uint64]int64{},
|
|
map[uint64]uint64{},
|
|
map[uint64]int64{},
|
|
map[uint64]walletmod.Wallet{},
|
|
map[uint64]int{},
|
|
map[uint64]int{},
|
|
)
|
|
|
|
if len(records) != 1 {
|
|
t.Fatalf("record count = %d, want 1", len(records))
|
|
}
|
|
if !records[0].DramaExpire.Equal(expire) {
|
|
t.Fatalf("dramaExpire = %s, want %s", records[0].DramaExpire, expire)
|
|
}
|
|
}
|