aboutsummaryrefslogtreecommitdiff
path: root/internal/db/users_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/db/users_test.go')
-rw-r--r--internal/db/users_test.go262
1 files changed, 262 insertions, 0 deletions
diff --git a/internal/db/users_test.go b/internal/db/users_test.go
new file mode 100644
index 00000000..e4d8fb86
--- /dev/null
+++ b/internal/db/users_test.go
@@ -0,0 +1,262 @@
+// Copyright 2020 The Gogs Authors. All rights reserved.
+// Use of this source code is governed by a MIT-style
+// license that can be found in the LICENSE file.
+
+package db
+
+import (
+ "testing"
+ "time"
+
+ "github.com/jinzhu/gorm"
+ "github.com/stretchr/testify/assert"
+
+ "gogs.io/gogs/internal/errutil"
+)
+
+func Test_users(t *testing.T) {
+ if testing.Short() {
+ t.Skip()
+ }
+
+ t.Parallel()
+
+ tables := []interface{}{new(User), new(EmailAddress)}
+ db := &users{
+ DB: initTestDB(t, "users", tables...),
+ }
+
+ for _, tc := range []struct {
+ name string
+ test func(*testing.T, *users)
+ }{
+ {"Authenticate", test_users_Authenticate},
+ {"Create", test_users_Create},
+ {"GetByEmail", test_users_GetByEmail},
+ {"GetByID", test_users_GetByID},
+ {"GetByUsername", test_users_GetByUsername},
+ } {
+ t.Run(tc.name, func(t *testing.T) {
+ t.Cleanup(func() {
+ err := clearTables(t, db.DB, tables...)
+ if err != nil {
+ t.Fatal(err)
+ }
+ })
+ tc.test(t, db)
+ })
+ }
+}
+
+// TODO: Only local account is tested, tests for external account will be added
+// along with addressing https://github.com/gogs/gogs/issues/6115.
+func test_users_Authenticate(t *testing.T, db *users) {
+ password := "pa$$word"
+ alice, err := db.Create(CreateUserOpts{
+ Name: "alice",
+ Email: "alice@example.com",
+ Password: password,
+ })
+ if err != nil {
+ t.Fatal(err)
+ }
+
+ t.Run("user not found", func(t *testing.T) {
+ _, err := db.Authenticate("bob", password, -1)
+ expErr := ErrUserNotExist{args: map[string]interface{}{"login": "bob"}}
+ assert.Equal(t, expErr, err)
+ })
+
+ t.Run("invalid password", func(t *testing.T) {
+ _, err := db.Authenticate(alice.Name, "bad_password", -1)
+ expErr := ErrUserNotExist{args: map[string]interface{}{"userID": alice.ID, "name": alice.Name}}
+ assert.Equal(t, expErr, err)
+ })
+
+ t.Run("via email and password", func(t *testing.T) {
+ user, err := db.Authenticate(alice.Email, password, -1)
+ if err != nil {
+ t.Fatal(err)
+ }
+ assert.Equal(t, alice.Name, user.Name)
+ })
+
+ t.Run("via username and password", func(t *testing.T) {
+ user, err := db.Authenticate(alice.Name, password, -1)
+ if err != nil {
+ t.Fatal(err)
+ }
+ assert.Equal(t, alice.Name, user.Name)
+ })
+}
+
+func test_users_Create(t *testing.T, db *users) {
+ alice, err := db.Create(CreateUserOpts{
+ Name: "alice",
+ Email: "alice@example.com",
+ Activated: true,
+ })
+ if err != nil {
+ t.Fatal(err)
+ }
+
+ t.Run("name not allowed", func(t *testing.T) {
+ _, err := db.Create(CreateUserOpts{
+ Name: "-",
+ })
+ expErr := ErrNameNotAllowed{args: errutil.Args{"reason": "reserved", "name": "-"}}
+ assert.Equal(t, expErr, err)
+ })
+
+ t.Run("name already exists", func(t *testing.T) {
+ _, err := db.Create(CreateUserOpts{
+ Name: alice.Name,
+ })
+ expErr := ErrUserAlreadyExist{args: errutil.Args{"name": alice.Name}}
+ assert.Equal(t, expErr, err)
+ })
+
+ t.Run("email already exists", func(t *testing.T) {
+ _, err := db.Create(CreateUserOpts{
+ Name: "bob",
+ Email: alice.Email,
+ })
+ expErr := ErrEmailAlreadyUsed{args: errutil.Args{"email": alice.Email}}
+ assert.Equal(t, expErr, err)
+ })
+
+ user, err := db.GetByUsername(alice.Name)
+ if err != nil {
+ t.Fatal(err)
+ }
+ assert.Equal(t, gorm.NowFunc().Format(time.RFC3339), user.Created.Format(time.RFC3339))
+ assert.Equal(t, gorm.NowFunc().Format(time.RFC3339), user.Updated.Format(time.RFC3339))
+}
+
+func test_users_GetByEmail(t *testing.T, db *users) {
+ t.Run("empty email", func(t *testing.T) {
+ _, err := db.GetByEmail("")
+ expErr := ErrUserNotExist{args: errutil.Args{"email": ""}}
+ assert.Equal(t, expErr, err)
+ })
+
+ t.Run("ignore organization", func(t *testing.T) {
+ // TODO: Use Orgs.Create to replace SQL hack when the method is available.
+ org, err := db.Create(CreateUserOpts{
+ Name: "gogs",
+ Email: "gogs@exmaple.com",
+ })
+ if err != nil {
+ t.Fatal(err)
+ }
+
+ err = db.Exec(`UPDATE user SET type = ? WHERE id = ?`, UserOrganization, org.ID).Error
+ if err != nil {
+ t.Fatal(err)
+ }
+
+ _, err = db.GetByEmail(org.Email)
+ expErr := ErrUserNotExist{args: errutil.Args{"email": org.Email}}
+ assert.Equal(t, expErr, err)
+ })
+
+ t.Run("by primary email", func(t *testing.T) {
+ alice, err := db.Create(CreateUserOpts{
+ Name: "alice",
+ Email: "alice@exmaple.com",
+ })
+ if err != nil {
+ t.Fatal(err)
+ }
+
+ _, err = db.GetByEmail(alice.Email)
+ expErr := ErrUserNotExist{args: errutil.Args{"email": alice.Email}}
+ assert.Equal(t, expErr, err)
+
+ // Mark user as activated
+ // TODO: Use UserEmails.Verify to replace SQL hack when the method is available.
+ err = db.Exec(`UPDATE user SET is_active = ? WHERE id = ?`, true, alice.ID).Error
+ if err != nil {
+ t.Fatal(err)
+ }
+
+ user, err := db.GetByEmail(alice.Email)
+ if err != nil {
+ t.Fatal(err)
+ }
+ assert.Equal(t, alice.Name, user.Name)
+ })
+
+ t.Run("by secondary email", func(t *testing.T) {
+ bob, err := db.Create(CreateUserOpts{
+ Name: "bob",
+ Email: "bob@example.com",
+ })
+ if err != nil {
+ t.Fatal(err)
+ }
+
+ // TODO: Use UserEmails.Create to replace SQL hack when the method is available.
+ email2 := "bob2@exmaple.com"
+ err = db.Exec(`INSERT INTO email_address (uid, email) VALUES (?, ?)`, bob.ID, email2).Error
+ if err != nil {
+ t.Fatal(err)
+ }
+
+ _, err = db.GetByEmail(email2)
+ expErr := ErrUserNotExist{args: errutil.Args{"email": email2}}
+ assert.Equal(t, expErr, err)
+
+ // TODO: Use UserEmails.Verify to replace SQL hack when the method is available.
+ err = db.Exec(`UPDATE email_address SET is_activated = ? WHERE email = ?`, true, email2).Error
+ if err != nil {
+ t.Fatal(err)
+ }
+
+ user, err := db.GetByEmail(email2)
+ if err != nil {
+ t.Fatal(err)
+ }
+ assert.Equal(t, bob.Name, user.Name)
+ })
+}
+
+func test_users_GetByID(t *testing.T, db *users) {
+ alice, err := db.Create(CreateUserOpts{
+ Name: "alice",
+ Email: "alice@exmaple.com",
+ })
+ if err != nil {
+ t.Fatal(err)
+ }
+
+ user, err := db.GetByID(alice.ID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ assert.Equal(t, alice.Name, user.Name)
+
+ _, err = db.GetByID(404)
+ expErr := ErrUserNotExist{args: errutil.Args{"userID": int64(404)}}
+ assert.Equal(t, expErr, err)
+}
+
+func test_users_GetByUsername(t *testing.T, db *users) {
+ alice, err := db.Create(CreateUserOpts{
+ Name: "alice",
+ Email: "alice@exmaple.com",
+ })
+ if err != nil {
+ t.Fatal(err)
+ }
+
+ user, err := db.GetByUsername(alice.Name)
+ if err != nil {
+ t.Fatal(err)
+ }
+ assert.Equal(t, alice.Name, user.Name)
+
+ _, err = db.GetByUsername("bad_username")
+ expErr := ErrUserNotExist{args: errutil.Args{"name": "bad_username"}}
+ assert.Equal(t, expErr, err)
+}
Powered by cgit, Themed by Ted Yin.