diff --git a/.env b/.env index 5ea7f421d5..2ce06fbeed 100644 --- a/.env +++ b/.env @@ -3,7 +3,7 @@ PG_PORT=5490 PG_USER=postgres PG_PASSWORD=postgres PG_DATABASE=stacks_blockchain_api -PG_SCHEMA=public +PG_SCHEMA=stacks_blockchain_api PG_SSL=false # Idle connection timeout in seconds, defaults to 30 # PG_IDLE_TIMEOUT=30 diff --git a/tests/event-replay/import-export.test.ts b/tests/event-replay/import-export.test.ts index c2f7780902..5a5575582e 100644 --- a/tests/event-replay/import-export.test.ts +++ b/tests/event-replay/import-export.test.ts @@ -7,7 +7,7 @@ import { exportEventsAsTsv, importEventsFromTsv } from '../../src/event-replay/e import { startEventServer } from '../../src/event-stream/event-server'; import { httpPostRequest } from '../../src/helpers'; import { useWithCleanup } from '../api/test-helpers'; -import { migrate } from '../utils/test-helpers'; +import { createSchema, migrate } from '../utils/test-helpers'; import { PgSqlClient, dangerousDropAllTables, databaseHasData } from '@hirosystems/api-toolkit'; import { getConnectionArgs } from '../../src/datastore/connection'; @@ -27,7 +27,9 @@ describe('import/export tests', () => { }); test('event import and export cycle - remote', async () => { + const args = getConnectionArgs(); // Import from mocknet TSV + await createSchema(args); await importEventsFromTsv('tests/event-replay/tsv/mocknet.tsv', 'archival', true, true); const chainTip = await db.getChainTip(db.sql); expect(chainTip.block_height).toBe(28); @@ -60,7 +62,9 @@ describe('import/export tests', () => { }); test('event import and export cycle - local', async () => { + const args = getConnectionArgs(); // Import from mocknet TSV + await createSchema(args); await importEventsFromTsv('tests/event-replay/tsv/mocknet.tsv', 'archival', true, true); const chainTip = await db.getChainTip(db.sql); expect(chainTip.block_height).toBe(28); diff --git a/tests/utils/test-helpers.ts b/tests/utils/test-helpers.ts index 0245f8920e..1e4ecbabbb 100644 --- a/tests/utils/test-helpers.ts +++ b/tests/utils/test-helpers.ts @@ -45,7 +45,13 @@ import { CoreRpcPoxInfo, StacksCoreRpcClient } from '../../src/core-rpc/client'; import { DbBlock, DbTx, DbTxStatus } from '../../src/datastore/common'; import { PgWriteStore } from '../../src/datastore/pg-write-store'; import { BitcoinAddressFormat, ECPair, getBitcoinAddressFromKey } from '../../src/ec-helpers'; -import { coerceToBuffer, connectPostgres, runMigrations, timeout } from '@hirosystems/api-toolkit'; +import { + coerceToBuffer, + connectPostgres, + PgConnectionArgs, + runMigrations, + timeout, +} from '@hirosystems/api-toolkit'; import { MIGRATIONS_DIR } from '../../src/datastore/pg-store'; import { getConnectionArgs } from '../../src/datastore/connection'; import { AddressStxBalance } from '../../src/api/schemas/entities/addresses'; @@ -53,6 +59,11 @@ import { ServerStatusResponse } from '../../src/api/schemas/responses/responses' export async function migrate(direction: 'up' | 'down') { const connArgs = getConnectionArgs(); + await createSchema(connArgs); + await runMigrations(MIGRATIONS_DIR, direction, connArgs); +} + +export async function createSchema(connArgs: PgConnectionArgs) { if (typeof connArgs !== 'string' && connArgs.schema) { const sql = await connectPostgres({ usageName: 'tests-migrations-setup', @@ -61,7 +72,6 @@ export async function migrate(direction: 'up' | 'down') { await sql`CREATE SCHEMA IF NOT EXISTS ${sql(connArgs.schema)}`; await sql.end(); } - await runMigrations(MIGRATIONS_DIR, direction, connArgs); } export interface TestEnvContext {