mirror of
https://github.com/0rangebananaspy/authelia.git
synced 2024-09-14 22:47:21 +07:00
6276883f04
This enhances the existing time.Duration parser to allow multiple units, and implements a decode hook which can be used by koanf to decode string/integers into time.Durations as applicable.
125 lines
3.4 KiB
Go
125 lines
3.4 KiB
Go
package utils
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestParseDurationString_ShouldParseDurationString(t *testing.T) {
|
|
duration, err := ParseDurationString("1h")
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 60*time.Minute, duration)
|
|
}
|
|
|
|
func TestParseDurationString_ShouldParseBlankString(t *testing.T) {
|
|
duration, err := ParseDurationString("")
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, time.Second*0, duration)
|
|
}
|
|
|
|
func TestParseDurationString_ShouldParseDurationStringAllUnits(t *testing.T) {
|
|
duration, err := ParseDurationString("1y")
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, time.Hour*24*365, duration)
|
|
|
|
duration, err = ParseDurationString("1M")
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, time.Hour*24*30, duration)
|
|
|
|
duration, err = ParseDurationString("1w")
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, time.Hour*24*7, duration)
|
|
|
|
duration, err = ParseDurationString("1d")
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, time.Hour*24, duration)
|
|
|
|
duration, err = ParseDurationString("1h")
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, time.Hour, duration)
|
|
|
|
duration, err = ParseDurationString("1s")
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, time.Second, duration)
|
|
}
|
|
|
|
func TestParseDurationString_ShouldParseSecondsString(t *testing.T) {
|
|
duration, err := ParseDurationString("100")
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 100*time.Second, duration)
|
|
}
|
|
|
|
func TestParseDurationString_ShouldNotParseDurationStringWithOutOfOrderQuantitiesAndUnits(t *testing.T) {
|
|
duration, err := ParseDurationString("h1")
|
|
|
|
assert.EqualError(t, err, "could not parse 'h1' as a duration")
|
|
assert.Equal(t, time.Duration(0), duration)
|
|
}
|
|
|
|
func TestParseDurationString_ShouldNotParseBadDurationString(t *testing.T) {
|
|
duration, err := ParseDurationString("10x")
|
|
|
|
assert.EqualError(t, err, "could not parse the units portion of '10x' in duration string '10x': the unit 'x' is not valid")
|
|
assert.Equal(t, time.Duration(0), duration)
|
|
}
|
|
|
|
func TestParseDurationString_ShouldParseDurationStringWithMultiValueUnits(t *testing.T) {
|
|
duration, err := ParseDurationString("10ms")
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, time.Duration(10)*time.Millisecond, duration)
|
|
}
|
|
|
|
func TestParseDurationString_ShouldParseDurationStringWithLeadingZero(t *testing.T) {
|
|
duration, err := ParseDurationString("005h")
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, time.Duration(5)*time.Hour, duration)
|
|
}
|
|
|
|
func TestParseDurationString_ShouldParseMultiUnitValues(t *testing.T) {
|
|
duration, err := ParseDurationString("1d3w10ms")
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t,
|
|
(time.Hour*time.Duration(24))+
|
|
(time.Hour*time.Duration(24)*time.Duration(7)*time.Duration(3))+
|
|
(time.Millisecond*time.Duration(10)), duration)
|
|
}
|
|
|
|
func TestParseDurationString_ShouldParseDuplicateUnitValues(t *testing.T) {
|
|
duration, err := ParseDurationString("1d4d2d")
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t,
|
|
(time.Hour*time.Duration(24))+
|
|
(time.Hour*time.Duration(24)*time.Duration(4))+
|
|
(time.Hour*time.Duration(24)*time.Duration(2)), duration)
|
|
}
|
|
|
|
func TestStandardizeDurationString_ShouldParseStringWithSpaces(t *testing.T) {
|
|
result, err := StandardizeDurationString("1d 1h 20m")
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, result, "24h1h20m")
|
|
}
|
|
|
|
func TestShouldTimeIntervalsMakeSense(t *testing.T) {
|
|
assert.Equal(t, Hour, time.Minute*60)
|
|
assert.Equal(t, Day, Hour*24)
|
|
assert.Equal(t, Week, Day*7)
|
|
assert.Equal(t, Year, Day*365)
|
|
assert.Equal(t, Month, Year/12)
|
|
}
|