Skip to content

Commit

Permalink
Merge pull request #25 from OllieJones/information-schema
Browse files Browse the repository at this point in the history 
Information schema
  • Loading branch information
@aristath
aristath authored Mar 28, 2023
2 parents d50ead7 + 0cc9cea commit 34d46b0
Show file tree
Hide file tree
Showing 8 changed files with 1,747 additions and 102 deletions.
2 changes: 2 additions & 0 deletions .gitignore
View file
Original file line number Diff line number Diff line change
@@ -1,2 +1,4 @@
vendor/
composer.lock
.idea/
.phpunit.result.cache
322 changes: 322 additions & 0 deletions tests/WP_SQLite_Metadata_Tests.php
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,322 @@
<?php

use PHPUnit\Framework\TestCase;

class WP_SQLite_Metadata_Tests extends TestCase {

private $engine;
private $sqlite;

public static function setUpBeforeClass(): void {
// if ( ! defined( 'PDO_DEBUG' )) {
// define( 'PDO_DEBUG', true );
// }
if ( ! defined( 'FQDB' ) ) {
define( 'FQDB', ':memory:' );
define( 'FQDBDIR', __DIR__ . '/../testdb' );
}
error_reporting( E_ALL & ~E_DEPRECATED );
if ( ! isset( $GLOBALS['table_prefix'] ) ) {
$GLOBALS['table_prefix'] = 'wptests_';
}
if ( ! isset( $GLOBALS['wpdb'] ) ) {
$GLOBALS['wpdb'] = new stdClass();
$GLOBALS['wpdb']->suppress_errors = false;
$GLOBALS['wpdb']->show_errors = true;
}
}

// Before each test, we create a new database
public function setUp(): void {
global $blog_tables;
$queries = explode( ';', $blog_tables );

$this->sqlite = new PDO( 'sqlite::memory:' );
$this->engine = new WP_SQLite_Translator( $this->sqlite );

$translator = $this->engine;

try {
$translator->begin_transaction();
foreach ( $queries as $query ) {
$query = trim( $query );
if ( empty( $query ) ) {
continue;
}

$result = $translator->execute_sqlite_query( $query );
if ( false === $result ) {
throw new PDOException( $translator->get_error_message() );
}
}
$translator->commit();
} catch ( PDOException $err ) {
$err_data =
$err->errorInfo; // phpcs:ignore WordPress.NamingConventions.ValidVariableName.UsedPropertyNotSnakeCase
$err_code = $err_data[1];
$translator->rollback();
$message = sprintf(
'Error occurred while creating tables or indexes...<br />Query was: %s<br />',
var_export( $query, true )
);
$message .= sprintf( 'Error message is: %s', $err_data[2] );
wp_die( $message, 'Database Error!' );
}
}

public function testCountTables() {
$this->assertQuery( "SELECT count(*) FROM information_schema.tables WHERE table_schema = 'wpdata'" );

$actual = $this->engine->get_query_results();
$count = array_values( get_object_vars( $actual[0] ) )[0];
self::assertIsNumeric( $count );
}

private function assertQuery( $sql, $error_substring = null ) {
$retval = $this->engine->query( $sql );
if ( null === $error_substring ) {
$this->assertEquals(
'',
$this->engine->get_error_message()
);
$this->assertNotFalse(
$retval
);
} else {
$this->assertStringContainsStringIgnoringCase( $error_substring, $this->engine->get_error_message() );
}

return $retval;
}

public function testCheckTable() {

/* a good table */
$table_name = 'wp_options';
$expected_result = array(
(object) array(
'Table' => $table_name,
'Op' => 'check',
'Msg_type' => 'status',
'Msg_text' => 'OK',
),
);

$this->assertQuery(
"CHECK TABLE $table_name;"
);

$this->assertEquals(
$expected_result,
$this->engine->get_query_results()
);

/* a different good table */
$table_name = 'wp_postmeta';
$expected_result = array(
(object) array(
'Table' => $table_name,
'Op' => 'check',
'Msg_type' => 'status',
'Msg_text' => 'OK',
),
);

$this->assertQuery(
"CHECK TABLE $table_name;"
);
$this->assertEquals(
$expected_result,
$this->engine->get_query_results()
);

/* a bogus, missing, table */
$table_name = 'wp_sqlite_rocks';
$expected_result = array(
(object) array(
'Table' => $table_name,
'Op' => 'check',
'Msg_type' => 'Error',
'Msg_text' => "Table '$table_name' doesn't exist",
),
(object) array(
'Table' => $table_name,
'Op' => 'check',
'Msg_type' => 'status',
'Msg_text' => 'Operation failed',
),
);

$this->assertQuery(
"CHECK TABLE $table_name;"
);

$this->assertEquals(
$expected_result,
$this->engine->get_query_results()
);
}

public function testOptimizeTable() {

/* a good table */
$table_name = 'wp_options';

$this->assertQuery(
"OPTIMIZE TABLE $table_name;"
);

$actual = $this->engine->get_query_results();

array_map(
function ( $row ) {
$this->assertIsObject( $row );
$row = (array) $row;
$this->assertIsString( $row['Table'] );
$this->assertIsString( $row['Op'] );
$this->assertIsString( $row['Msg_type'] );
$this->assertIsString( $row['Msg_text'] );
},
$actual
);

$ok = array_filter(
$actual,
function ( $row ) {
$row = (array) $row;

return strtolower( $row['Msg_type'] ) === 'status' && strtolower( $row['Msg_text'] ) === 'ok';
}
);
$this->assertIsArray( $ok );
$this->assertGreaterThan( 0, count( $ok ) );
}

public function testRepairTable() {

/* a good table */
$table_name = 'wp_options';

$this->assertQuery(
"REPAIR TABLE $table_name;"
);

$actual = $this->engine->get_query_results();

array_map(
function ( $r ) {
$this->assertIsObject( $r );
$row = $r;
$row = (array) $row;
$this->assertIsString( $row['Table'] );
$this->assertIsString( $row['Op'] );
$this->assertIsString( $row['Msg_type'] );
$this->assertIsString( $row['Msg_text'] );
},
$actual
);

$ok = array_filter(
$actual,
function ( $row ) {
return strtolower( $row->Msg_type ) === 'status' && strtolower( $row->Msg_text ) === 'ok';
}
);
$this->assertIsArray( $ok );
$this->assertGreaterThan( 0, count( $ok ) );
}

// this tests for successful rejection of a bad query

public function testShowTableStatus() {

$this->assertQuery(
"INSERT INTO wp_comments ( comment_author, comment_content ) VALUES ( 'PhpUnit', 'Testing' )"
);

$this->assertQuery(
"INSERT INTO wp_comments ( comment_author, comment_content ) VALUES ( 'PhpUnit0', 'Testing0' ), ( 'PhpUnit1', 'Testing1' ), ( 'PhpUnit2', 'Testing2' )"
);

$this->assertTableEmpty( 'wp_comments', false );

$this->assertQuery(
'SHOW TABLE STATUS FROM wp;'
);

$actual = $this->engine->get_query_results();

$this->assertIsArray( $actual );
$this->assertGreaterThanOrEqual(
1,
count( $actual )
);
$this->assertIsObject( $actual[0] );

$rows = array_values(
array_filter(
$actual,
function ( $row ) {
$this->assertIsObject( $row );
$this->assertIsString( $row->Name );
$this->assertIsNumeric( $row->Rows );

return str_ends_with( $row->Name, 'comments' );
}
)
);
$this->assertEquals( 'wp_comments', $rows[0]->Name );
$this->assertEquals( 4, $rows[0]->Rows );
}

private function assertTableEmpty( $table_name, $empty ) {

$this->assertQuery(
"SELECT COUNT(*) num FROM $table_name"
);

$actual = $this->engine->get_query_results();
if ( $empty ) {
$this->assertEquals( 0, $actual[0]->num, "$table_name is not empty" );
} else {
$this->assertGreaterThan( 0, $actual[0]->num, "$table_name is empty" );
}
}

public function testTruncateTable() {

$this->assertQuery(
"INSERT INTO wp_comments ( comment_author, comment_content ) VALUES ( 'PhpUnit', 'Testing' )"
);

$this->assertQuery(
"INSERT INTO wp_comments ( comment_author, comment_content ) VALUES ( 'PhpUnit0', 'Testing0' ), ( 'PhpUnit1', 'Testing1' ), ( 'PhpUnit2', 'Testing2' )"
);

$this->assertTableEmpty( 'wp_comments', false );

$this->assertQuery(
'TRUNCATE TABLE wp_comments;'
);
$actual = $this->engine->get_query_results();
$this->assertEquals(
true,
$actual
);
$this->assertTableEmpty( 'wp_comments', true );
}

public function testBogusQuery() {

$this->assertQuery(
'SELECT 1, BOGUS(1) FROM bogus;',
'no such table: bogus'
);
$actual = $this->engine->get_query_results();
$this->assertEquals(
null,
$actual
);
}

}
Loading

0 comments on commit 34d46b0

Please sign in to comment.