fix(migrate): seed - await exported function (#7139)

Jolg42 · web-flow · commit 47decbd9d0bf · 2021-05-18T16:34:04.000+02:00
diff --git a/src/packages/migrate/src/__tests__/fixtures/seed-sqlite-js-ts-default-export/tsconfig.json b/src/packages/migrate/src/__tests__/fixtures/seed-sqlite-js-ts-default-export/tsconfig.json
@@ -0,0 +1,5 @@
+{
+  "compilerOptions": {
+    "lib": ["DOM", "ES2016"]
+  }
+}
diff --git a/src/packages/migrate/src/__tests__/fixtures/seed-sqlite-js-ts-named-export/prisma/seed.js b/src/packages/migrate/src/__tests__/fixtures/seed-sqlite-js-ts-named-export/prisma/seed.js
@@ -1,5 +1,6 @@
-function seed() {
-  return 'hello from "seed" named export from seed.js'
+async function seed() {
+  await new Promise((_) => null) // eslint-disable-line @typescript-eslint/no-unused-vars
+  return 'hello from async "seed" named export from seed.js'
 }
 
 if (require.main === module) {
diff --git a/src/packages/migrate/src/__tests__/fixtures/seed-sqlite-js-ts-named-export/prisma/seed.ts b/src/packages/migrate/src/__tests__/fixtures/seed-sqlite-js-ts-named-export/prisma/seed.ts
@@ -2,8 +2,9 @@
 declare const require: any
 declare const module: any
 
-function seed() {
-  return 'hello from "seed" named export from seed.ts'
+async function seed() {
+  await new Promise((_) => null) // eslint-disable-line @typescript-eslint/no-unused-vars
+  return 'hello from async "seed" named export from seed.ts'
 }
 
 if (require.main === module) {
diff --git a/src/packages/migrate/src/__tests__/fixtures/seed-sqlite-js-ts-named-export/tsconfig.json b/src/packages/migrate/src/__tests__/fixtures/seed-sqlite-js-ts-named-export/tsconfig.json
@@ -0,0 +1,5 @@
+{
+  "compilerOptions": {
+    "lib": ["DOM", "ES2016"]
+  }
+}
diff --git a/src/packages/migrate/src/__tests__/fixtures/seed-sqlite/tsconfig.json b/src/packages/migrate/src/__tests__/fixtures/seed-sqlite/tsconfig.json
@@ -0,0 +1,5 @@
+{
+  "compilerOptions": {
+    "lib": ["DOM", "ES2016"]
+  }
+}
diff --git a/src/packages/migrate/src/utils/seed.ts b/src/packages/migrate/src/utils/seed.ts
@@ -69,19 +69,26 @@ export function detectSeedFiles(schemaPath) {
 
 function getSeedScript(type: 'TS' | 'JS', seedFilepath: string) {
   let script = `
+console.info('Result:')
+
 const __seed = require('./${seedFilepath}')
 const __keys = Object.keys(__seed)
 
-// Execute "seed" named export or default export
-if (__keys && __keys.length) {
-  if (__keys.indexOf('seed') !== -1) {
-    __seed.seed()
-  } else if (__keys.indexOf('default') !== -1) {
-    __seed.default()
+async function runSeed() {
+  // Execute "seed" named export or default export
+  if (__keys && __keys.length) {
+    if (__keys.indexOf('seed') !== -1) {
+      return await __seed.seed()
+    } else if (__keys.indexOf('default') !== -1) {
+      return await __seed.default()
+    }
+  } else {
+    return ''
   }
-} else {
-  ''
-}`
+}
+
+runSeed()
+`
 
   if (type === 'TS') {
     script = `

-Original file line number
+Diff line change
@@ @@ -0,0 +1,5 @@ @@
 +{
 +  "compilerOptions": {
 +    "lib": ["DOM", "ES2016"]
 +  }
 +}
Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,6 @@`
`1`		`-function seed() {`
`2`		`- return 'hello from "seed" named export from seed.js'`
	`1`	`+async function seed() {`
	`2`	`+ await new Promise((_) => null) // eslint-disable-line @typescript-eslint/no-unused-vars`
	`3`	`+ return 'hello from async "seed" named export from seed.js'`
`3`	`4`	`}`
`4`	`5`
`5`	`6`	`if (require.main === module) {`