config/duration_test.go at master · tokenized/config · 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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
package config

import (
	"encoding/json"
	"fmt"
	"testing"
	"time"
)

type DurationTestStruct struct {
	Duration1  Duration `json:"duration_1"`
	OtherField string   `json:"other_field"`
	Duration2  Duration `json:"duration_2"`
}

type DurationTestStringStruct struct {
	Duration1  string `json:"duration_1"`
	OtherField string `json:"other_field"`
	Duration2  string `json:"duration_2"`
}

func Test_Duration(t *testing.T) {
	var tests = []struct {
		Struct          DurationTestStruct
		duration1       time.Duration
		duration2       time.Duration
		duration1String string
		duration2String string
	}{
		{
			Struct: DurationTestStruct{
				Duration1:  NewDuration(time.Second * 5),
				OtherField: "test1",
				Duration2:  NewDuration(time.Minute * 30),
			},
			duration1:       time.Second * 5,
			duration2:       time.Minute * 30,
			duration1String: "5s",
			duration2String: "30m0s",
		},
		{
			Struct: DurationTestStruct{
				Duration1:  NewDuration(time.Second),
				OtherField: "test2",
				Duration2:  NewDuration(time.Hour * 4),
			},
			duration1:       time.Second,
			duration2:       time.Hour * 4,
			duration1String: "1s",
			duration2String: "4h0m0s",
		},
	}

	for i, test := range tests {
		js, _ := json.MarshalIndent(test.Struct, "", "  ")
		t.Logf("JSON (%d) : %s", i, js)

		jsTest := &DurationTestStruct{}
		if err := json.Unmarshal(js, jsTest); err != nil {
			t.Fatalf("Failed to unmarshal json : %s", err)
		}

		if jsTest.Duration1 != test.Struct.Duration1 {
			t.Errorf("Wrong duration 1 (%d) : got %s, want %s", i, jsTest.Duration1,
				test.Struct.Duration1)
		}

		if jsTest.Duration1.Duration != test.duration1 {
			t.Errorf("Wrong duration 1 (%d) : got %s, want %s", i, jsTest.Duration1,
				test.duration1)
		}

		if jsTest.OtherField != test.Struct.OtherField {
			t.Errorf("Wrong other field (%d) : got %s, want %s", i, jsTest.OtherField,
				test.Struct.OtherField)
		}

		if jsTest.Duration2 != test.Struct.Duration2 {
			t.Errorf("Wrong duration 2 (%d) : got %s, want %s", i, jsTest.Duration2,
				test.Struct.Duration2)
		}

		if jsTest.Duration2.Duration != test.duration2 {
			t.Errorf("Wrong duration 2 (%d) : got %s, want %s", i, jsTest.Duration2,
				test.duration2)
		}

		stringTest := &DurationTestStringStruct{}
		if err := json.Unmarshal(js, stringTest); err != nil {
			t.Fatalf("Failed to unmarshal json to string test : %s", err)
		}

		if stringTest.Duration1 != test.duration1String {
			t.Errorf("Wrong duration 1 string (%d) : got %s, want %s", i, stringTest.Duration1,
				test.duration1String)
		}

		if stringTest.OtherField != test.Struct.OtherField {
			t.Errorf("Wrong other field (%d) : got %s, want %s", i, stringTest.OtherField,
				test.Struct.OtherField)
		}

		if stringTest.Duration2 != test.duration2String {
			t.Errorf("Wrong duration 2 string (%d) : got %s, want %s", i, stringTest.Duration2,
				test.duration2String)
		}

		// Test conversion to string
		duration1String := fmt.Sprintf("%s", test.Struct.Duration1)
		if duration1String != test.duration1String {
			t.Errorf("Wrong duration 1 string (%d) : got %s, want %s", i, duration1String,
				test.duration1String)
		}
		t.Logf("Duration 1 String (%d) : %s", i, duration1String)

		duration2String := fmt.Sprintf("%s", test.Struct.Duration2)
		if duration2String != test.duration2String {
			t.Errorf("Wrong duration 1 string (%d) : got %s, want %s", i, duration2String,
				test.duration2String)
		}
		t.Logf("Duration 2 String (%d) : %s", i, duration2String)

		// Test conversion from string
		var duration1 Duration
		if err := duration1.UnmarshalText([]byte(test.duration1String)); err != nil {
			t.Fatalf("Failed to unmarshal duration 1 to text : %s", err)
		}
		if duration1 != test.Struct.Duration1 {
			t.Errorf("Wrong duration 1 (%d) : got %s, want %s", i, duration1,
				test.Struct.Duration1)
		}

		var duration2 Duration
		if err := duration2.UnmarshalText([]byte(test.duration2String)); err != nil {
			t.Fatalf("Failed to unmarshal duration 2 to text : %s", err)
		}
		if duration2 != test.Struct.Duration2 {
			t.Errorf("Wrong duration 2 (%d) : got %s, want %s", i, duration2,
				test.Struct.Duration2)
		}
	}
}