Ticket #32029: 32029.diff

src/wp-includes/wp-db.php

@@ -2399 +2399 @@
                         return true;
                 }
 
+                // We don't need to check the collation for queries that don't read data.
+                $query = ltrim( $query, "\r\n\t (" );
+                if ( preg_match( '/^(?:SHOW|DESCRIBE|DESC|EXPLAIN)\s/i', $query ) ) {
+                        return true;
+                }
+
                 $table = $this->get_table_from_query( $query );
                 if ( ! $table ) {
                         return false;
@@ -2405 +2411 @@
                 }
 
                 $this->checking_collation = true;
-                $this->get_table_charset( $table );
+                $collation = $this->get_table_charset( $table );
                 $this->checking_collation = false;
 
+                // Tables with no collation, or latin1 only, don't need extra checking.
+                if ( false === $collation || 'latin1' === $collation ) {
+                        return true;
+                }
+
                 $table = strtolower( $table );
                 if ( empty( $this->col_meta[ $table ] ) ) {
                         return false;
                 }
 
+                // If any of the columns don't have one of these collations, it needs more sanity checking.
                 foreach( $this->col_meta[ $table ] as $col ) {
                         if ( empty( $col->Collation ) ) {
                                 continue;

tests/phpunit/tests/db/charset.php

@@ -505 +505 @@
                         $value[0] = "CREATE TABLE $this_table_name {$value[0]}";
                         $value[2] = "SELECT * FROM $this_table_name";
                         $value[3] = "DROP TABLE IF EXISTS $this_table_name";
+                        $value[4] = array(
+                                "SHOW FULL TABLES LIKE $this_table_name",
+                                "DESCRIBE $this_table_name",
+                                "DESC $this_table_name",
+                                "EXPLAIN SELECT * FROM $this_table_name",
+                        );
                 }
                 unset( $value );
 
@@ -516 +522 @@
          * @dataProvider data_table_collation_check
          * @ticket 21212
          */
-        function test_table_collation_check( $create, $expected, $query, $drop ) {
+        function test_table_collation_check( $create, $expected, $query, $drop, $always_true ) {
                 self::$_wpdb->query( $drop );
 
                 self::$_wpdb->query( $create );
@@ -524 +530 @@
                 $return = self::$_wpdb->check_safe_collation( $query );
                 $this->assertEquals( $expected, $return );
 
+                foreach( $always_true as $true_query ) {
+                        $return = self::$_wpdb->check_safe_collation( $true_query );
+                        $this->assertTrue( $return );
+                }
+
                 self::$_wpdb->query( $drop );
         }
 }