Ticket #36086: 36086.diff

tests/phpunit/tests/general/template.php

@@ -192 +192 @@
                 $this->site_icon_id = $this->_make_attachment( $upload );
                 return $this->site_icon_id;
         }
+
+        /**
+         * @group custom_logo
+         *
+         * @since 4.5.0
+         */
+        function test_has_custom_logo() {
+                $this->assertFalse( has_custom_logo() );
+
+                $this->_set_custom_logo();
+                $this->assertTrue( has_custom_logo() );
+
+                $this->_remove_custom_logo();
+                $this->assertFalse( has_custom_logo() );
+        }
+
+        /**
+         * @group custom_logo
+         *
+         * @since 4.5.0
+         */
+        function test_get_custom_logo() {
+                $this->assertEmpty( get_custom_logo() );
+
+                $this->_set_custom_logo();
+                $custom_logo = get_custom_logo();
+                $this->assertNotEmpty( $custom_logo );
+                $this->assertInternalType( 'string', $custom_logo );
+
+                $this->_remove_custom_logo();
+                $this->assertEmpty( get_custom_logo() );
+        }
+
+        /**
+         * @group custom_logo
+         *
+         * @since 4.5.0
+         */
+        function test_the_custom_logo() {
+                $this->expectOutputString( '' );
+                the_custom_logo();
+
+                $this->_set_custom_logo();
+                $this->expectOutputString( get_custom_logo() );
+                the_custom_logo();
+                $this->_remove_custom_logo();
+        }
+
+        /**
+         * Sets a site icon in options for testing.
+         *
+         * @since 4.5.0
+         */
+        function _set_custom_logo() {
+                if ( ! $this->custom_logo_id ) {
+                        $this->_insert_custom_logo();
+                }
+
+                set_theme_mod( 'custom_logo', $this->custom_logo_id );
+        }
+
+        /**
+         * Removes the site icon from options.
+         *
+         * @since 4.5.0
+         */
+        function _remove_custom_logo() {
+                remove_theme_mod( 'custom_logo' );
+        }
+
+        /**
+         * Inserts an attachment for testing custom logos.
+         *
+         * @since 4.5.0
+         */
+        function _insert_custom_logo() {
+                $filename = DIR_TESTDATA . '/images/test-image.jpg';
+                $contents = file_get_contents( $filename );
+                $upload   = wp_upload_bits( basename( $filename ), null, $contents );
+
+                // Save the data.
+                $this->custom_logo_id = $this->_make_attachment( $upload );
+                return $this->custom_logo_id;
+        }
 }

tests/phpunit/tests/image/custom_logo.php

@@ +1 @@
+<?php
+
+/**
+ * Tests for the WP_Custom_logo class.
+ *
+ * @group custom_logo
+ */
+class Tests_WP_Custom_Logo extends WP_UnitTestCase {
+        public $wp_custom_logo;
+        public $attachment_id = 0;
+
+        function setUp() {
+                parent::setUp();
+
+                require_once ABSPATH . 'wp-admin/includes/class-wp-custom-logo.php';
+                $this->wp_custom_logo = $GLOBALS['wp_custom_logo'];
+        }
+
+        function tearDown() {
+                $this->custom_logo = null;
+                $this->remove_added_uploads();
+                parent::tearDown();
+        }
+
+        function test_delete_attachment_data() {
+                $attachment_id = $this->_insert_attachment();
+                set_theme_mod( 'custom_logo', $attachment_id );
+
+                wp_delete_attachment( $attachment_id, true );
+
+                $this->assertFalse( get_theme_mod( 'custom_logo' ) );
+        }
+
+        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( basename( $filename ), null, $contents );
+
+                $this->attachment_id = $this->_make_attachment( $upload );
+                return $this->attachment_id;
+        }
+}