Skip to content

Commit 83a31cb

Browse files
EmyrkEmyrk
committed
System outputs
1 parent cbb4502 commit 83a31cb

File tree

1 file changed

+41
-38
lines changed

1 file changed

+41
-38
lines changed

‎coderd/authzquery/system_test.go

Lines changed: 41 additions & 38 deletions
Original file line numberDiff line numberDiff line change
@@ -44,51 +44,51 @@ func (suite *MethodTestSuite) TestSystemFunctions() {
4444
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
4545
dbgen.WorkspaceBuild(t, db, database.WorkspaceBuild{})
4646
dbgen.WorkspaceBuild(t, db, database.WorkspaceBuild{})
47-
return methodCase(values(), asserts())
47+
return methodCase(values(), asserts(), nil)
4848
})
4949
})
5050
suite.Run("GetWorkspaceAgentByAuthToken", func() {
5151
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
52-
agent := dbgen.WorkspaceAgent(t, db, database.WorkspaceAgent{})
53-
return methodCase(values(agent.AuthToken), asserts())
52+
agt := dbgen.WorkspaceAgent(t, db, database.WorkspaceAgent{})
53+
return methodCase(values(agt.AuthToken), asserts(), values(agt))
5454
})
5555
})
5656
suite.Run("GetActiveUserCount", func() {
5757
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
58-
return methodCase(values(), asserts())
58+
return methodCase(values(), asserts(), values(int64(0)))
5959
})
6060
})
6161
suite.Run("GetUnexpiredLicenses", func() {
6262
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
63-
return methodCase(values(), asserts())
63+
return methodCase(values(), asserts(), nil)
6464
})
6565
})
6666
suite.Run("GetAuthorizationUserRoles", func() {
6767
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
6868
u := dbgen.User(t, db, database.User{})
69-
return methodCase(values(u.ID), asserts())
69+
return methodCase(values(u.ID), asserts(), nil)
7070
})
7171
})
7272
suite.Run("GetDERPMeshKey", func() {
7373
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
74-
return methodCase(values(), asserts())
74+
return methodCase(values(), asserts(), nil)
7575
})
7676
})
7777
suite.Run("InsertDERPMeshKey", func() {
7878
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
79-
return methodCase(values("value"), asserts())
79+
return methodCase(values("value"), asserts(), values())
8080
})
8181
})
8282
suite.Run("InsertDeploymentID", func() {
8383
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
84-
return methodCase(values("value"), asserts())
84+
return methodCase(values("value"), asserts(), values())
8585
})
8686
})
8787
suite.Run("InsertReplica", func() {
8888
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
8989
return methodCase(values(database.InsertReplicaParams{
9090
ID: uuid.New(),
91-
}), asserts())
91+
}), asserts(), nil)
9292
})
9393
})
9494
suite.Run("UpdateReplica", func() {
@@ -98,107 +98,109 @@ func (suite *MethodTestSuite) TestSystemFunctions() {
9898
return methodCase(values(database.UpdateReplicaParams{
9999
ID: replica.ID,
100100
DatabaseLatency: 100,
101-
}), asserts())
101+
}), asserts(), nil)
102102
})
103103
})
104104
suite.Run("DeleteReplicasUpdatedBefore", func() {
105105
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
106106
_, err := db.InsertReplica(context.Background(), database.InsertReplicaParams{ID: uuid.New(), UpdatedAt: time.Now()})
107107
require.NoError(t, err)
108-
return methodCase(values(time.Now().Add(time.Hour)), asserts())
108+
return methodCase(values(time.Now().Add(time.Hour)), asserts(), nil)
109109
})
110110
})
111111
suite.Run("GetReplicasUpdatedAfter", func() {
112112
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
113113
_, err := db.InsertReplica(context.Background(), database.InsertReplicaParams{ID: uuid.New(), UpdatedAt: time.Now()})
114114
require.NoError(t, err)
115-
return methodCase(values(time.Now().Add(time.Hour*-1)), asserts())
115+
return methodCase(values(time.Now().Add(time.Hour*-1)), asserts(), nil)
116116
})
117117
})
118118
suite.Run("GetUserCount", func() {
119119
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
120-
return methodCase(values(), asserts())
120+
return methodCase(values(), asserts(), values(0))
121121
})
122122
})
123123
suite.Run("GetTemplates", func() {
124124
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
125125
_ = dbgen.Template(t, db, database.Template{})
126-
return methodCase(values(), asserts())
126+
return methodCase(values(), asserts(), nil)
127127
})
128128
})
129129
suite.Run("UpdateWorkspaceBuildCostByID", func() {
130130
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
131131
b := dbgen.WorkspaceBuild(t, db, database.WorkspaceBuild{})
132+
o := b
133+
b.DailyCost = 10
132134
return methodCase(values(database.UpdateWorkspaceBuildCostByIDParams{
133135
ID: b.ID,
134136
DailyCost: 10,
135-
}), asserts())
137+
}), asserts(), values(o))
136138
})
137139
})
138140
suite.Run("InsertOrUpdateLastUpdateCheck", func() {
139141
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
140-
return methodCase(values("value"), asserts())
142+
return methodCase(values("value"), asserts(), nil)
141143
})
142144
})
143145
suite.Run("GetLastUpdateCheck", func() {
144146
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
145147
err := db.InsertOrUpdateLastUpdateCheck(context.Background(), "value")
146148
require.NoError(t, err)
147-
return methodCase(values(), asserts())
149+
return methodCase(values(), asserts(), nil)
148150
})
149151
})
150152
suite.Run("GetWorkspaceBuildsCreatedAfter", func() {
151153
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
152154
_ = dbgen.WorkspaceBuild(t, db, database.WorkspaceBuild{CreatedAt: time.Now().Add(-time.Hour)})
153-
return methodCase(values(time.Now()), asserts())
155+
return methodCase(values(time.Now()), asserts(), nil)
154156
})
155157
})
156158
suite.Run("GetWorkspaceAgentsCreatedAfter", func() {
157159
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
158160
_ = dbgen.WorkspaceAgent(t, db, database.WorkspaceAgent{CreatedAt: time.Now().Add(-time.Hour)})
159-
return methodCase(values(time.Now()), asserts())
161+
return methodCase(values(time.Now()), asserts(), nil)
160162
})
161163
})
162164
suite.Run("GetWorkspaceAppsCreatedAfter", func() {
163165
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
164166
_ = dbgen.WorkspaceApp(t, db, database.WorkspaceApp{CreatedAt: time.Now().Add(-time.Hour)})
165-
return methodCase(values(time.Now()), asserts())
167+
return methodCase(values(time.Now()), asserts(), nil)
166168
})
167169
})
168170
suite.Run("GetWorkspaceResourcesCreatedAfter", func() {
169171
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
170172
_ = dbgen.WorkspaceResource(t, db, database.WorkspaceResource{CreatedAt: time.Now().Add(-time.Hour)})
171-
return methodCase(values(time.Now()), asserts())
173+
return methodCase(values(time.Now()), asserts(), nil)
172174
})
173175
})
174176
suite.Run("GetWorkspaceResourceMetadataCreatedAfter", func() {
175177
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
176178
_ = dbgen.WorkspaceResourceMetadata(t, db, database.WorkspaceResourceMetadatum{})
177-
return methodCase(values(time.Now()), asserts())
179+
return methodCase(values(time.Now()), asserts(), nil)
178180
})
179181
})
180182
suite.Run("DeleteOldAgentStats", func() {
181183
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
182-
return methodCase(values(), asserts())
184+
return methodCase(values(), asserts(), nil)
183185
})
184186
})
185187
suite.Run("GetParameterSchemasCreatedAfter", func() {
186188
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
187189
_ = dbgen.ParameterSchema(t, db, database.ParameterSchema{CreatedAt: time.Now().Add(-time.Hour)})
188-
return methodCase(values(time.Now()), asserts())
190+
return methodCase(values(time.Now()), asserts(), nil)
189191
})
190192
})
191193
suite.Run("GetProvisionerJobsCreatedAfter", func() {
192194
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
193195
_ = dbgen.ProvisionerJob(t, db, database.ProvisionerJob{CreatedAt: time.Now().Add(-time.Hour)})
194-
return methodCase(values(time.Now()), asserts())
196+
return methodCase(values(time.Now()), asserts(), nil)
195197
})
196198
})
197199
suite.Run("InsertWorkspaceAgent", func() {
198200
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
199201
return methodCase(values(database.InsertWorkspaceAgentParams{
200202
ID: uuid.New(),
201-
}), asserts())
203+
}), asserts(), nil)
202204
})
203205
})
204206
suite.Run("InsertWorkspaceApp", func() {
@@ -207,30 +209,31 @@ func (suite *MethodTestSuite) TestSystemFunctions() {
207209
ID: uuid.New(),
208210
Health: database.WorkspaceAppHealthDisabled,
209211
SharingLevel: database.AppSharingLevelOwner,
210-
}), asserts())
212+
}), asserts(), nil)
211213
})
212214
})
213215
suite.Run("InsertWorkspaceResourceMetadata", func() {
214216
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
215217
return methodCase(values(database.InsertWorkspaceResourceMetadataParams{
216218
WorkspaceResourceID: uuid.New(),
217-
}), asserts())
219+
}), asserts(), nil)
218220
})
219221
})
220222
suite.Run("AcquireProvisionerJob", func() {
221223
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
222224
j := dbgen.ProvisionerJob(t, db, database.ProvisionerJob{
223225
StartedAt: sql.NullTime{Valid: false},
224226
})
225-
return methodCase(values(database.AcquireProvisionerJobParams{Types: []database.ProvisionerType{j.Provisioner}}), asserts())
227+
return methodCase(values(database.AcquireProvisionerJobParams{Types: []database.ProvisionerType{j.Provisioner}}),
228+
asserts(), nil)
226229
})
227230
})
228231
suite.Run("UpdateProvisionerJobWithCompleteByID", func() {
229232
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
230233
j := dbgen.ProvisionerJob(t, db, database.ProvisionerJob{})
231234
return methodCase(values(database.UpdateProvisionerJobWithCompleteByIDParams{
232235
ID: j.ID,
233-
}), asserts())
236+
}), asserts(), nil)
234237
})
235238
})
236239
suite.Run("UpdateProvisionerJobByID", func() {
@@ -239,7 +242,7 @@ func (suite *MethodTestSuite) TestSystemFunctions() {
239242
return methodCase(values(database.UpdateProvisionerJobByIDParams{
240243
ID: j.ID,
241244
UpdatedAt: time.Now(),
242-
}), asserts())
245+
}), asserts(), nil)
243246
})
244247
})
245248
suite.Run("InsertProvisionerJob", func() {
@@ -249,30 +252,30 @@ func (suite *MethodTestSuite) TestSystemFunctions() {
249252
Provisioner: database.ProvisionerTypeEcho,
250253
StorageMethod: database.ProvisionerStorageMethodFile,
251254
Type: database.ProvisionerJobTypeWorkspaceBuild,
252-
}), asserts())
255+
}), asserts(), nil)
253256
})
254257
})
255258
suite.Run("InsertProvisionerJobLogs", func() {
256259
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
257260
j := dbgen.ProvisionerJob(t, db, database.ProvisionerJob{})
258261
return methodCase(values(database.InsertProvisionerJobLogsParams{
259262
JobID: j.ID,
260-
}), asserts())
263+
}), asserts(), nil)
261264
})
262265
})
263266
suite.Run("InsertProvisionerDaemon", func() {
264267
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
265268
return methodCase(values(database.InsertProvisionerDaemonParams{
266269
ID: uuid.New(),
267-
}), asserts())
270+
}), asserts(), nil)
268271
})
269272
})
270273
suite.Run("InsertTemplateVersionParameter", func() {
271274
suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
272275
v := dbgen.TemplateVersion(t, db, database.TemplateVersion{})
273276
return methodCase(values(database.InsertTemplateVersionParameterParams{
274277
TemplateVersionID: v.ID,
275-
}), asserts())
278+
}), asserts(), nil)
276279
})
277280
})
278281
suite.Run("InsertWorkspaceResource", func() {
@@ -281,7 +284,7 @@ func (suite *MethodTestSuite) TestSystemFunctions() {
281284
return methodCase(values(database.InsertWorkspaceResourceParams{
282285
ID: r.ID,
283286
Transition: database.WorkspaceTransitionStart,
284-
}), asserts())
287+
}), asserts(), nil)
285288
})
286289
})
287290
suite.Run("InsertParameterSchema", func() {
@@ -291,7 +294,7 @@ func (suite *MethodTestSuite) TestSystemFunctions() {
291294
DefaultSourceScheme: database.ParameterSourceSchemeNone,
292295
DefaultDestinationScheme: database.ParameterDestinationSchemeNone,
293296
ValidationTypeSystem: database.ParameterTypeSystemNone,
294-
}), asserts())
297+
}), asserts(), nil)
295298
})
296299
})
297300
}

0 commit comments

Comments
 (0)