Fix dbgen to set url and wildcard

Emyrk · Emyrk · commit 481a0323bbc3 · 2023-04-20T11:25:16.000-05:00
diff --git a/coderd/database/dbgen/generator.go b/coderd/database/dbgen/generator.go
@@ -343,7 +343,7 @@ func WorkspaceProxy(t testing.TB, db database.Store, orig database.WorkspaceProx
 	require.NoError(t, err, "generate secret")
 	hashedSecret := sha256.Sum256([]byte(secret))
 
-	resource, err := db.InsertWorkspaceProxy(context.Background(), database.InsertWorkspaceProxyParams{
+	proxy, err := db.InsertWorkspaceProxy(context.Background(), database.InsertWorkspaceProxyParams{
 		ID:                takeFirst(orig.ID, uuid.New()),
 		Name:              takeFirst(orig.Name, namesgenerator.GetRandomName(1)),
 		DisplayName:       takeFirst(orig.DisplayName, namesgenerator.GetRandomName(1)),
@@ -353,7 +353,17 @@ func WorkspaceProxy(t testing.TB, db database.Store, orig database.WorkspaceProx
 		UpdatedAt:         takeFirst(orig.UpdatedAt, database.Now()),
 	})
 	require.NoError(t, err, "insert proxy")
-	return resource, secret
+
+	// Also set these fields if the caller wants them.
+	if orig.Url != "" || orig.WildcardHostname != "" {
+		proxy, err = db.RegisterWorkspaceProxy(context.Background(), database.RegisterWorkspaceProxyParams{
+			Url:              orig.Url,
+			WildcardHostname: orig.WildcardHostname,
+			ID:               proxy.ID,
+		})
+		require.NoError(t, err, "update proxy")
+	}
+	return proxy, secret
 }
 
 func File(t testing.TB, db database.Store, orig database.File) database.File {
