wordpress-develop/tests/phpunit/tests/image/siteIcon.php

<?php
/**
 * Tests for the WP_Site_Icon class.
 *
 * @group site_icon
 */

require_once ABSPATH . 'wp-admin/includes/class-wp-site-icon.php';

class Tests_WP_Site_Icon extends WP_UnitTestCase {
	protected $wp_site_icon;

	public $attachment_id = 0;

	public function set_up() {
		parent::set_up();

		$this->wp_site_icon = new WP_Site_Icon();
	}

	public function tear_down() {
		$this->remove_custom_logo();
		$this->remove_added_uploads();
		parent::tear_down();
	}

	private function remove_custom_logo() {
		remove_theme_mod( 'custom_logo' );
	}

	public function test_intermediate_image_sizes() {
		$image_sizes = $this->wp_site_icon->intermediate_image_sizes( array() );

		$sizes = array();
		foreach ( $this->wp_site_icon->site_icon_sizes as $size ) {
			$sizes[] = 'site_icon-' . $size;
		}

		$this->assertSame( $sizes, $image_sizes );
	}

	public function test_intermediate_image_sizes_with_filter() {
		add_filter( 'site_icon_image_sizes', array( $this, 'custom_test_sizes' ) );
		$image_sizes = $this->wp_site_icon->intermediate_image_sizes( array() );

		$sizes = array();
		foreach ( $this->wp_site_icon->site_icon_sizes as $size ) {
			$sizes[] = 'site_icon-' . $size;
		}

		// Is our custom icon size there?
		$this->assertContains( 'site_icon-321', $image_sizes );

		// All icon sizes should be part of the array, including sizes added through the filter.
		$this->assertSame( $sizes, $image_sizes );

		// Remove custom size.
		unset( $this->wp_site_icon->site_icon_sizes[ array_search( 321, $this->wp_site_icon->site_icon_sizes, true ) ] );
		// Remove the filter we added.
		remove_filter( 'site_icon_image_sizes', array( $this, 'custom_test_sizes' ) );
	}

	public function test_additional_sizes() {
		$image_sizes = $this->wp_site_icon->additional_sizes( array() );

		$sizes = array();
		foreach ( $this->wp_site_icon->site_icon_sizes as $size ) {
			$sizes[ 'site_icon-' . $size ] = array(
				'width ' => $size,
				'height' => $size,
				'crop'   => true,
			);
		}

		$this->assertSame( $sizes, $image_sizes );
	}

	public function test_additional_sizes_with_filter() {
		add_filter( 'site_icon_image_sizes', array( $this, 'custom_test_sizes' ) );
		$image_sizes = $this->wp_site_icon->additional_sizes( array() );

		$sizes = array();
		foreach ( $this->wp_site_icon->site_icon_sizes as $size ) {
			$sizes[ 'site_icon-' . $size ] = array(
				'width ' => $size,
				'height' => $size,
				'crop'   => true,
			);
		}

		// Is our custom icon size there?
		$this->assertArrayHasKey( 'site_icon-321', $image_sizes );

		// All icon sizes should be part of the array, including sizes added through the filter.
		$this->assertSame( $sizes, $image_sizes );

		// Remove custom size.
		unset( $this->wp_site_icon->site_icon_sizes[ array_search( 321, $this->wp_site_icon->site_icon_sizes, true ) ] );
	}

	public function test_create_attachment_object() {
		$attachment_id = $this->insert_attachment();
		$parent_url    = get_post( $attachment_id )->guid;
		$cropped       = str_replace( wp_basename( $parent_url ), 'cropped-test-image.jpg', $parent_url );

		$object = $this->wp_site_icon->create_attachment_object( $cropped, $attachment_id );

		$this->assertSame( $object['post_title'], 'cropped-test-image.jpg' );
		$this->assertSame( $object['context'], 'site-icon' );
		$this->assertSame( $object['post_mime_type'], 'image/jpeg' );
		$this->assertSame( $object['post_content'], $cropped );
		$this->assertSame( $object['guid'], $cropped );
	}

	public function test_insert_cropped_attachment() {
		$attachment_id = $this->insert_attachment();
		$parent_url    = get_post( $attachment_id )->guid;
		$cropped       = str_replace( wp_basename( $parent_url ), 'cropped-test-image.jpg', $parent_url );

		$object     = $this->wp_site_icon->create_attachment_object( $cropped, $attachment_id );
		$cropped_id = $this->wp_site_icon->insert_attachment( $object, $cropped );

		$this->assertIsInt( $cropped_id );
		$this->assertGreaterThan( 0, $cropped_id );
	}

	public function test_delete_attachment_data() {
		$attachment_id = $this->insert_attachment();
		update_option( 'site_icon', $attachment_id );

		wp_delete_attachment( $attachment_id, true );

		$this->assertFalse( get_option( 'site_icon', false ) );
	}

	/**
	 * @ticket 34368
	 */
	public function test_get_post_metadata() {
		$attachment_id = $this->insert_attachment();
		update_option( 'site_icon', $attachment_id );

		$this->wp_site_icon->get_post_metadata( '', $attachment_id, '_some_post_meta', true );
		$this->assertFalse( has_filter( 'intermediate_image_sizes', array( $this->wp_site_icon, 'intermediate_image_sizes' ) ) );

		$this->wp_site_icon->get_post_metadata( '', $attachment_id, '_wp_attachment_backup_sizes', true );
		$this->assertSame( 10, has_filter( 'intermediate_image_sizes', array( $this->wp_site_icon, 'intermediate_image_sizes' ) ) );

		wp_delete_attachment( $attachment_id, true );
	}

	public function custom_test_sizes( $sizes ) {
		$sizes[] = 321;

		return $sizes;
	}

	private function insert_attachment() {
		if ( $this->attachment_id ) {
			return $this->attachment_id;
		}

		$filename = DIR_TESTDATA . '/images/test-image.jpg';
		$contents = file_get_contents( $filename );

		$upload = wp_upload_bits( wp_basename( $filename ), null, $contents );

		$this->attachment_id = $this->_make_attachment( $upload );
		return $this->attachment_id;
	}
}