feat: add assertion to check if staging table is empty

Chia-Network · Aug 23, 2022 · e2fc4da · e2fc4da
1 parent 488dba2
commit e2fc4da
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/utils/data-assertions.js b/src/utils/data-assertions.js
@@ -223,3 +223,11 @@ export const assertProjectRecordExists = async (
 
   return record.dataValues;
 };
+
+export const assertStagingTableIsEmpty = async (customMessage) => {
+  const count = await Staging.count({ raw: true });
+
+  if (count > 0) {
+    throw new Error(customMessage || `Staging table is not empty.`);
+  }
+};