aspireupdate/tests/Branding/Branding_OutputAdminNoticeTest.php

<?php
/**
 * Class Branding_OutputAdminNoticeTest
 *
 * @package AspireUpdate
 */

/**
 * Tests for Branding::output_admin_notice()
 *
 * @covers \AspireUpdate\Branding::output_admin_notice
 */
class Branding_OutputAdminNoticeTest extends WP_UnitTestCase {
	/**
	 * Test that the expected admin notice is output.
	 *
	 * @dataProvider data_screen_specific_messages
	 *
	 * @param string $hook     The current screen's hook.
	 * @param string $expected The expected substring to find.
	 */
	public function test_should_output_admin_notice( $hook, $expected ) {
		set_current_screen( $hook );

		$branding = new AspireUpdate\Branding();
		$this->assertStringContainsString( $expected, get_echo( [ $branding, 'output_admin_notice' ] ) );
	}

	/**
	 * Data provider.
	 *
	 * @return array[]
	 */
	public function data_screen_specific_messages() {
		return [
			'update-core.php'    => [
				'hook'     => 'update-core.php',
				'expected' => 'WordPress, plugin, theme and translation updates',
			],
			'plugins.php'        => [
				'hook'     => 'plugins.php',
				'expected' => 'plugin updates',
			],
			'plugin-install.php' => [
				'hook'     => 'plugin-install.php',
				'expected' => 'plugin updates',
			],
			'themes.php'         => [
				'hook'     => 'themes.php',
				'expected' => 'theme updates',
			],
			'theme-install.php'  => [
				'hook'     => 'theme-install.php',
				'expected' => 'theme updates',
			],
		];
	}

	/**
	 * Test that no admin notice is output on adjacent screens.
	 *
	 * @dataProvider data_adjacent_screens
	 *
	 * @param string $hook The current screen's hook.
	 */
	public function test_should_not_output_notice_on_adjacent_screens( $hook ) {
		set_current_screen( $hook );

		$branding = new AspireUpdate\Branding();
		$this->assertSame( '', get_echo( [ $branding, 'output_admin_notice' ] ) );
	}

	/**
	 * Data provider.
	 *
	 * @return array[]
	 */
	public function data_adjacent_screens() {
		return self::text_array_to_dataprovider(
			[
				'index.php',
				'nav-menus.php',
				'plugin-editor.php',
			]
		);
	}

	/**
	 * Test that no admin notice is output when there is no screen.
	 */
	public function test_should_not_output_notice_when_there_is_no_screen() {
		global $current_screen;
		$current_screen_backup = $current_screen;
		unset( $current_screen );

		$branding       = new AspireUpdate\Branding();
		$actual         = get_echo( [ $branding, 'output_admin_notice' ] );
		$current_screen = $current_screen_backup;

		$this->assertSame( '', $actual );
	}

	/**
	 * Test that no admin notice is output when AP_REMOVE_UI is set to true.
	 *
	 * @runInSeparateProcess
	 * @preserveGlobalState disabled
	 */
	public function test_should_not_output_notice_when_ap_remove_ui_is_true() {
		// Set to a screen that should display an admin notice.
		set_current_screen( 'plugins.php' );

		// Prevent the notice from being displayed.
		define( 'AP_REMOVE_UI', true );

		$branding = new AspireUpdate\Branding();
		$this->assertSame( '', get_echo( [ $branding, 'output_admin_notice' ] ) );
	}
}