Don't loose mysql FOREIGN_KEY_CHECKS status

sebastianbergmann · Jan 23, 2018 · b8b728e · b8b728e
1 parent fa2b2a3
commit b8b728e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/Operation/Truncate.php b/src/Operation/Truncate.php
@@ -59,14 +59,15 @@ public function execute(Connection $connection, IDataSet $dataSet)
     private function disableForeignKeyChecksForMysql(Connection $connection)
     {
         if ($this->isMysql($connection)) {
+            $connection->getConnection()->query('SET @PHPUNIT_OLD_FOREIGN_KEY_CHECKS = @@FOREIGN_KEY_CHECKS');
             $connection->getConnection()->query('SET FOREIGN_KEY_CHECKS = 0');
         }
     }
 
     private function enableForeignKeyChecksForMysql(Connection $connection)
     {
         if ($this->isMysql($connection)) {
-            $connection->getConnection()->query('SET FOREIGN_KEY_CHECKS = 1');
+            $connection->getConnection()->query('SET FOREIGN_KEY_CHECKS=@PHPUNIT_OLD_FOREIGN_KEY_CHECKS');
         }
     }