Set postgres enabled state via request body in E2E and drizzle next…

… tests (#1570) Co-authored-by: Xata Bot <[email protected]> Co-authored-by: github-merge-queue[bot] <github-merge-queue[bot]@users.noreply.github.com>
xataio · Sep 9, 2024 · 00a8db0 · 00a8db0
1 parent eed3363
commit 00a8db0
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/packages/client/src/api/controlPlaneComponents.ts b/packages/client/src/api/controlPlaneComponents.ts
@@ -1494,6 +1494,12 @@ export type CreateDatabaseRequestBody = {
    * @minLength 1
    */
   region: string;
+  /**
+   * Enable postgres access for this database
+   *
+   * @default false
+   */
+  postgresEnabled?: boolean;
   /**
    * The dedicated cluster where branches from this database will be created. Defaults to 'shared-cluster'.
    *

diff --git a/packages/plugin-client-drizzle/test/drizzle.test.ts b/packages/plugin-client-drizzle/test/drizzle.test.ts
@@ -80,8 +80,7 @@ describe.concurrent.each([{ type: 'pg' }, { type: 'http' }])('Drizzle $type', ({
   beforeAll(async () => {
     await api.databases.createDatabase({
       pathParams: { workspaceId: workspace, dbName: database },
-      body: { region, branchName: 'main' },
-      headers: { 'X-Features': 'feat-pgroll-migrations=1' }
+      body: { region, branchName: 'main', postgresEnabled: true }
     });
 
     await waitForReplication();

diff --git a/test/utils/setup.ts b/test/utils/setup.ts
@@ -80,8 +80,7 @@ export async function setUpTestEnvironment(
 
   const { databaseName: database } = await api.databases.createDatabase({
     pathParams: { workspaceId: workspace, dbName: `sdk-integration-test-${prefix}-${id}` },
-    body: { region, defaultClusterID: clusterId },
-    headers: { 'X-Features': 'feat-pgroll-migrations=1' }
+    body: { region, defaultClusterID: clusterId, postgresEnabled: true }
   });
 
   const workspaceUrl = getHostUrl(host, 'workspaces').replace('{workspaceId}', workspace).replace('{region}', region);