69 lines
2.0 KiB
Go
69 lines
2.0 KiB
Go
|
package cast_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"apigo.cc/go/cast"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestParseTime(t *testing.T) {
|
||
|
|
tests := []struct {
|
||
|
|
input any
|
||
|
|
expected string
|
||
|
|
}{
|
||
|
|
{"2023-05-04 12:34:56", "2023-05-04 12:34:56"},
|
||
|
|
{"2023/05/04 12:34:56", "2023-05-04 12:34:56"},
|
||
|
|
{"2023.05.04 12:34:56", "2023-05-04 12:34:56"},
|
||
|
|
{"20230504123456", "2023-05-04 12:34:56"},
|
||
|
|
{1683196496, "2023-05-04 10:34:56"}, // Unix timestamp (Example)
|
||
|
|
{"2023年5月4日 12时34分56秒", "2023-05-04 12:34:56"},
|
||
|
|
}
|
||
|
|
|
||
|
|
// Set UTC for test stability if needed, or just compare formatted strings
|
||
|
|
loc, _ := time.LoadLocation("UTC")
|
||
|
|
cast.SetDefaultTimeZone(loc)
|
||
|
|
|
||
|
|
for _, tt := range tests {
|
||
|
|
tm := cast.ParseTime(tt.input)
|
||
|
|
// We use a relative check for timestamp since it depends on the timezone if not careful
|
||
|
|
if tt.input == 1683196496 {
|
||
|
|
// 1683196496 is 2023-05-04 10:34:56 UTC
|
||
|
|
if tm.Format("2006-01-02 15:04:05") != tt.expected {
|
||
|
|
t.Errorf("ParseTime(%v) = %v, want %v", tt.input, tm.Format("2006-01-02 15:04:05"), tt.expected)
|
||
|
|
}
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
if tm.Format("2006-01-02 15:04:05") != tt.expected {
|
||
|
|
t.Errorf("ParseTime(%v) = %v, want %v", tt.input, tm.Format("2006-01-02 15:04:05"), tt.expected)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFormatTime(t *testing.T) {
|
||
|
|
tm := time.Date(2023, 5, 4, 12, 34, 56, 0, time.UTC)
|
||
|
|
formatted := cast.FormatTime("YYYY-MM-DD HH:mm:ss", tm)
|
||
|
|
if formatted != "2023-05-04 12:34:56" {
|
||
|
|
t.Errorf("FormatTime failed, got %s", formatted)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestAddTime(t *testing.T) {
|
||
|
|
tm := time.Date(2023, 5, 4, 12, 34, 56, 0, time.UTC)
|
||
|
|
added := cast.AddTime("+1Y+1M+1D", tm)
|
||
|
|
expected := time.Date(2024, 6, 5, 12, 34, 56, 0, time.UTC)
|
||
|
|
if !added.Equal(expected) {
|
||
|
|
t.Errorf("AddTime failed, got %v, want %v", added, expected)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestTimeZoneNow(t *testing.T) {
|
||
|
|
loc, _ := time.LoadLocation("Asia/Shanghai")
|
||
|
|
tz := cast.NewTimeZone(loc)
|
||
|
|
now := tz.Now()
|
||
|
|
if now.Location().String() != "Asia/Shanghai" {
|
||
|
|
t.Errorf("TimeZone.Now() failed to set location, got %s", now.Location().String())
|
||
|
|
}
|
||
|
|
}
|