wordpress-develop/tests/phpunit/tests/link/getEditPostLink.php

<?php
/**
 * Tests the `get_edit_post_link()` function.
 *
 * @since 6.3.0
 *
 * @group link
 *
 * @covers ::get_edit_post_link
 */
class Tests_Link_GetEditPostLink extends WP_UnitTestCase {
	/**
	 * The name of the theme to use for the test.
	 *
	 * @since 6.3.0
	 * @var string
	 */
	const TEST_THEME = 'block-theme';

	/**
	 * The id of the user to use for the test.
	 *
	 * @since 6.3.0
	 * @var int
	 */
	private static $admin_id;

	/**
	 * Creates admin user before tests run.
	 *
	 * @param WP_UnitTest_Factory $factory
	 */
	public static function wpSetUpBeforeClass( WP_UnitTest_Factory $factory ) {
		// Create an admin user because get_edit_post_link() requires 'edit_post' capability.
		self::$admin_id = $factory->user->create( array( 'role' => 'administrator' ) );
	}

	/**
	 * Performs setup tasks for every test.
	 *
	 * @since 6.3.0
	 */
	public function set_up() {
		parent::set_up();
		wp_set_current_user( self::$admin_id );
		switch_theme( self::TEST_THEME );
	}

	/**
	 * Tests getting the edit post link for a post.
	 */
	public function test_get_edit_post_link() {
		$post                 = self::factory()->post->create_and_get(
			array(
				'post_type'   => 'post',
				'post_title'  => 'Test Post',
				'post_name'   => 'test-post',
				'post_status' => 'publish',
			)
		);
		$post_type_object     = get_post_type_object( $post->post_type );
		$link_default_context = admin_url( sprintf( $post_type_object->_edit_link . '&amp;action=edit', $post->ID ) );
		$link_custom_context  = admin_url( sprintf( $post_type_object->_edit_link . '&action=edit', $post->ID ) );

		$this->assertSame( $link_default_context, get_edit_post_link( $post ), 'Second argument `$context` has a default context of `"display"`.' );
		$this->assertSame( $link_custom_context, get_edit_post_link( $post, 'something-else' ), 'Pass non-default value in second argument.' );
	}

	/**
	 * Tests getting the edit post link for a template post type.
	 *
	 * @ticket 57709
	 */
	public function test_get_edit_post_link_for_wp_template_post_type() {
		$template_post = self::factory()->post->create_and_get(
			array(
				'post_type'    => 'wp_template',
				'post_name'    => 'my_template',
				'post_title'   => 'My Template',
				'post_content' => 'Content',
				'post_excerpt' => 'Description of my template',
				'tax_input'    => array(
					'wp_theme' => array(
						self::TEST_THEME,
					),
				),
			)
		);

		wp_set_post_terms( $template_post->ID, self::TEST_THEME, 'wp_theme' );

		$post_type_object     = get_post_type_object( $template_post->post_type );
		$link_default_context = admin_url( sprintf( $post_type_object->_edit_link, $template_post->post_type, get_stylesheet() . '%2F%2Fmy_template' ) );
		$link_custom_context  = admin_url( sprintf( $post_type_object->_edit_link, $template_post->post_type, get_stylesheet() . '%2F%2Fmy_template' ) );

		$this->assertSame( $link_default_context, get_edit_post_link( $template_post ), 'Second argument `$context` has a default context of `"display"`.' );
		$this->assertSame( $link_custom_context, get_edit_post_link( $template_post, 'something-else' ), 'Pass non-default value in second argument.' );
	}

	/**
	 * Tests getting the edit post link for a template part post type.
	 *
	 * @ticket 57709
	 */
	public function test_get_edit_post_link_for_wp_template_part_post_type() {
		$template_part_post = self::factory()->post->create_and_get(
			array(
				'post_type'    => 'wp_template_part',
				'post_name'    => 'my_template_part',
				'post_title'   => 'My Template Part',
				'post_content' => 'Content',
				'post_excerpt' => 'Description of my template part',
				'tax_input'    => array(
					'wp_theme'              => array(
						self::TEST_THEME,
					),
					'wp_template_part_area' => array(
						WP_TEMPLATE_PART_AREA_HEADER,
					),
				),
			)
		);

		wp_set_post_terms( $template_part_post->ID, WP_TEMPLATE_PART_AREA_HEADER, 'wp_template_part_area' );
		wp_set_post_terms( $template_part_post->ID, self::TEST_THEME, 'wp_theme' );

		$post_type_object     = get_post_type_object( $template_part_post->post_type );
		$link_default_context = admin_url( sprintf( $post_type_object->_edit_link, $template_part_post->post_type, get_stylesheet() . '%2F%2Fmy_template_part' ) );
		$link_custom_context  = admin_url( sprintf( $post_type_object->_edit_link, $template_part_post->post_type, get_stylesheet() . '%2F%2Fmy_template_part' ) );

		$this->assertSame( $link_default_context, get_edit_post_link( $template_part_post ), 'Second argument `$context` has a default context of `"display"`.' );
		$this->assertSame( $link_custom_context, get_edit_post_link( $template_part_post, 'something-else' ), 'Pass non-default value in second argument.' );
	}

	/**
	 * Tests getting the edit post link for a wp_navigation post type.
	 *
	 * @ticket 58589
	 * */
	public function test_get_edit_post_link_for_wp_navigation_post_type() {
		$navigation_post = self::factory()->post->create_and_get(
			array(
				'post_type'    => 'wp_navigation',
				'post_name'    => 'my_navigation',
				'post_title'   => 'My Navigation',
				'post_content' => '<!-- wp:navigation-link {"label":"WordPress","type":"custom","url":"http://www.wordpress.org/","kind":"custom"} /-->',
				'post_excerpt' => 'Description of my Navigation',
			)
		);

		$post_type_object = get_post_type_object( $navigation_post->post_type );

		$link_default_context = admin_url( sprintf( $post_type_object->_edit_link, $navigation_post->ID ) );
		$link_custom_context  = admin_url( sprintf( $post_type_object->_edit_link, $navigation_post->ID ) );

		$this->assertSame( $link_default_context, get_edit_post_link( $navigation_post ), 'Second argument `$context` has a default context of `"display"`.' );
		$this->assertSame( $link_custom_context, get_edit_post_link( $navigation_post, 'something-else' ), 'Pass non-default value in second argument.' );
	}
}