382 lines
12 KiB
PHP
382 lines
12 KiB
PHP
|
<?php
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Search Replace class unit tests
|
||
|
|
* Written for PHPUnit
|
||
|
|
*
|
||
|
|
* Requires a mysql database with the following schema and utf8_unicode_ci collation:
|
||
|
|
*
|
||
|
|
* posts
|
||
|
|
* id INT
|
||
|
|
* content VARCHAR(2000)
|
||
|
|
* url VARCHAR(2000)
|
||
|
|
* serialised VARCHAR(2000)
|
||
|
|
*
|
||
|
|
*/
|
||
|
|
|
||
|
|
date_default_timezone_set( 'Europe/London' );
|
||
|
|
|
||
|
|
class SrdbTest extends PHPUnit_Extensions_Database_TestCase {
|
||
|
|
|
||
|
|
static private $pdo;
|
||
|
|
|
||
|
|
private $conn;
|
||
|
|
|
||
|
|
public $testdb = array(
|
||
|
|
'host' => '127.0.0.1',
|
||
|
|
'name' => 'srdbtest',
|
||
|
|
'user' => 'root',
|
||
|
|
'pass' => '123',
|
||
|
|
'table'=> 'posts'
|
||
|
|
);
|
||
|
|
|
||
|
|
public function setUp() {
|
||
|
|
parent::setUp();
|
||
|
|
|
||
|
|
// get class to test
|
||
|
|
require_once( dirname( __FILE__ ) . '/../srdb.class.php' );
|
||
|
|
}
|
||
|
|
|
||
|
|
public function tearDown() {
|
||
|
|
parent::tearDown();
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @return PHPUnit_Extensions_Database_DB_IDatabaseConnection
|
||
|
|
*/
|
||
|
|
public function getConnection() {
|
||
|
|
if ( $this->conn === null ) {
|
||
|
|
if ( self::$pdo == null )
|
||
|
|
self::$pdo = new PDO( "mysql:host={$this->testdb['host']}",
|
||
|
|
$this->testdb[ 'user' ],
|
||
|
|
$this->testdb[ 'pass' ],
|
||
|
|
array( PDO::MYSQL_ATTR_INIT_COMMAND => 'SET NAMES utf8mb4' ) );
|
||
|
|
|
||
|
|
self::$pdo->query( "CREATE DATABASE IF NOT EXISTS `{$this->testdb[ 'name' ]}` CHARACTER SET = 'utf8mb4' COLLATE = 'utf8mb4_general_ci';" );
|
||
|
|
self::$pdo->query( "CREATE TABLE IF NOT EXISTS `{$this->testdb[ 'name' ]}`.`{$this->testdb[ 'table' ]}` (
|
||
|
|
`id` int(11) NOT NULL AUTO_INCREMENT,
|
||
|
|
`content` blob,
|
||
|
|
`url` varchar(1024) CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci DEFAULT NULL,
|
||
|
|
`serialised` varchar(2000) CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci DEFAULT NULL,
|
||
|
|
PRIMARY KEY (`id`)
|
||
|
|
) ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8mb4;" );
|
||
|
|
|
||
|
|
self::$pdo->query( "USE `{$this->testdb[ 'name' ]}`;" );
|
||
|
|
|
||
|
|
$this->conn = $this->createDefaultDBConnection( self::$pdo, $this->testdb[ 'name' ] );
|
||
|
|
|
||
|
|
// Get the charset of the table.
|
||
|
|
$charset = $this->get_table_character_set( );
|
||
|
|
if ( $charset )
|
||
|
|
self::$pdo->query( "SET NAMES {$charset};" );
|
||
|
|
}
|
||
|
|
return $this->conn;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
public function get_table_character_set( ) {
|
||
|
|
$charset = self::$pdo->query( "SELECT c.`character_set_name`
|
||
|
|
FROM information_schema.`TABLES` t
|
||
|
|
LEFT JOIN information_schema.`COLLATION_CHARACTER_SET_APPLICABILITY` c
|
||
|
|
ON (t.`TABLE_COLLATION` = c.`COLLATION_NAME`)
|
||
|
|
WHERE t.table_schema = '{$this->testdb[ 'name' ]}'
|
||
|
|
AND t.table_name = '{$this->testdb[ 'table' ]}'
|
||
|
|
LIMIT 1;" );
|
||
|
|
|
||
|
|
$encoding = false;
|
||
|
|
if ( $charset ) {
|
||
|
|
$result = $charset->fetch( );
|
||
|
|
if ( isset( $result[ 'character_set_name' ] ) )
|
||
|
|
$encoding = $result[ 'character_set_name' ];
|
||
|
|
}
|
||
|
|
|
||
|
|
return $encoding;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @return PHPUnit_Extensions_Database_DataSet_IDataSet
|
||
|
|
*/
|
||
|
|
public function getDataSet() {
|
||
|
|
return $this->createXMLDataSet( dirname( __FILE__ ) . '/DataSet.xml' );
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
* @test search replace
|
||
|
|
*/
|
||
|
|
public function testSearchReplace() {
|
||
|
|
|
||
|
|
// search replace strings
|
||
|
|
$search = 'example.com';
|
||
|
|
$replace = 'example.org';
|
||
|
|
|
||
|
|
// runs search/replace
|
||
|
|
$srdb = new icit_srdb( array_merge( array(
|
||
|
|
'search' => $search,
|
||
|
|
'replace' => $replace,
|
||
|
|
'dry_run' => false
|
||
|
|
), $this->testdb ) );
|
||
|
|
|
||
|
|
// results from sample data
|
||
|
|
|
||
|
|
// no errors
|
||
|
|
$this->assertEmpty( $srdb->errors[ 'results' ], "Search replace script errors were found: \n" . implode( "\n", $srdb->errors[ 'results' ] ) );
|
||
|
|
$this->assertEmpty( $srdb->errors[ 'db' ], "Search replace script database errors were found: \n" . implode( "\n", $srdb->errors[ 'db' ] ) );
|
||
|
|
|
||
|
|
// update statements run
|
||
|
|
$updates = $srdb->report[ 'updates' ];
|
||
|
|
$this->assertEquals( 50, $updates, 'Wrong number of updates reported' );
|
||
|
|
|
||
|
|
// cells changed
|
||
|
|
$changes = $srdb->report[ 'change' ];
|
||
|
|
$this->assertEquals( 150, $changes, 'Wrong number of cells changed reported' );
|
||
|
|
|
||
|
|
// test the database is actually changed
|
||
|
|
$modified = self::$pdo->query( "SELECT url FROM `{$this->testdb[ 'table' ]}` LIMIT 1;" )->fetchColumn();
|
||
|
|
$this->assertRegExp( "/{$replace}/", $modified );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
public function testSearchReplaceUnicode() {
|
||
|
|
|
||
|
|
// search replace strings
|
||
|
|
$search = 'perspiciatis';
|
||
|
|
$replace = '😸';
|
||
|
|
|
||
|
|
// runs search/replace
|
||
|
|
$srdb = new icit_srdb( array_merge( array(
|
||
|
|
'search' => $search,
|
||
|
|
'replace' => $replace,
|
||
|
|
'dry_run' => false
|
||
|
|
), $this->testdb ) );
|
||
|
|
|
||
|
|
// results from sample data
|
||
|
|
|
||
|
|
// no errors
|
||
|
|
$this->assertEmpty( $srdb->errors[ 'results' ], "Search replace script errors were found: \n" . implode( "\n", $srdb->errors[ 'results' ] ) );
|
||
|
|
$this->assertEmpty( $srdb->errors[ 'db' ], "Search replace script database errors were found: \n" . implode( "\n", $srdb->errors[ 'db' ] ) );
|
||
|
|
|
||
|
|
// update statements run
|
||
|
|
$updates = $srdb->report[ 'updates' ];
|
||
|
|
$this->assertEquals( 50, $updates, 'Wrong number of updates reported' );
|
||
|
|
|
||
|
|
// cells changed
|
||
|
|
$changes = $srdb->report[ 'change' ];
|
||
|
|
$this->assertEquals( 50, $changes, 'Wrong number of cells changed reported' );
|
||
|
|
|
||
|
|
// test the database is actually changed
|
||
|
|
$modified = self::$pdo->query( "SELECT content FROM `{$this->testdb[ 'table' ]}` LIMIT 1;" )->fetchColumn();
|
||
|
|
$this->assertRegExp( "/{$replace}/", $modified );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/*
|
||
|
|
* @test str_replace regex
|
||
|
|
*/
|
||
|
|
public function testRegexReplace() {
|
||
|
|
|
||
|
|
// search replace strings
|
||
|
|
$search = '#https?://([a-z0-9\.-]+)/?#';
|
||
|
|
$replace = 'https://$1/';
|
||
|
|
|
||
|
|
// class instance with regex enabled
|
||
|
|
$srdb = new icit_srdb( array_merge( array(
|
||
|
|
'search' => $search,
|
||
|
|
'replace' => $replace,
|
||
|
|
'regex' => true,
|
||
|
|
'dry_run' => false
|
||
|
|
), $this->testdb ) );
|
||
|
|
|
||
|
|
// direct method invocation
|
||
|
|
$subject = 'http://example.com/';
|
||
|
|
$result = 'https://example.com/';
|
||
|
|
$replaced = $srdb->str_replace( $search, $replace, $subject );
|
||
|
|
$this->assertEquals( $result, $replaced );
|
||
|
|
|
||
|
|
// results from sample data
|
||
|
|
|
||
|
|
// no errors
|
||
|
|
$this->assertEmpty( $srdb->errors[ 'results' ], "Search replace script errors were found: \n" . implode( "\n", $srdb->errors[ 'results' ] ) );
|
||
|
|
$this->assertEmpty( $srdb->errors[ 'db' ], "Search replace script database errors were found: \n" . implode( "\n", $srdb->errors[ 'db' ] ) );
|
||
|
|
|
||
|
|
// update statements run
|
||
|
|
$updates = $srdb->report[ 'updates' ];
|
||
|
|
$this->assertEquals( 50, $updates, 'Wrong number of updates reported' );
|
||
|
|
|
||
|
|
// cells changed
|
||
|
|
$changes = $srdb->report[ 'change' ];
|
||
|
|
$this->assertEquals( 150, $changes, 'Wrong number of changes reported' );
|
||
|
|
|
||
|
|
// test the database is actually changed
|
||
|
|
$modified = self::$pdo->query( "SELECT url FROM `{$this->testdb[ 'table' ]}` LIMIT 1;" )->fetchColumn();
|
||
|
|
$this->assertRegExp( "#{$result}#", $modified, 'Database not updated, modified result is ' . $modified );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @test str_replace serialised data
|
||
|
|
*/
|
||
|
|
public function testStrReplaceSerialised() {
|
||
|
|
|
||
|
|
// search replace strings
|
||
|
|
$search = 'serialised string';
|
||
|
|
$replace = 'longer serialised string';
|
||
|
|
|
||
|
|
// class instance with regex enabled
|
||
|
|
$srdb = new icit_srdb( array_merge( array(
|
||
|
|
'search' => $search,
|
||
|
|
'replace' => $replace,
|
||
|
|
'dry_run' => false
|
||
|
|
), $this->testdb ) );
|
||
|
|
|
||
|
|
// results from sample data
|
||
|
|
|
||
|
|
// no errors
|
||
|
|
$this->assertEmpty( $srdb->errors[ 'results' ], "Search replace script errors were found: \n" . implode( "\n", $srdb->errors[ 'results' ] ) );
|
||
|
|
$this->assertEmpty( $srdb->errors[ 'db' ], "Search replace script database errors were found: \n" . implode( "\n", $srdb->errors[ 'db' ] ) );
|
||
|
|
|
||
|
|
// update statements run
|
||
|
|
$updates = $srdb->report[ 'updates' ];
|
||
|
|
$this->assertEquals( 50, $updates, 'Wrong number of updates reported' );
|
||
|
|
|
||
|
|
// cells changed
|
||
|
|
$changes = $srdb->report[ 'change' ];
|
||
|
|
$this->assertEquals( 50, $changes, 'Wrong number of changes reported' );
|
||
|
|
|
||
|
|
// check unserialised values are what they should be
|
||
|
|
$modified = self::$pdo->query( "SELECT serialised FROM `{$this->testdb[ 'table' ]}` LIMIT 1;" )->fetchColumn();
|
||
|
|
$from = unserialize( $modified );
|
||
|
|
|
||
|
|
$this->assertEquals( $replace, $from[ 'string' ], 'Unserialised array value not updated' );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
* @test recursive unserialize replace
|
||
|
|
*/
|
||
|
|
public function testRecursiveUnserializeReplace() {
|
||
|
|
|
||
|
|
// search replace strings
|
||
|
|
$search = 'serialised string';
|
||
|
|
$replace = 'longer serialised string';
|
||
|
|
|
||
|
|
// class instance with regex enabled
|
||
|
|
$srdb = new icit_srdb( array_merge( array(
|
||
|
|
'search' => $search,
|
||
|
|
'replace' => $replace,
|
||
|
|
'dry_run' => false
|
||
|
|
), $this->testdb ) );
|
||
|
|
|
||
|
|
// results from sample data
|
||
|
|
|
||
|
|
// no errors
|
||
|
|
$this->assertEmpty( $srdb->errors[ 'results' ], "Search replace script errors were found: \n" . implode( "\n", $srdb->errors[ 'results' ] ) );
|
||
|
|
$this->assertEmpty( $srdb->errors[ 'db' ], "Search replace script database errors were found: \n" . implode( "\n", $srdb->errors[ 'db' ] ) );
|
||
|
|
|
||
|
|
// check unserialised values are what they should be
|
||
|
|
$modified = self::$pdo->query( "SELECT serialised FROM `{$this->testdb[ 'table' ]}` LIMIT 1;" )->fetchColumn();
|
||
|
|
$from = unserialize( $modified );
|
||
|
|
|
||
|
|
$this->assertEquals( $replace, $from[ 'nested' ][ 'string' ], 'Unserialised nested array value not updated' );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
* @test include columns
|
||
|
|
*/
|
||
|
|
public function testIncludeColumns() {
|
||
|
|
|
||
|
|
// search replace strings
|
||
|
|
$search = 'example.com';
|
||
|
|
$replace = 'example.org';
|
||
|
|
|
||
|
|
// class instance with regex enabled
|
||
|
|
$srdb = new icit_srdb( array_merge( array(
|
||
|
|
'search' => $search,
|
||
|
|
'replace' => $replace,
|
||
|
|
'dry_run' => false,
|
||
|
|
'include_cols' => array( 'url' )
|
||
|
|
), $this->testdb ) );
|
||
|
|
|
||
|
|
// results from sample data
|
||
|
|
|
||
|
|
// no errors
|
||
|
|
$this->assertEmpty( $srdb->errors[ 'results' ], "Search replace script errors were found: \n" . implode( "\n", $srdb->errors[ 'results' ] ) );
|
||
|
|
$this->assertEmpty( $srdb->errors[ 'db' ], "Search replace script database errors were found: \n" . implode( "\n", $srdb->errors[ 'db' ] ) );
|
||
|
|
|
||
|
|
// update statements run
|
||
|
|
$updates = $srdb->report[ 'updates' ];
|
||
|
|
$this->assertEquals( 50, $updates, 'Wrong number of updates reported' );
|
||
|
|
|
||
|
|
// cells changed
|
||
|
|
$changes = $srdb->report[ 'change' ];
|
||
|
|
$this->assertEquals( 50, $changes, 'Wrong number of changes reported' );
|
||
|
|
|
||
|
|
|
||
|
|
// check unserialised values are what they should be
|
||
|
|
$modified = self::$pdo->query( "SELECT content, url FROM `{$this->testdb[ 'table' ]}` LIMIT 1;" )->fetchAll();
|
||
|
|
$content = $modified[ 0 ][ 'content' ];
|
||
|
|
$url = $modified[ 0 ][ 'url' ];
|
||
|
|
|
||
|
|
$this->assertRegExp( "/$search/", $content, 'Content column was modified' );
|
||
|
|
$this->assertRegExp( "/$replace/", $url, 'URL column was not modified' );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
* @test exclude columns
|
||
|
|
*/
|
||
|
|
public function testExcludeColumns() {
|
||
|
|
|
||
|
|
// search replace strings
|
||
|
|
$search = 'example.com';
|
||
|
|
$replace = 'example.org';
|
||
|
|
|
||
|
|
// class instance with regex enabled
|
||
|
|
$srdb = new icit_srdb( array_merge( array(
|
||
|
|
'search' => $search,
|
||
|
|
'replace' => $replace,
|
||
|
|
'dry_run' => false,
|
||
|
|
'exclude_cols' => array( 'url' )
|
||
|
|
), $this->testdb ) );
|
||
|
|
|
||
|
|
// results from sample data
|
||
|
|
|
||
|
|
// no errors
|
||
|
|
$this->assertEmpty( $srdb->errors[ 'results' ], "Search replace script errors were found: \n" . implode( "\n", $srdb->errors[ 'results' ] ) );
|
||
|
|
$this->assertEmpty( $srdb->errors[ 'db' ], "Search replace script database errors were found: \n" . implode( "\n", $srdb->errors[ 'db' ] ) );
|
||
|
|
|
||
|
|
// update statements run
|
||
|
|
$updates = $srdb->report[ 'updates' ];
|
||
|
|
$this->assertEquals( 50, $updates, 'Wrong number of updates reported' );
|
||
|
|
|
||
|
|
// cells changed
|
||
|
|
$changes = $srdb->report[ 'change' ];
|
||
|
|
$this->assertEquals( 100, $changes, 'Wrong number of changes reported' );
|
||
|
|
|
||
|
|
// check unserialised values are what they should be
|
||
|
|
$modified = self::$pdo->query( "SELECT content, url FROM `{$this->testdb[ 'table' ]}` LIMIT 1;" )->fetchAll();
|
||
|
|
$content = $modified[ 0 ][ 'content' ];
|
||
|
|
$url = $modified[ 0 ][ 'url' ];
|
||
|
|
|
||
|
|
$this->assertRegExp( "/$replace/", $content, 'Content column was not modified' );
|
||
|
|
$this->assertRegExp( "/$search/", $url, 'URL column was modified' );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @test multibyte string replacement method
|
||
|
|
*/
|
||
|
|
public function testMultibyteStrReplace() {
|
||
|
|
|
||
|
|
$subject = 'föö ❤ ☀ ☆ ☂ ☻ ♞ ☯';
|
||
|
|
$result = 'föö ❤ ☻ ♞ ☯ ☻ ♞ ☯';
|
||
|
|
$replaced = icit_srdb::mb_str_replace( '☀ ☆ ☂', '☻ ♞ ☯', $subject );
|
||
|
|
|
||
|
|
$this->assertEquals( $result, $replaced );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|