scheibling
217
timex/time_test.go
Normal file
217
timex/time_test.go
Normal file
@@ -0,0 +1,217 @@
|
||||
package timex
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestIsAmericaSummerTime(t *testing.T) {
|
||||
testCases := []struct {
|
||||
Date string
|
||||
SummerTime bool
|
||||
}{
|
||||
{"0001-01-01", false},
|
||||
{"2021-11-10", false},
|
||||
{"2021-12-10", false},
|
||||
{"2021-03-10", false},
|
||||
{"2021-03-14", true},
|
||||
{"2021-11-01", true},
|
||||
{"2021-10-10", true},
|
||||
{"2021-10-11", true},
|
||||
{"2021-10-12", true},
|
||||
{"2021-12-12", false},
|
||||
{"2022-03-15", true},
|
||||
}
|
||||
for _, testCase := range testCases {
|
||||
d, _ := time.Parse("2006-01-02", testCase.Date)
|
||||
v := IsAmericaSummerTime(d)
|
||||
if v != testCase.SummerTime {
|
||||
t.Errorf("%s except %v, actual %v", testCase.Date, testCase.SummerTime, v)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestBetween(t *testing.T) {
|
||||
testCases := []struct {
|
||||
tag string
|
||||
t string
|
||||
begin string
|
||||
end string
|
||||
expected bool
|
||||
}{
|
||||
{"t1", "2022-01-01", "2022-01-01", "2022-01-01", true},
|
||||
{"t2", "2022-01-02", "2022-01-01", "2022-01-01", false},
|
||||
{"t3", "2022-01-02", "2022-01-01", "2022-01-02", true},
|
||||
}
|
||||
layout := "2006-01-02"
|
||||
for _, testCase := range testCases {
|
||||
tv, _ := time.Parse(layout, testCase.t)
|
||||
begin, _ := time.Parse(layout, testCase.begin)
|
||||
end, _ := time.Parse(layout, testCase.end)
|
||||
v := Between(tv, begin, end)
|
||||
assert.Equal(t, testCase.expected, v, testCase.tag)
|
||||
}
|
||||
}
|
||||
|
||||
func TestDayStart(t *testing.T) {
|
||||
testCases := []struct {
|
||||
tag string
|
||||
time string
|
||||
layout string
|
||||
expected string
|
||||
}{
|
||||
{"t1", "2022-01-01T12:12:00.924Z", "2006-01-02T15:04:05Z", "2022-01-01 00:00:00"},
|
||||
{"t2", "2022-01-01 00:00:00", "2006-01-02 15:04:05", "2022-01-01 00:00:00"},
|
||||
}
|
||||
|
||||
for _, testCase := range testCases {
|
||||
tv, err := time.Parse(testCase.layout, testCase.time)
|
||||
if err != nil {
|
||||
t.Errorf(err.Error())
|
||||
}
|
||||
v := DayStart(tv).Format("2006-01-02 15:04:05")
|
||||
assert.Equal(t, testCase.expected, v, testCase.tag)
|
||||
}
|
||||
}
|
||||
|
||||
func TestDayEnd(t *testing.T) {
|
||||
testCases := []struct {
|
||||
tag string
|
||||
time string
|
||||
layout string
|
||||
expected string
|
||||
}{
|
||||
{"t1", "2022-01-01 12:12:00", "2006-01-02 15:04:05", "2022-01-01 23:59:59"},
|
||||
{"t2", "2022-01-01 00:00:00", "2006-01-02 15:04:05", "2022-01-01 23:59:59"},
|
||||
}
|
||||
for _, testCase := range testCases {
|
||||
tv, _ := time.Parse(testCase.layout, testCase.time)
|
||||
v := DayEnd(tv).Format("2006-01-02 15:04:05")
|
||||
assert.Equal(t, testCase.expected, v, testCase.tag)
|
||||
}
|
||||
}
|
||||
|
||||
func TestMonthStart(t *testing.T) {
|
||||
testCases := []struct {
|
||||
tag string
|
||||
time string
|
||||
layout string
|
||||
expected string
|
||||
}{
|
||||
{"t1", "2022-01-12 12:12:00", "2006-01-02 15:04:05", "2022-01-01 00:00:00"},
|
||||
{"t2", "2022-01-21 00:00:00", "2006-01-02 15:04:05", "2022-01-01 00:00:00"},
|
||||
}
|
||||
for _, testCase := range testCases {
|
||||
tv, _ := time.Parse(testCase.layout, testCase.time)
|
||||
v := MonthStart(tv).Format("2006-01-02 15:04:05")
|
||||
assert.Equal(t, testCase.expected, v, testCase.tag)
|
||||
}
|
||||
}
|
||||
|
||||
func TestMonthEnd(t *testing.T) {
|
||||
testCases := []struct {
|
||||
tag string
|
||||
time string
|
||||
layout string
|
||||
expected string
|
||||
}{
|
||||
{"t1", "2022-01-12 12:12:00", "2006-01-02 15:04:05", "2022-01-31 23:59:59"},
|
||||
{"t2", "2022-02-21 00:00:00", "2006-01-02 15:04:05", "2022-02-28 23:59:59"},
|
||||
}
|
||||
for _, testCase := range testCases {
|
||||
tv, _ := time.Parse(testCase.layout, testCase.time)
|
||||
v := MonthEnd(tv).Format("2006-01-02 15:04:05")
|
||||
assert.Equal(t, testCase.expected, v, testCase.tag)
|
||||
}
|
||||
}
|
||||
|
||||
func TestWeekStart(t *testing.T) {
|
||||
testCases := []struct {
|
||||
tag string
|
||||
yearWeek int
|
||||
expected string
|
||||
}{
|
||||
{"t1", 202201, "2022-01-03 00:00:00"},
|
||||
{"t2", 202202, "2022-01-10 00:00:00"},
|
||||
}
|
||||
for _, testCase := range testCases {
|
||||
v := WeekStart(testCase.yearWeek).Format("2006-01-02 15:04:05")
|
||||
assert.Equal(t, testCase.expected, v, testCase.tag)
|
||||
}
|
||||
}
|
||||
|
||||
func TestWeekEnd(t *testing.T) {
|
||||
testCases := []struct {
|
||||
tag string
|
||||
yearWeek int
|
||||
expected string
|
||||
}{
|
||||
{"t1", 202201, "2022-01-09 23:59:59"},
|
||||
{"t2", 202202, "2022-01-16 23:59:59"},
|
||||
}
|
||||
for _, testCase := range testCases {
|
||||
v := WeekEnd(testCase.yearWeek).Format("2006-01-02 15:04:05")
|
||||
assert.Equal(t, testCase.expected, v, testCase.tag)
|
||||
}
|
||||
}
|
||||
|
||||
func TestYearWeeksByWeek(t *testing.T) {
|
||||
testCases := []struct {
|
||||
tag string
|
||||
beginYearWeek int
|
||||
endYearWeek int
|
||||
expected []int
|
||||
}{
|
||||
{"t1", 202201, 202202, []int{202201, 202202}},
|
||||
{"t2", 202201, 202204, []int{202201, 202202, 202203, 202204}},
|
||||
}
|
||||
for _, testCase := range testCases {
|
||||
v := YearWeeksByWeek(testCase.beginYearWeek, testCase.endYearWeek)
|
||||
assert.Equal(t, testCase.expected, v, testCase.tag)
|
||||
}
|
||||
}
|
||||
|
||||
func TestYearWeeksByTime(t *testing.T) {
|
||||
testCases := []struct {
|
||||
tag string
|
||||
beginDate string
|
||||
endDate string
|
||||
expected []int
|
||||
}{
|
||||
{"t1", "2022-01-01", "2022-01-02", []int{202152}},
|
||||
{"t2", "2022-01-01", "2022-02-02", []int{202152, 202201, 202202, 202203, 202204, 202205}},
|
||||
}
|
||||
for _, testCase := range testCases {
|
||||
beginDate, _ := time.Parse("2006-01-02", testCase.beginDate)
|
||||
endDate, _ := time.Parse("2006-01-02", testCase.endDate)
|
||||
v := YearWeeksByTime(beginDate, endDate)
|
||||
assert.Equal(t, testCase.expected, v, testCase.tag)
|
||||
}
|
||||
}
|
||||
|
||||
// https://savvytime.com/week-number
|
||||
func TestXISOWeek(t *testing.T) {
|
||||
testCases := []struct {
|
||||
tag string
|
||||
date string
|
||||
expected string
|
||||
}{
|
||||
{"t1", "2022-01-01", "202152"},
|
||||
{"t2", "2022-01-02", "202201"},
|
||||
{"t3", "2022-01-09", "202202"},
|
||||
{"t4", "2022-01-10", "202202"},
|
||||
{"t5", "2022-01-15", "202202"},
|
||||
{"t6", "2022-01-16", "202203"},
|
||||
{"t7", "2022-01-17", "202203"},
|
||||
{"t8", "2022-01-29", "202204"},
|
||||
{"t9", "2022-12-25", "202252"},
|
||||
{"t10", "2023-01-01", "202301"},
|
||||
}
|
||||
for _, testCase := range testCases {
|
||||
d, _ := time.Parse("2006-01-02", testCase.date)
|
||||
year, week := XISOWeek(d)
|
||||
assert.Equal(t, testCase.expected, fmt.Sprintf("%d%02d", year, week), testCase.tag)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user