Skip to content

Commit 2c768f5

Browse files
committed
add more test asserts
1 parent ac86171 commit 2c768f5

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

cmd/admin_user_create_test.go

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,11 +70,13 @@ func TestAdminUserCreate(t *testing.T) {
7070
// no generated access token
7171
reset()
7272
assert.NoError(t, createUser("u", "--random-password"))
73+
assert.Equal(t, 1, unittest.GetCount(t, &user_model.User{}))
7374
assert.Equal(t, 0, unittest.GetCount(t, &auth_model.AccessToken{}))
7475

7576
// using "--access-token" only means "all" access
7677
reset()
7778
assert.NoError(t, createUser("u", "--random-password --access-token"))
79+
assert.Equal(t, 1, unittest.GetCount(t, &user_model.User{}))
7880
assert.Equal(t, 1, unittest.GetCount(t, &auth_model.AccessToken{}))
7981
accessToken := unittest.AssertExistsAndLoadBean(t, &auth_model.AccessToken{Name: "gitea-admin"})
8082
hasScopes, err := accessToken.Scope.HasScope(auth_model.AccessTokenScopeWriteAdmin, auth_model.AccessTokenScopeWriteRepository)
@@ -84,6 +86,7 @@ func TestAdminUserCreate(t *testing.T) {
8486
// using "--access-token" with name & scopes
8587
reset()
8688
assert.NoError(t, createUser("u", "--random-password --access-token --access-token-name new-token-name --access-token-scopes read:issue,read:user"))
89+
assert.Equal(t, 1, unittest.GetCount(t, &user_model.User{}))
8790
assert.Equal(t, 1, unittest.GetCount(t, &auth_model.AccessToken{}))
8891
accessToken = unittest.AssertExistsAndLoadBean(t, &auth_model.AccessToken{Name: "new-token-name"})
8992
hasScopes, err = accessToken.Scope.HasScope(auth_model.AccessTokenScopeReadIssue, auth_model.AccessTokenScopeReadUser)
@@ -96,16 +99,22 @@ func TestAdminUserCreate(t *testing.T) {
9699
// using "--access-token-name" without "--access-token"
97100
reset()
98101
err = createUser("u", "--random-password --access-token-name new-token-name")
102+
assert.Equal(t, 0, unittest.GetCount(t, &user_model.User{}))
103+
assert.Equal(t, 0, unittest.GetCount(t, &auth_model.AccessToken{}))
99104
assert.ErrorContains(t, err, "access-token-name and access-token-scopes flags are only valid when access-token flag is set")
100105

101106
// using "--access-token-scopes" without "--access-token"
102107
reset()
103108
err = createUser("u", "--random-password --access-token-scopes read:issue")
109+
assert.Equal(t, 0, unittest.GetCount(t, &user_model.User{}))
110+
assert.Equal(t, 0, unittest.GetCount(t, &auth_model.AccessToken{}))
104111
assert.ErrorContains(t, err, "access-token-name and access-token-scopes flags are only valid when access-token flag is set")
105112

106113
// empty permission
107114
reset()
108115
err = createUser("u", "--random-password --access-token --access-token-scopes public-only")
116+
assert.Equal(t, 0, unittest.GetCount(t, &user_model.User{}))
117+
assert.Equal(t, 0, unittest.GetCount(t, &auth_model.AccessToken{}))
109118
assert.ErrorContains(t, err, "access token does not have any permission")
110119
})
111120
}

0 commit comments

Comments
 (0)