#14862 - Refactor method clear() by adding support of Postgres

phalcon · Feb 25, 2020 · f59d6cb · f59d6cb
1 parent df71cf1
commit f59d6cb
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/tests/_data/fixtures/Migrations/AbstractMigration.php b/tests/_data/fixtures/Migrations/AbstractMigration.php
@@ -69,14 +69,25 @@ public function clear()
                 ->connection
                 ->getAttribute(PDO::ATTR_DRIVER_NAME)
             ;
-            if ('sqlite' !== $driver) {
+            if ($driver === 'mysql') {
                 $this->connection->exec(
                     'truncate table ' . $this->table . ';'
                 );
-            } else {
+            } elseif ($driver === 'sqlite') {
                 $this->connection->exec(
                     'delete from ' . $this->table . ';'
                 );
+            } else {
+                $schema = getenv('DATA_POSTGRES_SCHEMA');
+                $exists = $this
+                    ->connection
+                    ->query("SELECT to_regclass('$schema.$this->table') AS exists;")
+                    ->fetchColumn();
+                if ($exists) {
+                    $this->connection->exec(
+                        'truncate table ' . $this->table . ' cascade;'
+                    );
+                }
             }
         }
     }