245 lines
7.1 KiB
PHP
245 lines
7.1 KiB
PHP
<?php
|
|
/**
|
|
* Admin notes tests
|
|
*
|
|
* @package WooCommerce\Admin\Tests\Notes
|
|
*/
|
|
|
|
use \Automattic\WooCommerce\Admin\Notes\WC_Admin_Notes;
|
|
use \Automattic\WooCommerce\Admin\Notes\WC_Admin_Note;
|
|
|
|
/**
|
|
* Class WC_Tests_Notes_Data_Store
|
|
*/
|
|
class WC_Tests_Notes_Data_Store extends WC_Unit_Test_Case {
|
|
|
|
/**
|
|
* Tests that the read data store method works as expected.
|
|
*/
|
|
public function test_read() {
|
|
$note = new WC_Admin_Note();
|
|
$note->set_title( 'PHPUNIT_TEST_NOTE' );
|
|
$note->set_content( 'PHPUNIT_TEST_NOTE_CONTENT' );
|
|
$note->set_type( WC_Admin_Note::E_WC_ADMIN_NOTE_INFORMATIONAL );
|
|
$note->set_name( 'PHPUNIT_TEST_NOTE_NAME' );
|
|
$note->set_source( 'PHPUNIT_TEST' );
|
|
$note->set_is_snoozable( false );
|
|
$note->set_layout( 'plain' );
|
|
$note->set_image( '' );
|
|
$note->add_action(
|
|
'PHPUNIT_TEST_ACTION_SLUG',
|
|
'PHPUNIT_TEST_ACTION_LABEL',
|
|
'?s=PHPUNIT_TEST_ACTION_URL'
|
|
);
|
|
$note->set_is_deleted( false );
|
|
$note->save();
|
|
|
|
// Load in a new instance of the note.
|
|
$read_note = new WC_Admin_Note();
|
|
$read_note->set_id( $note->get_id() );
|
|
WC_Data_Store::load( 'admin-note' )->read( $read_note );
|
|
|
|
$this->assertEquals( $note->get_title(), $read_note->get_title() );
|
|
$this->assertEquals( $note->get_content(), $read_note->get_content() );
|
|
$this->assertEquals( $note->get_type(), $read_note->get_type() );
|
|
$this->assertEquals( $note->get_name(), $read_note->get_name() );
|
|
$this->assertEquals( $note->get_source(), $read_note->get_source() );
|
|
$this->assertEquals( $note->get_is_snoozable(), '0' !== $read_note->get_is_snoozable() );
|
|
$this->assertEquals( $note->get_layout(), $read_note->get_layout() );
|
|
$this->assertEquals( $note->get_image(), $read_note->get_image() );
|
|
$this->assertEquals( $note->get_actions(), $read_note->get_actions() );
|
|
$this->assertEquals( $note->get_is_deleted(), $read_note->get_is_deleted() );
|
|
}
|
|
|
|
/**
|
|
* Tests that the read data store method does not fail for invalid content data fields.
|
|
*/
|
|
public function test_read_with_invalid_content_data() {
|
|
global $wpdb;
|
|
$data_store = WC_Data_Store::load( 'admin-note' );
|
|
|
|
$note = new WC_Admin_Note();
|
|
$note->set_title( 'PHPUNIT_TEST_NOTE' );
|
|
$note->set_content( 'PHPUNIT_TEST_NOTE_CONTENT' );
|
|
$note->save();
|
|
|
|
// Make sure that empty content_data does not break the note.
|
|
$wpdb->update(
|
|
$wpdb->prefix . 'wc_admin_notes',
|
|
array(
|
|
'content_data' => '',
|
|
),
|
|
array( 'note_id' => $note->get_id() )
|
|
);
|
|
$data_store->read( $note );
|
|
|
|
$this->assertEquals( 'PHPUNIT_TEST_NOTE', $note->get_title() );
|
|
$this->assertEquals( 'PHPUNIT_TEST_NOTE_CONTENT', $note->get_content() );
|
|
|
|
// We also want to make sure an empty array does not break the content.
|
|
$wpdb->update(
|
|
$wpdb->prefix . 'wc_admin_notes',
|
|
array(
|
|
'content_data' => '[]',
|
|
),
|
|
array( 'note_id' => $note->get_id() )
|
|
);
|
|
$data_store->read( $note );
|
|
|
|
$this->assertEquals( 'PHPUNIT_TEST_NOTE', $note->get_title() );
|
|
$this->assertEquals( 'PHPUNIT_TEST_NOTE_CONTENT', $note->get_content() );
|
|
}
|
|
|
|
/**
|
|
* Tests that the delete_notes_with_name method works when a single name is
|
|
* passed in.
|
|
*/
|
|
public function test_delete_notes_with_single_name() {
|
|
$note_name_to_delete = 'PHPUNIT_TEST_NOTE_TO_DELETE_' . gmdate( 'U' );
|
|
$note_name_to_keep = 'PHPUNIT_TEST_NOTE_TO_KEEP_' . gmdate( 'U' );
|
|
$note_names = array(
|
|
$note_name_to_delete,
|
|
$note_name_to_keep,
|
|
);
|
|
|
|
$data_store = WC_Data_Store::load( 'admin-note' );
|
|
|
|
// Create notes.
|
|
foreach ( $note_names as $note_name ) {
|
|
for ( $i = 0; $i < 3; $i++ ) {
|
|
$note = new WC_Admin_Note();
|
|
$note->set_name( $note_name );
|
|
$note->set_title( 'PHPUNIT_TEST_NOTE' );
|
|
$note->set_content( 'PHPUNIT_TEST_NOTE_CONTENT' );
|
|
$note->save();
|
|
}
|
|
}
|
|
|
|
// Make sure that we know that the notes were added properly.
|
|
$this->assertEquals(
|
|
3,
|
|
count( $data_store->get_notes_with_name( $note_name_to_delete ) )
|
|
);
|
|
$this->assertEquals(
|
|
3,
|
|
count( $data_store->get_notes_with_name( $note_name_to_keep ) )
|
|
);
|
|
|
|
// Delete the notes.
|
|
WC_Admin_Notes::delete_notes_with_name( $note_name_to_delete );
|
|
|
|
// Make sure the notes were deleted.
|
|
$this->assertEquals(
|
|
0,
|
|
count( $data_store->get_notes_with_name( $note_name_to_delete ) )
|
|
);
|
|
$this->assertEquals(
|
|
3,
|
|
count( $data_store->get_notes_with_name( $note_name_to_keep ) )
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Tests that the delete_notes_with_name method works when multiple names
|
|
* are passed in.
|
|
*/
|
|
public function test_delete_notes_with_multiple_names() {
|
|
$note_name_to_delete_1 = 'PHPUNIT_TEST_NOTE_TO_DELETE_1_' . gmdate( 'U' );
|
|
$note_name_to_delete_2 = 'PHPUNIT_TEST_NOTE_TO_DELETE_2_' . gmdate( 'U' );
|
|
$note_name_to_keep = 'PHPUNIT_TEST_NOTE_TO_KEEP_' . gmdate( 'U' );
|
|
$note_names = array(
|
|
$note_name_to_delete_1,
|
|
$note_name_to_delete_2,
|
|
$note_name_to_keep,
|
|
);
|
|
|
|
$data_store = WC_Data_Store::load( 'admin-note' );
|
|
|
|
// Create notes.
|
|
foreach ( $note_names as $note_name ) {
|
|
for ( $i = 0; $i < 3; $i++ ) {
|
|
$note = new WC_Admin_Note();
|
|
$note->set_name( $note_name );
|
|
$note->set_title( 'PHPUNIT_TEST_NOTE' );
|
|
$note->set_content( 'PHPUNIT_TEST_NOTE_CONTENT' );
|
|
$note->save();
|
|
}
|
|
}
|
|
|
|
// Make sure that we know that the notes were added properly.
|
|
$this->assertEquals(
|
|
3,
|
|
count( $data_store->get_notes_with_name( $note_name_to_delete_1 ) )
|
|
);
|
|
$this->assertEquals(
|
|
3,
|
|
count( $data_store->get_notes_with_name( $note_name_to_delete_2 ) )
|
|
);
|
|
$this->assertEquals(
|
|
3,
|
|
count( $data_store->get_notes_with_name( $note_name_to_keep ) )
|
|
);
|
|
|
|
// Delete the notes.
|
|
WC_Admin_Notes::delete_notes_with_name(
|
|
array(
|
|
$note_name_to_delete_1,
|
|
$note_name_to_delete_2,
|
|
)
|
|
);
|
|
|
|
// Make sure the notes were deleted.
|
|
$this->assertEquals(
|
|
0,
|
|
count( $data_store->get_notes_with_name( $note_name_to_delete_1 ) )
|
|
);
|
|
$this->assertEquals(
|
|
0,
|
|
count( $data_store->get_notes_with_name( $note_name_to_delete_2 ) )
|
|
);
|
|
$this->assertEquals(
|
|
3,
|
|
count( $data_store->get_notes_with_name( $note_name_to_keep ) )
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Test that fatal errors aren't thrown when the Notes datastore read() throws an exception.
|
|
*/
|
|
public function test_read_throws_exception() {
|
|
// Create a test note.
|
|
$note = new WC_Admin_Note();
|
|
$note->set_name( 'PHPUNIT_TEST_NOTE_NAME' );
|
|
$note->set_title( 'PHPUNIT_TEST_NOTE_TITLE' );
|
|
$note->set_content( 'PHPUNIT_TEST_NOTE_CONTENT' );
|
|
$note->save();
|
|
|
|
$note_id = $note->get_id();
|
|
|
|
// Sub in a mock datastore that throws an error on read().
|
|
$mock_datastore = $this->createMock( \Automattic\WooCommerce\Admin\Notes\DataStore::class );
|
|
$mock_datastore
|
|
->method( 'read' )
|
|
->will( $this->throwException( new \Exception() ) );
|
|
|
|
// Suppress deliberately caused errors.
|
|
$log_file = ini_set( 'error_log', '/dev/null' );
|
|
|
|
$filter_datastore = function() use ( $mock_datastore ) {
|
|
return $mock_datastore;
|
|
};
|
|
|
|
add_filter( 'woocommerce_admin-note_data_store', $filter_datastore );
|
|
|
|
// Attempt to retrieve the test note.
|
|
$note = WC_Admin_Notes::get_note( $note_id );
|
|
|
|
remove_filter( 'woocommerce_admin-note_data_store', $filter_datastore );
|
|
|
|
ini_set( 'error_log', $log_file );
|
|
|
|
$this->assertFalse( $note );
|
|
$this->assertEquals( 1, did_action( 'woocommerce_caught_exception' ) );
|
|
}
|
|
}
|