fix unit tests

core-haxe · Aug 7, 2024 · 0d6b694 · 0d6b694
1 parent 606409f
commit 0d6b694
Show file tree

Hide file tree

Showing 6 changed files with 43 additions and 0 deletions.
diff --git a/tests/cases/DBCreatorBase.hx b/tests/cases/DBCreatorBase.hx
@@ -21,6 +21,7 @@ class DBCreatorBase {
                 File.saveContent(sqliteFilename, "");
             }
 
+            resetEntities();
             EntityManager.instance.database = db;
             @:privateAccess EntityManager.instance.connect().then(_ -> {
                 return db.delete();
@@ -40,6 +41,10 @@ class DBCreatorBase {
         });
     }
 
+    public function resetEntities() {
+
+    }
+
     public function createDummyData() {
         return new Promise((resolve, reject) -> {
             resolve(true);

diff --git a/tests/cases/basic/DBCreator.hx b/tests/cases/basic/DBCreator.hx
@@ -10,6 +10,12 @@ class DBCreator extends DBCreatorBase {
         sqliteFilename = "basic.db";
     }
 
+    public override function resetEntities() {
+        super.resetEntities();
+        @:privateAccess BasicEntity._checkedTables = false;
+        @:privateAccess BasicEntityStructInit._checkedTables = false;
+    }
+
     public override function createDummyData() {
         return new Promise((resolve, reject) -> {
             var list = [];

diff --git a/tests/cases/books/DBCreator.hx b/tests/cases/books/DBCreator.hx
@@ -9,6 +9,14 @@ class DBCreator extends DBCreatorBase {
         sqliteFilename = "books.db";
     }
 
+    public override function resetEntities() {
+        super.resetEntities();
+        @:privateAccess Author._checkedTables = false;
+        @:privateAccess Book._checkedTables = false;
+        @:privateAccess Category._checkedTables = false;
+        @:privateAccess Publisher._checkedTables = false;
+    }
+
     public override function createDummyData() {
         return new Promise((resolve, reject) -> {
             var list:Array<() -> promises.Promise<Any>> = [];

diff --git a/tests/cases/fakeorg/DBCreator.hx b/tests/cases/fakeorg/DBCreator.hx
@@ -12,6 +12,16 @@ class DBCreator extends DBCreatorBase {
         sqliteFilename = "fakeorg.db";
     }
 
+    public override function resetEntities() {
+        super.resetEntities();
+        @:privateAccess Address._checkedTables = false;
+        @:privateAccess AddressLine._checkedTables = false;
+        @:privateAccess Icon._checkedTables = false;
+        @:privateAccess Image._checkedTables = false;
+        @:privateAccess Organization._checkedTables = false;
+        @:privateAccess Worker._checkedTables = false;
+    }
+
     public override function createDummyData() {
         return new Promise((resolve, reject) -> {
             var list:Array<() -> promises.Promise<Any>> = [];

diff --git a/tests/cases/simple/DBCreator.hx b/tests/cases/simple/DBCreator.hx
@@ -1,6 +1,14 @@
 package cases.simple;
 
 class DBCreator extends DBCreatorBase {
+
+    public override function resetEntities() {
+        super.resetEntities();
+        @:privateAccess A._checkedTables = false;
+        @:privateAccess B._checkedTables = false;
+        @:privateAccess C._checkedTables = false;
+    }
+
     public function new() {
         super();
         sqliteFilename = "simple.db";

diff --git a/tests/cases/user/DBCreator.hx b/tests/cases/user/DBCreator.hx
@@ -9,6 +9,12 @@ class DBCreator extends DBCreatorBase {
         sqliteFilename = "users.db";
     }
 
+    public override function resetEntities() {
+        super.resetEntities();
+        @:privateAccess User._checkedTables = false;
+        @:privateAccess Property._checkedTables = false;
+    }
+
     public override function createDummyData() {
         return new Promise((resolve, reject) -> {
             var list = [];