sergiotarxz
/
Wordpress
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Test framework: Parse CREATE TABLE queries the same way we do DROP TABLE queries. 

props soulseekah.
fixes #24800.


git-svn-id: https://develop.svn.wordpress.org/trunk@27041 602fd350-edb4-49c9-b593-d223f7449a82
This commit is contained in:
Andrew Nacin 2014-01-27 03:09:13 +00:00
parent 9b414abc2d
commit c759a54b2c
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tests/phpunit/includes/testcase.php
View File

@ -35,8 +35,8 @@ class WP_UnitTestCase extends PHPUnit_Framework_TestCase {
global $wpdb; global $wpdb;
$this->expectedDeprecated(); $this->expectedDeprecated();
$wpdb->query( 'ROLLBACK' ); $wpdb->query( 'ROLLBACK' );
remove_filter( 'dbdelta_create_queries', array( $this, '_create_temporary_tables' ) ); remove_filter( 'query', array( $this, '_create_temporary_table' ) );
remove_filter( 'query', array( $this, '_drop_temporary_tables' ) ); remove_filter( 'query', array( $this, '_drop_temporary_table' ) );
remove_filter( 'wp_die_handler', array( $this, 'get_wp_die_handler' ) ); remove_filter( 'wp_die_handler', array( $this, 'get_wp_die_handler' ) );
} }
@ -64,17 +64,19 @@ class WP_UnitTestCase extends PHPUnit_Framework_TestCase {
global $wpdb; global $wpdb;
$wpdb->query( 'SET autocommit = 0;' ); $wpdb->query( 'SET autocommit = 0;' );
$wpdb->query( 'START TRANSACTION;' ); $wpdb->query( 'START TRANSACTION;' );
add_filter( 'dbdelta_create_queries', array( $this, '_create_temporary_tables' ) ); add_filter( 'query', array( $this, '_create_temporary_table' ) );
add_filter( 'query', array( $this, '_drop_temporary_tables' ) ); add_filter( 'query', array( $this, '_drop_temporary_table' ) );
} }
function _create_temporary_tables( $queries ) { function _create_temporary_table( $query ) {
return str_replace( 'CREATE TABLE', 'CREATE TEMPORARY TABLE', $queries ); if ( 'CREATE TABLE' === substr( $query, 0, 12 ) )
return substr_replace( $query, 'CREATE TEMPORARY TABLE', 0, 12 );
return $query;
} }
function _drop_temporary_tables( $query ) { function _drop_temporary_table( $query ) {
if ( 'DROP TABLE' === substr( $query, 0, 10 ) ) if ( 'DROP TABLE' === substr( $query, 0, 10 ) )
return 'DROP TEMPORARY TABLE ' . substr( $query, 10 ); return substr_replace( $query, 'DROP TEMPORARY TABLE', 0, 10 );
return $query; return $query;
} }