timeexpression/hour_unit_test.go at master · mingforpc/timeexpression · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
package timeexpression

import (
	"errors"
	"github.com/stretchr/testify/assert"
	"strconv"
	"testing"
)

func TestHourUnit_new(t *testing.T) {
	testDatas := []struct {
		exp  string
		err  error
		hour int
		min  int
		sec  int
	}{
		{
			exp:  "22:33:44",
			err:  nil,
			hour: 22,
			min:  33,
			sec:  44,
		},
		{
			exp: "2233:44",
			err: ErrHourUnitFormat,
		},
		{
			exp: "22:60:44",
			err: ErrHourUnitFormat,
		},
		{
			exp: "22:33:60",
			err: ErrHourUnitFormat,
		},
		{
			exp: "25:33:44",
			err: ErrHourUnitFormat,
		},
		{
			exp: "a:33:44",
			err: &strconv.NumError{
				Func: "Atoi",
				Num:  "a",
				Err:  errors.New("invalid syntax"),
			},
		},
		{
			exp: "23:a:44",
			err: &strconv.NumError{
				Func: "Atoi",
				Num:  "a",
				Err:  errors.New("invalid syntax"),
			},
		},
		{
			exp: "23:33:a",
			err: &strconv.NumError{
				Func: "Atoi",
				Num:  "a",
				Err:  errors.New("invalid syntax"),
			},
		},
	}

	for _, data := range testDatas {
		unit, err := newHourTimeUnit(data.exp)
		if err != nil {
			assert.EqualError(t, data.err, err.Error())
			continue
		}

		assert.Equal(t, data.hour, unit.Hour)
		assert.Equal(t, data.min, unit.Minute)
		assert.Equal(t, data.sec, unit.Sec)
	}
}

func TestHourUnit_before(t *testing.T) {
	unitBefore, err := newHourTimeUnit("22:33:44")
	if err != nil {
		panic(err)
	}
	unitAfter1, err := newHourTimeUnit("22:33:55")
	if err != nil {
		panic(err)
	}
	unitAfter2, err := newHourTimeUnit("22:33:44")
	if err != nil {
		panic(err)
	}
	unitAfter3, err := newHourTimeUnit("22:33:33")
	if err != nil {
		panic(err)
	}

	assert.True(t, unitBefore.before(unitAfter1))
	assert.True(t, unitBefore.before(unitAfter2))
	assert.False(t, unitBefore.before(unitAfter3))
}

func TestHourUnit_after(t *testing.T) {
	unitBefore, err := newHourTimeUnit("22:33:44")
	if err != nil {
		panic(err)
	}
	unitAfter1, err := newHourTimeUnit("22:33:55")
	if err != nil {
		panic(err)
	}
	unitAfter2, err := newHourTimeUnit("22:33:44")
	if err != nil {
		panic(err)
	}
	unitAfter3, err := newHourTimeUnit("22:33:33")
	if err != nil {
		panic(err)
	}

	assert.False(t, unitBefore.after(unitAfter1))
	assert.False(t, unitBefore.after(unitAfter2))
	assert.True(t, unitBefore.after(unitAfter3))
}