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

Timestamp:

10/03/2015 05:14:12 PM (10 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();
+    }
 }