Changeset 34802 for trunk/tests/phpunit/includes/testcase.php

Timestamp:

10/03/2015 05:14:12 PM (11 years ago)

Author:

DrewAPicture

Message:

Tests: Introduce WP_UnitTestCase::reset_permalinks(), an attempt to DRY up logic for resetting and restoring default permalinks on setUp() and tearDown().

See #33968.

File:

• trunk/tests/phpunit/includes/testcase.php (modified) (4 diffs)

trunk/tests/phpunit/includes/testcase.php

@@ -16 +16 @@
 
         protected $db_version;
+
+        public static $default_permalink_structure;
 
         /**
@@ -71 +73 @@
                         add_filter( 'pre_option_db_version', array( $this, 'db_version' ) );
                 }
+
+                self::$default_permalink_structure = get_option( 'permalink_structure' );
+
+                $this->reset_permalinks();
         }
 
@@ -100 +106 @@
                 $this->_restore_hooks();
                 wp_set_current_user( 0 );
+
+                $this->reset_permalinks( $restore = true );
         }
 
@@ -646 +654 @@
                 return $this->db_version;
         }
+
+        /**
+         * Utility method that resets permalinks and flushes rewrites.
+         *
+         * Collects the current permalink structure and stores it in a class property for use
+         * by sub-classes.
+         *
+         * @since 4.4.0
+         *
+         * @global WP_Rewrite $wp_rewrite
+         *
+         * @param bool $restore_default Optional. Whether to restore the default permalink structure.
+         *                              Default false.
+         */
+        public function reset_permalinks( $restore_default = false ) {
+                global $wp_rewrite;
+
+                if ( ! $restore_default ) {
+                        $wp_rewrite->init();
+                        $wp_rewrite->set_permalink_structure( '' );
+                } else {
+                        $wp_rewrite->set_permalink_structure( self::$default_permalink_structure );
+                }
+
+                $wp_rewrite->flush_rules();
+        }
 }