Ticket #31271: 31271.4.diff

src/wp-includes/post-template.php

@@ -1629 +1629 @@
  * Whether currently in a page template.
  *
  * This template tag allows you to determine if you are in a page template.
- * You can optionally provide a template name and then the check will be
- * specific to that template.
+ * You can optionally provide a template name or array of template names
+ * and then the check will be specific to that template.
  *
  * @since 2.5.0
+ * @since 4.2.0 The `$template` parameter was changed to accept an array of page templates.
  *
- * @param string $template The specific template name if specific matching is required.
+ * @param string|array $template The specific template name or array of templates to match.
  * @return bool True on success, false on failure.
  */
 function is_page_template( $template = '' ) {
@@ -1649 +1650 @@
         if ( $template == $page_template )
                 return true;
 
+        if ( is_array( $template ) ) {
+                if ( in_array( 'default', $template, true ) && ! $page_template
+                        || in_array( $page_template, $template, true )
+                ) {
+                        return true;
+                }
+        }
+
         if ( 'default' == $template && ! $page_template )
                 return true;
 

tests/phpunit/tests/query/conditionals.php

@@ -960 +960 @@
                 $this->assertFalse( $q->is_page( $p2_name ) );
                 $this->assertFalse( $q->is_page( $p2 ) );
         }
+
+        function test_is_page_template() {
+                $post_id = $this->factory->post->create( array( 'post_type' => 'page' ) );
+                update_post_meta($post_id, '_wp_page_template', 'example.php');
+                $this->go_to( "/?page_id=$post_id" );
+                $this->assertTrue( is_page_template( 'example.php' ) );
+        }
+
+        function test_is_page_template_default() {
+                $post_id = $this->factory->post->create( array( 'post_type' => 'page' ) );
+                $this->go_to( "/?page_id=$post_id" );
+                $this->assertTrue( is_page_template( 'default' ) );
+                $this->assertTrue( is_page_template( array( 'random', 'default' ) ) );
+        }
+
+        /**
+         * @ticket 31271
+         */
+        function test_is_page_template_array() {
+                $post_id = $this->factory->post->create( array( 'post_type' => 'page' ) );
+                update_post_meta($post_id, '_wp_page_template', 'example.php');
+                $this->go_to( "/?page_id=$post_id" );
+                $this->assertFalse( is_page_template( array( 'test.php' ) ) );
+                $this->assertTrue( is_page_template( array('test.php', 'example.php') ) );
+        }
 }