More tests

acuminous · Jan 7, 2024 · 29bc13b · 29bc13b
1 parent 62faf1b
commit 29bc13b
Show file tree

Hide file tree

Showing 4 changed files with 132 additions and 40 deletions.
diff --git a/lib/marv-rdf-driver.js b/lib/marv-rdf-driver.js
@@ -25,6 +25,8 @@ module.exports = (options) => {
 
   function runMigration(migration, cb) {
     const fileType = getFileType(migration);
+    if (!['sql', 'yaml', 'json'].includes(fileType)) return cb(new Error(`Unsupported file type: ${fileType}`));
+
     if (fileType === 'sql') return pgDriver.runMigration(migration, cb);
 
     const script = parseMigrationScript(migration, fileType);
@@ -36,9 +38,7 @@ module.exports = (options) => {
   }
 
   function getFileType(migration) {
-    const fileType = path.extname(migration.filename).replace(/^\./, '');
-    if (!['sql', 'yaml', 'json'].includes(fileType)) return cb(new Error(`Unsupported file type: ${fileType}`));
-    return fileType;
+    return path.extname(migration.filename).replace(/^\./, '');
   }
 
   function parseMigrationScript(migration, fileType) {