chore: Add WorkspaceApps to dbgen

Emyrk · Emyrk · commit 13a4fabb2e9a · 2023-02-02T21:01:50.000-06:00
diff --git a/coderd/database/dbgen/generator.go b/coderd/database/dbgen/generator.go
@@ -254,6 +254,34 @@ func ProvisionerJob(t *testing.T, db database.Store, orig database.ProvisionerJo
 	return job
 }
 
+func WorkspaceApp(t *testing.T, db database.Store, orig database.WorkspaceApp) database.WorkspaceApp {
+	resource, err := db.InsertWorkspaceApp(context.Background(), database.InsertWorkspaceAppParams{
+		ID:          takeFirst(orig.ID, uuid.New()),
+		CreatedAt:   takeFirst(orig.CreatedAt, time.Now()),
+		AgentID:     takeFirst(orig.AgentID, uuid.New()),
+		Slug:        takeFirst(orig.Slug, namesgenerator.GetRandomName(1)),
+		DisplayName: takeFirst(orig.DisplayName, namesgenerator.GetRandomName(1)),
+		Icon:        takeFirst(orig.Icon, namesgenerator.GetRandomName(1)),
+		Command: sql.NullString{
+			String: takeFirst(orig.Command.String, "ls"),
+			Valid:  orig.Command.Valid,
+		},
+		Url: sql.NullString{
+			String: takeFirst(orig.Url.String),
+			Valid:  orig.Url.Valid,
+		},
+		External:             orig.External,
+		Subdomain:            orig.Subdomain,
+		SharingLevel:         takeFirst(orig.SharingLevel, database.AppSharingLevelOwner),
+		HealthcheckUrl:       takeFirst(orig.HealthcheckUrl, "https://localhost:8000"),
+		HealthcheckInterval:  takeFirst(orig.HealthcheckInterval, 60),
+		HealthcheckThreshold: takeFirst(orig.HealthcheckThreshold, 60),
+		Health:               takeFirst(orig.Health, database.WorkspaceAppHealthHealthy),
+	})
+	require.NoError(t, err, "insert app")
+	return resource
+}
+
 func WorkspaceResource(t *testing.T, db database.Store, orig database.WorkspaceResource) database.WorkspaceResource {
 	resource, err := db.InsertWorkspaceResource(context.Background(), database.InsertWorkspaceResourceParams{
 		ID:         takeFirst(orig.ID, uuid.New()),
diff --git a/coderd/database/dbgen/generator_test.go b/coderd/database/dbgen/generator_test.go
@@ -51,6 +51,13 @@ func TestGenerator(t *testing.T) {
 		require.Equal(t, exp, must(db.GetWorkspaceResourceByID(context.Background(), exp.ID)))
 	})
 
+	t.Run("WorkspaceApp", func(t *testing.T) {
+		t.Parallel()
+		db := dbfake.New()
+		exp := dbgen.WorkspaceApp(t, db, database.WorkspaceApp{})
+		require.Equal(t, exp, must(db.GetWorkspaceAppsByAgentID(context.Background(), exp.AgentID))[0])
+	})
+
 	t.Run("WorkspaceResourceMetadatum", func(t *testing.T) {
 		t.Parallel()
 		db := dbfake.New()
