GitSSHKey, UserLink, GitAuthLink

Emyrk · Emyrk · commit 592a62b73fd5 · 2023-02-02T21:36:43.000-06:00
diff --git a/coderd/database/dbgen/generator.go b/coderd/database/dbgen/generator.go
@@ -186,6 +186,18 @@ func User(t *testing.T, db database.Store, orig database.User) database.User {
 	return user
 }
 
+func GitSSHKey(t *testing.T, db database.Store, orig database.GitSSHKey) database.GitSSHKey {
+	key, err := db.InsertGitSSHKey(context.Background(), database.InsertGitSSHKeyParams{
+		UserID:     takeFirst(orig.UserID, uuid.New()),
+		CreatedAt:  takeFirst(orig.CreatedAt, time.Now()),
+		UpdatedAt:  takeFirst(orig.UpdatedAt, time.Now()),
+		PrivateKey: takeFirst(orig.PrivateKey, ""),
+		PublicKey:  takeFirst(orig.PublicKey, ""),
+	})
+	require.NoError(t, err, "insert ssh key")
+	return key
+}
+
 func Organization(t *testing.T, db database.Store, orig database.Organization) database.Organization {
 	org, err := db.InsertOrganization(context.Background(), database.InsertOrganizationParams{
 		ID:          takeFirst(orig.ID, uuid.New()),
@@ -340,6 +352,21 @@ func UserLink(t *testing.T, db database.Store, orig database.UserLink) database.
 	return link
 }
 
+func GitAuthLink(t *testing.T, db database.Store, orig database.GitAuthLink) database.GitAuthLink {
+	link, err := db.InsertGitAuthLink(context.Background(), database.InsertGitAuthLinkParams{
+		ProviderID:        takeFirst(orig.ProviderID, uuid.New().String()),
+		UserID:            takeFirst(orig.UserID, uuid.New()),
+		OAuthAccessToken:  takeFirst(orig.OAuthAccessToken, uuid.NewString()),
+		OAuthRefreshToken: takeFirst(orig.OAuthAccessToken, uuid.NewString()),
+		OAuthExpiry:       takeFirst(orig.OAuthExpiry, time.Now().Add(time.Hour*24)),
+		CreatedAt:         takeFirst(orig.CreatedAt, time.Now()),
+		UpdatedAt:         takeFirst(orig.UpdatedAt, time.Now()),
+	})
+
+	require.NoError(t, err, "insert git auth link")
+	return link
+}
+
 func TemplateVersion(t *testing.T, db database.Store, orig database.TemplateVersion) database.TemplateVersion {
 	version, err := db.InsertTemplateVersion(context.Background(), database.InsertTemplateVersionParams{
 		ID: takeFirst(orig.ID, uuid.New()),
diff --git a/coderd/database/dbgen/generator_test.go b/coderd/database/dbgen/generator_test.go
@@ -44,6 +44,16 @@ func TestGenerator(t *testing.T) {
 		require.Equal(t, exp, must(db.GetUserLinkByLinkedID(context.Background(), exp.LinkedID)))
 	})
 
+	t.Run("GitAuthLink", func(t *testing.T) {
+		t.Parallel()
+		db := dbfake.New()
+		exp := dbgen.GitAuthLink(t, db, database.GitAuthLink{})
+		require.Equal(t, exp, must(db.GetGitAuthLink(context.Background(), database.GetGitAuthLinkParams{
+			ProviderID: exp.ProviderID,
+			UserID:     exp.UserID,
+		})))
+	})
+
 	t.Run("WorkspaceResource", func(t *testing.T) {
 		t.Parallel()
 		db := dbfake.New()
@@ -166,6 +176,13 @@ func TestGenerator(t *testing.T) {
 		exp := dbgen.User(t, db, database.User{})
 		require.Equal(t, exp, must(db.GetUserByID(context.Background(), exp.ID)))
 	})
+
+	t.Run("SSHKey", func(t *testing.T) {
+		t.Parallel()
+		db := dbfake.New()
+		exp := dbgen.GitSSHKey(t, db, database.GitSSHKey{})
+		require.Equal(t, exp, must(db.GetGitSSHKey(context.Background(), exp.UserID)))
+	})
 }
 
 func must[T any](value T, err error) T {
