CRM-19038 backport safe Remove FK code to 4.6
authorSeamus Lee <seamuslee001@gmail.com>
Fri, 1 Jul 2016 02:30:46 +0000 (12:30 +1000)
committerSeamus Lee <seamuslee001@gmail.com>
Mon, 11 Jul 2016 04:02:19 +0000 (04:02 +0000)
CRM/Core/BAO/SchemaHandler.php
tests/phpunit/CRM/Core/BAO/SchemaHandlerTest.php

index 6b2170141cd625a49ed081ae45feeca05d5475e4..c5ddefba107440b94c5542538acdd174845c754a 100644 (file)
@@ -604,4 +604,34 @@ MODIFY      {$columnName} varchar( $length )
     return FALSE;
   }
 
+  /**
+   * Remove a foreign key from a table if it exists
+   *
+   * @param $table_name
+   * @param $constraint_name
+   */
+  public static function safeRemoveFK($table_name, $constraint_name) {
+
+    $config = CRM_Core_Config::singleton();
+    $dbUf = DB::parseDSN($config->dsn);
+    $query = "
+      SELECT CONSTRAINT_NAME FROM INFORMATION_SCHEMA.TABLE_CONSTRAINTS
+      WHERE TABLE_SCHEMA = %1
+      AND TABLE_NAME = %2
+      AND CONSTRAINT_NAME = %3
+      AND CONSTRAINT_TYPE = 'FOREIGN KEY'
+    ";
+    $params = array(
+      1 => array($dbUf['database'], 'String'),
+      2 => array($table_name, 'String'),
+      3 => array($constraint_name, 'String'),
+    );
+    $dao = CRM_Core_DAO::executeQuery($query, $params);
+
+    if ($dao->fetch()) {
+      CRM_Core_DAO::executeQuery("ALTER TABLE {$table_name} DROP FOREIGN KEY {$constraint_name}", array());
+    }
+
+  }
+
 }
index 00a86bbc2d734da680cc8654b8a0129dd1e286b4..90089fb1ca10a6693d7ec5e60ad76c3a04c9ec42 100644 (file)
@@ -148,4 +148,20 @@ class CRM_Core_BAO_SchemaHandlerTest extends CiviUnitTestCase {
     }
   }
 
+  /**
+   * Test to see if we can drop foreign key
+   *
+   */
+  public function testSafeDropForeignKey() {
+    $tests = array('FK_civicrm_mailing_recipients_email_id', 'FK_civicrm_mailing_recipients_id');
+    foreach ($tests as $test) {
+      if ($test == 'FK_civicrm_mailing_recipients_id') {
+        $this->assertFalse(CRM_Core_BAO_SchemaHandler::safeRemoveFK('civicrm_mailing_recipients', $test));
+      }
+      else {
+        $this->assertTrue(CRM_Core_BAO_SchemaHandler::safeRemoveFK('civicrm_mailing_recipients', $test));
+      }
+    }
+  }
+
 }