50 lines
1.5 KiB
Go
50 lines
1.5 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package date_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"storj.io/storj/private/date"
|
|
)
|
|
|
|
func TestMonthBoundary(t *testing.T) {
|
|
now := time.Now()
|
|
|
|
start, end := date.MonthBoundary(now)
|
|
assert.Equal(t, start, time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, now.Location()))
|
|
assert.Equal(t, end, time.Date(now.Year(), now.Month()+1, 1, 0, 0, 0, -1, now.Location()))
|
|
}
|
|
|
|
func TestDayBoundary(t *testing.T) {
|
|
now := time.Now()
|
|
|
|
start, end := date.DayBoundary(now)
|
|
assert.Equal(t, start, time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location()))
|
|
assert.Equal(t, end, time.Date(now.Year(), now.Month(), now.Day()+1, 0, 0, 0, -1, now.Location()))
|
|
}
|
|
|
|
func TestPeriodToTime(t *testing.T) {
|
|
testCases := [...]struct {
|
|
period string
|
|
periodTime time.Time
|
|
}{
|
|
{"2020-01", time.Date(2020, 01, 01, 0, 0, 0, 0, &time.Location{})},
|
|
{"2020-02-01", time.Date(2020, 02, 01, 0, 0, 0, 0, &time.Location{})},
|
|
{"2019-11-04 14:14:14", time.Date(2019, 11, 01, 0, 0, 0, 0, &time.Location{})},
|
|
{"2020-03-04T15:04:05-0700", time.Date(2020, 03, 01, 0, 0, 0, 0, &time.Location{})},
|
|
{"2020-04gasgahsgnasjfasgjs", time.Date(2020, 04, 01, 0, 0, 0, 0, &time.Location{})},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
periodTime, err := date.PeriodToTime(tc.period)
|
|
require.NoError(t, err)
|
|
require.Equal(t, periodTime.String(), tc.periodTime.String())
|
|
}
|
|
}
|