Fix docker pull in integration tests

drizzle-team · Apr 7, 2023 · ecce0a2 · ecce0a2
1 parent 5a1f4b0
commit ecce0a2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/integration-tests/tests/mysql-schema.test.ts b/integration-tests/tests/mysql-schema.test.ts
@@ -81,7 +81,10 @@ async function createDockerDB(ctx: Context): Promise<string> {
 	const port = await getPort({ port: 3306 });
 	const image = 'mysql:8';
 
-	await docker.pull(image);
+	const pullStream = await docker.pull(image);
+	await new Promise((resolve, reject) =>
+		docker.modem.followProgress(pullStream, (err) => (err ? reject(err) : resolve(err)))
+	);
 
 	ctx.mysqlContainer = await docker.createContainer({
 		Image: image,

diff --git a/integration-tests/tests/pg-schema.test.ts b/integration-tests/tests/pg-schema.test.ts
@@ -58,7 +58,10 @@ async function createDockerDB(ctx: Context): Promise<string> {
 	const port = await getPort({ port: 5432 });
 	const image = 'postgres:14';
 
-	await docker.pull(image);
+	const pullStream = await docker.pull(image);
+	await new Promise((resolve, reject) =>
+		docker.modem.followProgress(pullStream, (err) => (err ? reject(err) : resolve(err)))
+	);
 
 	ctx.pgContainer = await docker.createContainer({
 		Image: image,