Fix database table loop

selective-php · Feb 4, 2022 · 94b283a · 94b283a
1 parent a5896a4
commit 94b283a
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/Traits/DatabaseTestTrait.php b/src/Traits/DatabaseTestTrait.php
@@ -128,8 +128,10 @@ protected function dropTables(): void
                 WHERE table_schema = database()'
         );
 
+        $rows = (array)$statement->fetchAll(PDO::FETCH_ASSOC);
+
         $sql = [];
-        while ($row = $statement->fetch(PDO::FETCH_ASSOC)) {
+        foreach ($rows as $row) {
             $sql[] = sprintf('DROP TABLE `%s`;', $row['TABLE_NAME']);
         }
 
@@ -202,8 +204,10 @@ protected function truncateTables(): void
                 AND update_time IS NOT NULL'
         );
 
+        $rows = (array)$statement->fetchAll(PDO::FETCH_ASSOC);
+
         $sql = [];
-        while ($row = $statement->fetch(PDO::FETCH_ASSOC)) {
+        foreach ($rows as $row) {
             $sql[] = sprintf('TRUNCATE TABLE `%s`;', $row['TABLE_NAME']);
         }