update display_no_table_notice() test

wp-media · May 14, 2024 · 2e19068 · 2e19068
1 parent 795f632
commit 2e19068
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 5 deletions.
diff --git a/tests/Fixtures/inc/Engine/Optimization/RUCSS/Admin/Subscriber/displayNoTableNotice.php b/tests/Fixtures/inc/Engine/Optimization/RUCSS/Admin/Subscriber/displayNoTableNotice.php
@@ -9,22 +9,34 @@
 Notice;
 
 return [
-	'tableExistsShouldDisplayNotice' => [
+	'testShouldDoNothingWhenRucssDisabled' => [
 		'config' => [
-			'remove_unused_css' => true,
+			'remove_unused_css' => false,
+			'table_exists'      => true,
 		],
 		'expected' => [
-			'contains' => true,
+			'contains' => false,
 			'content' => $content,
 		],
 	],
-	'disableShouldDoNothing' => [
+	'testShouldDoNothingWhenTableExists' => [
 		'config' => [
-			'remove_unused_css' => false,
+			'remove_unused_css' => true,
+			'table_exists'      => true,
 		],
 		'expected' => [
 			'contains' => false,
 			'content' => $content,
 		],
 	],
+	'testShouldDisplayNoticeWhenTableNotExists' => [
+		'config' => [
+			'remove_unused_css' => true,
+			'table_exists'      => false,
+		],
+		'expected' => [
+			'contains' => true,
+			'content' => $content,
+		],
+	],
 ];
diff --git a/tests/Integration/inc/Engine/Optimization/RUCSS/Admin/Subscriber/displayNoTableNotice.php b/tests/Integration/inc/Engine/Optimization/RUCSS/Admin/Subscriber/displayNoTableNotice.php
@@ -6,6 +6,7 @@
 use Mockery;
 use WP_Rocket\Tests\Fixtures\WP_Filesystem_Direct;
 use WP_Rocket\Tests\Integration\AdminTestCase;
+use WP_Rocket\Tests\Integration\DBTrait;
 
 /**
  * Test class covering \WP_Rocket\Engine\Optimization\RUCSS\Admin\Subscriber::display_no_table_notice
@@ -14,6 +15,8 @@
  * @group AdminOnly
  */
 class Test_DisplayNoTableNotice extends AdminTestCase {
+	use DBTrait;
+
 	protected $rucss;
 
 	public function set_up() {
@@ -39,6 +42,10 @@ public function tear_down() {
 	public function testShouldDoAsExpected( $config, $expected ) {
 		$this->rucss = $config['remove_unused_css'];
 
+		if ( ! $config['table_exists'] ) {
+			self::uninstallAll();
+		}
+
 		add_filter( 'pre_get_rocket_option_remove_unused_css', [ $this, 'rucss' ] );
 
 		$filesystem_mock = Mockery::mock( WP_Filesystem_Direct::class );