woocommerce/plugins/woocommerce-admin/tests/framework/helpers/class-wc-helper-admin-notes...

106 lines
3.4 KiB
PHP

<?php
/**
* Admin notes helper for wc-admin unit tests.
*
* @package WooCommerce\Admin\Tests\Framework\Helpers
*/
use Automattic\WooCommerce\Admin\Notes\Note;
/**
* Class WC_Helper_Admin_Notes.
*
* This helper class should ONLY be used for unit tests!.
*/
class WC_Helper_Admin_Notes {
/**
* Delete everything in the notes tables.
*/
public static function reset_notes_dbs() {
global $wpdb;
$wpdb->query( "TRUNCATE TABLE {$wpdb->prefix}wc_admin_notes" ); // @codingStandardsIgnoreLine.
$wpdb->query( "TRUNCATE TABLE {$wpdb->prefix}wc_admin_note_actions" ); // @codingStandardsIgnoreLine.
}
/**
* Create two notes that we can use for notes REST API tests
*/
public static function add_notes_for_tests() {
$data_store = WC_Data_Store::load( 'admin-note' );
$note_1 = new Note();
$note_1->set_title( 'PHPUNIT_TEST_NOTE_1_TITLE' );
$note_1->set_content( 'PHPUNIT_TEST_NOTE_1_CONTENT' );
$note_1->set_content_data( (object) array( 'amount' => 1.23 ) );
$note_1->set_type( Note::E_WC_ADMIN_NOTE_INFORMATIONAL );
$note_1->set_name( 'PHPUNIT_TEST_NOTE_NAME' );
$note_1->set_source( 'PHPUNIT_TEST' );
$note_1->set_is_snoozable( false );
$note_1->set_layout( 'plain' );
$note_1->set_image( '' );
$note_1->add_action(
'PHPUNIT_TEST_NOTE_1_ACTION_1_SLUG',
'PHPUNIT_TEST_NOTE_1_ACTION_1_LABEL',
'?s=PHPUNIT_TEST_NOTE_1_ACTION_1_URL'
);
$note_1->add_action(
'PHPUNIT_TEST_NOTE_1_ACTION_2_SLUG',
'PHPUNIT_TEST_NOTE_1_ACTION_2_LABEL',
'?s=PHPUNIT_TEST_NOTE_1_ACTION_2_URL'
);
$note_1->save();
$note_2 = new Note();
$note_2->set_title( 'PHPUNIT_TEST_NOTE_2_TITLE' );
$note_2->set_content( 'PHPUNIT_TEST_NOTE_2_CONTENT' );
$note_2->set_content_data( (object) array( 'amount' => 4.56 ) );
$note_2->set_type( Note::E_WC_ADMIN_NOTE_WARNING );
$note_2->set_name( 'PHPUNIT_TEST_NOTE_NAME' );
$note_2->set_source( 'PHPUNIT_TEST' );
$note_2->set_status( Note::E_WC_ADMIN_NOTE_ACTIONED );
$note_2->set_is_snoozable( true );
$note_2->set_layout( 'banner' );
$note_2->set_image( 'https://an-image.jpg' );
// This note has no actions.
$note_2->save();
$note_3 = new Note();
$note_3->set_title( 'PHPUNIT_TEST_NOTE_3_TITLE' );
$note_3->set_content( 'PHPUNIT_TEST_NOTE_3_CONTENT' );
$note_3->set_content_data( (object) array( 'amount' => 7.89 ) );
$note_3->set_type( Note::E_WC_ADMIN_NOTE_INFORMATIONAL );
$note_3->set_name( 'PHPUNIT_TEST_NOTE_NAME' );
$note_3->set_source( 'PHPUNIT_TEST' );
$note_3->set_status( Note::E_WC_ADMIN_NOTE_SNOOZED );
$note_3->set_date_reminder( time() - HOUR_IN_SECONDS );
$note_3->set_layout( 'thumbnail' );
$note_3->set_image( 'https://an-image.jpg' );
// This note has no actions.
$note_3->save();
$note_4 = new Note();
$note_4->set_title( 'PHPUNIT_TEST_NOTE_4_TITLE' );
$note_4->set_content( 'PHPUNIT_TEST_NOTE_4_CONTENT' );
$note_4->set_content_data( (object) array( 'amount' => 1.23 ) );
$note_4->set_type( Note::E_WC_ADMIN_NOTE_INFORMATIONAL );
$note_4->set_name( 'PHPUNIT_TEST_NOTE_NAME' );
$note_4->set_source( 'PHPUNIT_TEST' );
$note_4->set_is_snoozable( false );
$note_4->set_layout( 'plain' );
$note_4->set_image( '' );
$note_4->add_action(
'PHPUNIT_TEST_NOTE_4_ACTION_1_SLUG',
'PHPUNIT_TEST_NOTE_4_ACTION_1_LABEL',
'?s=PHPUNIT_TEST_NOTE_4_ACTION_1_URL'
);
$note_4->add_action(
'PHPUNIT_TEST_NOTE_4_ACTION_2_SLUG',
'PHPUNIT_TEST_NOTE_4_ACTION_2_LABEL',
'?s=PHPUNIT_TEST_NOTE_4_ACTION_2_URL'
);
$note_4->save();
}
}