Skip to content

Commit acf214c

Browse files
EmyrkEmyrk
committed
touch ups
1 parent 5253d11 commit acf214c

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

‎coderd/database/dbauthz/dbauthz_test.go

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1259,9 +1259,9 @@ func (s *MethodTestSuite) TestUser() {
12591259
check.Args(arg, emptyPreparedAuthorized{}).Asserts()
12601260
}))
12611261
s.Run("DeleteAPIKeysByUserID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
1262-
u := testutil.Fake(s.T(), faker, database.User{})
1263-
dbm.EXPECT().DeleteAPIKeysByUserID(gomock.Any(), u.ID).Return(nil).AnyTimes()
1264-
check.Args(u.ID).Asserts(rbac.ResourceApiKey.WithOwner(u.ID.String()), policy.ActionDelete).Returns()
1262+
key := testutil.Fake(s.T(), faker, database.APIKey{})
1263+
dbm.EXPECT().DeleteAPIKeysByUserID(gomock.Any(), key.UserID).Return(nil).AnyTimes()
1264+
check.Args(key.UserID).Asserts(key, policy.ActionDelete).Returns()
12651265
}))
12661266
s.Run("GetQuotaAllowanceForUser", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
12671267
u := testutil.Fake(s.T(), faker, database.User{})
@@ -1277,7 +1277,7 @@ func (s *MethodTestSuite) TestUser() {
12771277
}))
12781278
s.Run("GetUserByEmailOrUsername", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
12791279
u := testutil.Fake(s.T(), faker, database.User{})
1280-
arg := database.GetUserByEmailOrUsernameParams{Username: u.Username, Email: u.Email}
1280+
arg := database.GetUserByEmailOrUsernameParams{Email: u.Email}
12811281
dbm.EXPECT().GetUserByEmailOrUsername(gomock.Any(), arg).Return(u, nil).AnyTimes()
12821282
check.Args(arg).Asserts(u, policy.ActionRead).Returns(u)
12831283
}))
@@ -1301,7 +1301,7 @@ func (s *MethodTestSuite) TestUser() {
13011301
}))
13021302
s.Run("InsertUser", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) {
13031303
arg := database.InsertUserParams{ID: uuid.New(), LoginType: database.LoginTypePassword, RBACRoles: []string{}}
1304-
dbm.EXPECT().InsertUser(gomock.Any(), arg).Return(database.User{}, nil).AnyTimes()
1304+
dbm.EXPECT().InsertUser(gomock.Any(), arg).Return(database.User{ID: arg.ID, LoginType: arg.LoginType}, nil).AnyTimes()
13051305
check.Args(arg).Asserts(rbac.ResourceAssignRole, policy.ActionAssign, rbac.ResourceUser, policy.ActionCreate)
13061306
}))
13071307
s.Run("InsertUserLink", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
@@ -1332,7 +1332,7 @@ func (s *MethodTestSuite) TestUser() {
13321332
}))
13331333
s.Run("UpdateUserHashedOneTimePasscode", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
13341334
u := testutil.Fake(s.T(), faker, database.User{})
1335-
arg := database.UpdateUserHashedOneTimePasscodeParams{ID: u.ID, HashedOneTimePasscode: []byte{}, OneTimePasscodeExpiresAt: sql.NullTime{Time: u.CreatedAt, Valid: true}}
1335+
arg := database.UpdateUserHashedOneTimePasscodeParams{ID: u.ID}
13361336
dbm.EXPECT().UpdateUserHashedOneTimePasscode(gomock.Any(), arg).Return(nil).AnyTimes()
13371337
check.Args(arg).Asserts(rbac.ResourceSystem, policy.ActionUpdate).Returns()
13381338
}))

0 commit comments

Comments
 (0)