aboutsummaryrefslogtreecommitdiff
path: root/internal/db/users_test.go
diff options
context:
space:
mode:
authorJoe Chen <jc@unknwon.io>2022-10-09 22:55:32 +0800
committerGitHub <noreply@github.com>2022-10-09 22:55:32 +0800
commit8982a42d38c537de87fabcc59f062841f1c437c2 (patch)
treebae626255a1582feb14be4b73f5aa0a581c141ab /internal/db/users_test.go
parentfb00e3e56fd777df96e71df1b497e50ff3fbb069 (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.go23
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)
+}