@@ -48,12 +48,14 @@ func TestCache_TemplateUsers(t *testing.T) {
48
48
uniqueUsers int
49
49
}
50
50
tests := []struct {
51
- name string
52
- args args
53
- want want
51
+ name string
52
+ args args
53
+ tplWant want
54
+ // dauWant is optional
55
+ dauWant []codersdk.DAUEntry
54
56
tzOffset int
55
57
}{
56
- {name :"empty" ,args :args {},want :want {nil ,0 }},
58
+ {name :"empty" ,args :args {},tplWant :want {nil ,0 }},
57
59
{
58
60
name :"one hole" ,
59
61
args :args {
@@ -62,7 +64,7 @@ func TestCache_TemplateUsers(t *testing.T) {
62
64
statRow (zebra ,dateH (2022 ,8 ,30 ,0 )),
63
65
},
64
66
},
65
- want :want {[]codersdk.DAUEntry {
67
+ tplWant :want {[]codersdk.DAUEntry {
66
68
{
67
69
Date :date (2022 ,8 ,27 ),
68
70
Amount :1 ,
@@ -90,7 +92,7 @@ func TestCache_TemplateUsers(t *testing.T) {
90
92
statRow (zebra ,dateH (2022 ,8 ,29 ,0 )),
91
93
},
92
94
},
93
- want :want {[]codersdk.DAUEntry {
95
+ tplWant :want {[]codersdk.DAUEntry {
94
96
{
95
97
Date :date (2022 ,8 ,27 ),
96
98
Amount :1 ,
@@ -116,7 +118,7 @@ func TestCache_TemplateUsers(t *testing.T) {
116
118
statRow (tiger ,dateH (2022 ,1 ,7 ,0 )),
117
119
},
118
120
},
119
- want :want {[]codersdk.DAUEntry {
121
+ tplWant :want {[]codersdk.DAUEntry {
120
122
{
121
123
Date :date (2022 ,1 ,1 ),
122
124
Amount :2 ,
@@ -159,7 +161,13 @@ func TestCache_TemplateUsers(t *testing.T) {
159
161
statRow (tiger ,dateH (2022 ,1 ,2 ,0 )),
160
162
},
161
163
},
162
- want :want {[]codersdk.DAUEntry {
164
+ tplWant :want {[]codersdk.DAUEntry {
165
+ {
166
+ Date :date (2022 ,1 ,2 ),
167
+ Amount :2 ,
168
+ },
169
+ },2 },
170
+ dauWant : []codersdk.DAUEntry {
163
171
{
164
172
Date :date (2022 ,1 ,1 ),
165
173
Amount :2 ,
@@ -168,7 +176,7 @@ func TestCache_TemplateUsers(t *testing.T) {
168
176
Date :date (2022 ,1 ,2 ),
169
177
Amount :2 ,
170
178
},
171
- },2 },
179
+ },
172
180
},
173
181
{
174
182
name :"tzOffsetPreviousDay" ,
@@ -181,11 +189,17 @@ func TestCache_TemplateUsers(t *testing.T) {
181
189
statRow (tiger ,dateH (2022 ,1 ,2 ,0 )),
182
190
},
183
191
},
184
- want : want { []codersdk.DAUEntry {
192
+ dauWant : []codersdk.DAUEntry {
185
193
{
186
194
Date :date (2022 ,1 ,1 ),
187
195
Amount :2 ,
188
196
},
197
+ },
198
+ tplWant :want {[]codersdk.DAUEntry {
199
+ {
200
+ Date :date (2022 ,1 ,2 ),
201
+ Amount :2 ,
202
+ },
189
203
},2 },
190
204
},
191
205
}
@@ -223,11 +237,18 @@ func TestCache_TemplateUsers(t *testing.T) {
223
237
gotUniqueUsers ,ok := cache .TemplateUniqueUsers (template .ID )
224
238
require .True (t ,ok )
225
239
240
+ if tt .dauWant != nil {
241
+ _ ,dauResponse ,ok := cache .DeploymentDAUs (tt .tzOffset )
242
+ require .True (t ,ok )
243
+ require .Equal (t ,tt .dauWant ,dauResponse .Entries )
244
+ }
245
+
226
246
offset ,gotEntries ,ok := cache .TemplateDAUs (template .ID ,tt .tzOffset )
227
247
require .True (t ,ok )
228
- require .Equal (t ,offset ,tt .tzOffset )
229
- require .Equal (t ,tt .want .entries ,gotEntries .Entries )
230
- require .Equal (t ,tt .want .uniqueUsers ,gotUniqueUsers )
248
+ // Template only supports 0 offset.
249
+ require .Equal (t ,0 ,offset )
250
+ require .Equal (t ,tt .tplWant .entries ,gotEntries .Entries )
251
+ require .Equal (t ,tt .tplWant .uniqueUsers ,gotUniqueUsers )
231
252
})
232
253
}
233
254
}