diff options
Diffstat (limited to 'internal/db/users_test.go')
| -rw-r--r-- | internal/db/users_test.go | 175 |
1 files changed, 174 insertions, 1 deletions
diff --git a/internal/db/users_test.go b/internal/db/users_test.go index d83ffe50..8b2e7e59 100644 --- a/internal/db/users_test.go +++ b/internal/db/users_test.go @@ -116,6 +116,12 @@ func TestUsers(t *testing.T) { {"SearchByName", usersSearchByName}, {"Update", usersUpdate}, {"UseCustomAvatar", usersUseCustomAvatar}, + {"AddEmail", usersAddEmail}, + {"GetEmail", usersGetEmail}, + {"ListEmails", usersListEmails}, + {"MarkEmailActivated", usersMarkEmailActivated}, + {"MarkEmailPrimary", usersMarkEmailPrimary}, + {"DeleteEmail", usersDeleteEmail}, {"Follow", usersFollow}, {"IsFollowing", usersIsFollowing}, {"Unfollow", usersUnfollow}, @@ -1100,7 +1106,19 @@ func usersUpdate(t *testing.T, db *users) { }) t.Run("update email but already used", func(t *testing.T) { - // todo + bob, err := db.Create( + ctx, + "bob", + "bob@example.com", + CreateUserOptions{ + Activated: true, + }, + ) + require.NoError(t, err) + + got := db.Update(ctx, alice.ID, UpdateUserOptions{Email: &bob.Email}) + want := ErrEmailAlreadyUsed{args: errutil.Args{"email": bob.Email}} + assert.Equal(t, want, got) }) loginSource := int64(1) @@ -1204,6 +1222,161 @@ func TestIsUsernameAllowed(t *testing.T) { } } +func usersAddEmail(t *testing.T, db *users) { + ctx := context.Background() + + t.Run("multiple users can add the same unverified email", func(t *testing.T) { + alice, err := db.Create(ctx, "alice", "unverified@example.com", CreateUserOptions{}) + require.NoError(t, err) + err = db.AddEmail(ctx, alice.ID+1, "unverified@example.com", false) + require.NoError(t, err) + }) + + t.Run("only one user can add the same verified email", func(t *testing.T) { + bob, err := db.Create(ctx, "bob", "verified@example.com", CreateUserOptions{Activated: true}) + require.NoError(t, err) + got := db.AddEmail(ctx, bob.ID+1, "verified@example.com", true) + want := ErrEmailAlreadyUsed{args: errutil.Args{"email": "verified@example.com"}} + require.Equal(t, want, got) + }) +} + +func usersGetEmail(t *testing.T, db *users) { + ctx := context.Background() + + const testUserID = 1 + const testEmail = "alice@example.com" + _, err := db.GetEmail(ctx, testUserID, testEmail, false) + wantErr := ErrEmailNotExist{ + args: errutil.Args{ + "email": testEmail, + }, + } + assert.Equal(t, wantErr, err) + + err = db.AddEmail(ctx, testUserID, testEmail, false) + require.NoError(t, err) + got, err := db.GetEmail(ctx, testUserID, testEmail, false) + require.NoError(t, err) + assert.Equal(t, testEmail, got.Email) + + // Should not return if we ask for a different user + _, err = db.GetEmail(ctx, testUserID+1, testEmail, false) + assert.Equal(t, wantErr, err) + + // Should not return if we only want activated emails + _, err = db.GetEmail(ctx, testUserID, testEmail, true) + assert.Equal(t, wantErr, err) + + err = db.MarkEmailActivated(ctx, testUserID, testEmail) + require.NoError(t, err) + got, err = db.GetEmail(ctx, testUserID, testEmail, true) + require.NoError(t, err) + assert.Equal(t, testEmail, got.Email) +} + +func usersListEmails(t *testing.T, db *users) { + ctx := context.Background() + + t.Run("list emails with primary email", func(t *testing.T) { + alice, err := db.Create(ctx, "alice", "alice@example.com", CreateUserOptions{}) + require.NoError(t, err) + err = db.AddEmail(ctx, alice.ID, "alice2@example.com", true) + require.NoError(t, err) + err = db.MarkEmailPrimary(ctx, alice.ID, "alice2@example.com") + require.NoError(t, err) + + emails, err := db.ListEmails(ctx, alice.ID) + require.NoError(t, err) + got := make([]string, 0, len(emails)) + for _, email := range emails { + got = append(got, email.Email) + } + want := []string{"alice2@example.com", "alice@example.com"} + assert.Equal(t, want, got) + }) + + t.Run("list emails without primary email", func(t *testing.T) { + bob, err := db.Create(ctx, "bob", "bob@example.com", CreateUserOptions{}) + require.NoError(t, err) + err = db.AddEmail(ctx, bob.ID, "bob2@example.com", false) + require.NoError(t, err) + + emails, err := db.ListEmails(ctx, bob.ID) + require.NoError(t, err) + got := make([]string, 0, len(emails)) + for _, email := range emails { + got = append(got, email.Email) + } + want := []string{"bob2@example.com", "bob@example.com"} + assert.Equal(t, want, got) + }) +} + +func usersMarkEmailActivated(t *testing.T, db *users) { + ctx := context.Background() + + alice, err := db.Create(ctx, "alice", "alice@example.com", CreateUserOptions{}) + require.NoError(t, err) + + err = db.AddEmail(ctx, alice.ID, "alice2@example.com", false) + require.NoError(t, err) + err = db.MarkEmailActivated(ctx, alice.ID, "alice2@example.com") + require.NoError(t, err) + + gotEmail, err := db.GetEmail(ctx, alice.ID, "alice2@example.com", true) + require.NoError(t, err) + assert.True(t, gotEmail.IsActivated) + + gotAlice, err := db.GetByID(ctx, alice.ID) + require.NoError(t, err) + assert.NotEqual(t, alice.Rands, gotAlice.Rands) +} + +func usersMarkEmailPrimary(t *testing.T, db *users) { + ctx := context.Background() + alice, err := db.Create(ctx, "alice", "alice@example.com", CreateUserOptions{}) + require.NoError(t, err) + err = db.AddEmail(ctx, alice.ID, "alice2@example.com", false) + require.NoError(t, err) + + // Should fail because email not verified + gotError := db.MarkEmailPrimary(ctx, alice.ID, "alice2@example.com") + wantError := ErrEmailNotVerified{args: errutil.Args{"email": "alice2@example.com"}} + assert.Equal(t, wantError, gotError) + + // Mark email as verified and should succeed + err = db.MarkEmailActivated(ctx, alice.ID, "alice2@example.com") + require.NoError(t, err) + err = db.MarkEmailPrimary(ctx, alice.ID, "alice2@example.com") + require.NoError(t, err) + gotAlice, err := db.GetByID(ctx, alice.ID) + require.NoError(t, err) + assert.Equal(t, "alice2@example.com", gotAlice.Email) + + // Former primary email should be preserved + gotEmail, err := db.GetEmail(ctx, alice.ID, "alice@example.com", false) + require.NoError(t, err) + assert.False(t, gotEmail.IsActivated) +} + +func usersDeleteEmail(t *testing.T, db *users) { + ctx := context.Background() + alice, err := db.Create(ctx, "alice", "alice@example.com", CreateUserOptions{}) + require.NoError(t, err) + + err = db.AddEmail(ctx, alice.ID, "alice2@example.com", false) + require.NoError(t, err) + _, err = db.GetEmail(ctx, alice.ID, "alice2@example.com", false) + require.NoError(t, err) + + err = db.DeleteEmail(ctx, alice.ID, "alice2@example.com") + require.NoError(t, err) + _, got := db.GetEmail(ctx, alice.ID, "alice2@example.com", false) + want := ErrEmailNotExist{args: errutil.Args{"email": "alice2@example.com"}} + require.Equal(t, want, got) +} + func usersFollow(t *testing.T, db *users) { ctx := context.Background() |