diff options
author | Joe Chen <jc@unknwon.io> | 2022-10-09 22:55:32 +0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-10-09 22:55:32 +0800 |
commit | 8982a42d38c537de87fabcc59f062841f1c437c2 (patch) | |
tree | bae626255a1582feb14be4b73f5aa0a581c141ab /internal/db/users_test.go | |
parent | fb00e3e56fd777df96e71df1b497e50ff3fbb069 (diff) |
refactor(db): move `User.HasForkedRepository` to `users.HasForkedRepository` (#7176)
Co-authored-by: deepsource-autofix[bot] <62050782+deepsource-autofix[bot]@users.noreply.github.com>
Diffstat (limited to 'internal/db/users_test.go')
-rw-r--r-- | internal/db/users_test.go | 23 |
1 files changed, 22 insertions, 1 deletions
diff --git a/internal/db/users_test.go b/internal/db/users_test.go index 94922a18..68c1bfdc 100644 --- a/internal/db/users_test.go +++ b/internal/db/users_test.go @@ -24,7 +24,7 @@ func TestUsers(t *testing.T) { } t.Parallel() - tables := []interface{}{new(User), new(EmailAddress)} + tables := []interface{}{new(User), new(EmailAddress), new(Repository)} db := &users{ DB: dbtest.NewDB(t, "users", tables...), } @@ -38,6 +38,7 @@ func TestUsers(t *testing.T) { {"GetByEmail", usersGetByEmail}, {"GetByID", usersGetByID}, {"GetByUsername", usersGetByUsername}, + {"HasForkedRepository", usersHasForkedRepository}, } { t.Run(tc.name, func(t *testing.T) { t.Cleanup(func() { @@ -275,3 +276,23 @@ func usersGetByUsername(t *testing.T, db *users) { wantErr := ErrUserNotExist{args: errutil.Args{"name": "bad_username"}} assert.Equal(t, wantErr, err) } + +func usersHasForkedRepository(t *testing.T, db *users) { + ctx := context.Background() + + has := db.HasForkedRepository(ctx, 1, 1) + assert.False(t, has) + + _, err := NewReposStore(db.DB).Create( + ctx, + 1, + CreateRepoOptions{ + Name: "repo1", + ForkID: 1, + }, + ) + require.NoError(t, err) + + has = db.HasForkedRepository(ctx, 1, 1) + assert.True(t, has) +} |