Ticket #32322: 32322-unit-tests.diff

tests/phpunit/tests/link.php

@@ -328 +328 @@
 
                 $this->assertEquals( $non_pretty_permalink, get_permalink( $p ) );
         }
+
+        public function test_shortlink_post_statuses_filter() {
+                update_option( 'permalink_structure','/%postname%/' );
+                register_post_type( 'wptests_pt', array( 'public' => true, 'rewrite' => array( 'slug' => 'wptests_pt' ) ) );
+                flush_rewrite_rules();
+
+                $p = $this->factory->post->create( array(
+                        'post_title'  => 'from the future',
+                        'post_status' => 'publish',
+                        'post_date'   => strftime( '%Y-%m-%d %H:%M:%S', strtotime( '+1 day' ) )
+                ) );
+                $cpt_p = $this->factory->post->create( array(
+                        'post_title'  => 'cpt from the future',
+                        'post_type'   => 'wptests_pt',
+                        'post_status' => 'publish',
+                        'post_date'   => strftime( '%Y-%m-%d %H:%M:%S', strtotime( '+1 day' ) )
+                ) );
+
+                $pretty_permalink = site_url( 'from-the-future' );
+                $cpt_pretty_permalink = add_query_arg( array(
+                        'wptests_pt' => 'cpt-from-the-future',
+                ), trailingslashit( home_url() ) );
+
+                add_filter( 'shortlink_post_statuses', function( $statuses ) {
+                        unset( $statuses[ array_search( 'future', $statuses ) ] );
+                        return $statuses;
+                } );
+
+                $this->assertEquals( $pretty_permalink, get_permalink( $p ) );
+                $this->assertEquals( $cpt_pretty_permalink, get_permalink( $cpt_p ) );
+        }
 }