Browse Source

chore: Correctly pass DB to benchmark config

Michael Bromley 3 years ago
parent
commit
4b0001f4f3

+ 1 - 1
packages/dev-server/load-testing/benchmarks.ts

@@ -62,7 +62,7 @@ const opts = program.opts() as any;
 runBenchmark(opts).then(() => process.exit(0));
 
 async function runBenchmark(options: Options) {
-    const config = getLoadTestConfig('bearer', DATABASE_NAME);
+    const config = getLoadTestConfig('bearer', DATABASE_NAME, options.db);
     if (options.populate) {
         console.log(`Populating benchmark database "${DATABASE_NAME}"`);
         await clearAllTables(config, true);

+ 2 - 1
packages/dev-server/load-testing/load-test-config.ts

@@ -40,6 +40,7 @@ export function getPostgresConnectionOptions(databaseName: string) {
 export function getLoadTestConfig(
     tokenMethod: 'cookie' | 'bearer',
     databaseName: string,
+    db?: 'postgres' | 'mysql',
 ): Required<VendureConfig> {
     return mergeConfig(defaultConfig, {
         paymentOptions: {
@@ -50,7 +51,7 @@ export function getLoadTestConfig(
         },
         logger: new DefaultLogger({ level: LogLevel.Info }),
         dbConnectionOptions:
-            process.env.DB === 'postgres'
+            process.env.DB === 'postgres' || db === 'postgres'
                 ? getPostgresConnectionOptions(databaseName)
                 : getMysqlConnectionOptions(databaseName),
         authOptions: {