<?php
// phpcs:disable WordPress.DateTime.RestrictedFunctions.timezone_change_date_default_timezone_set

/**
 * Class ActionScheduler_UnitTestCase
 */
class ActionScheduler_UnitTestCase extends WP_UnitTestCase {

	/**
	 * Timezone string.
	 *
	 * @var string
	 */
	protected $existing_timezone;

	/**
	 * Perform test set-up work.
	 */
	public function set_up() {
		ActionScheduler_Callbacks::add_callbacks();
		parent::set_up();
	}

	/**
	 * Perform test tear-down work.
	 */
	public function tear_down() {
		ActionScheduler_Callbacks::remove_callbacks();
		parent::tear_down();
	}

	/**
	 * Counts the number of test cases executed by run(TestResult result).
	 *
	 * @return int
	 */
	public function count(): int {
		return ( 'UTC' === date_default_timezone_get() ) ? 2 : 3;
	}

	/**
	 * We want to run every test multiple times using a different timezone to make sure
	 * that they are unaffected by changes to PHP's timezone.
	 *
	 * @param null|\PHPUnit\Framework\TestResult $result Test result.
	 */
	public function run( ?PHPUnit\Framework\TestResult $result = null ): \PHPUnit\Framework\TestResult {

		if ( is_null( $result ) ) {
			$result = $this->createResult();
		}

		$this->existing_timezone = date_default_timezone_get();

		if ( 'UTC' !== $this->existing_timezone ) {
			date_default_timezone_set( 'UTC' );
			$result->run( $this );
		}

		date_default_timezone_set( 'Pacific/Fiji' ); // UTC+12.
		$result->run( $this );

		date_default_timezone_set( 'Pacific/Tahiti' ); // UTC-10: it's a magical place.
		$result->run( $this );

		date_default_timezone_set( $this->existing_timezone );

		return $result;
	}
}