Set postgres enabled state via request body in E2E and drizzle next tests (#1570)

Co-authored-by: Xata Bot <97746683+xata-bot@users.noreply.github.com>
Co-authored-by: github-merge-queue[bot] <github-merge-queue[bot]@users.noreply.github.com>

Author: 3 people

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'.
    *

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();

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);