Ensure setup_theme is triggered in test so that WP_Customize_Manager::$original_stylesheet will be set

westonruter · westonruter · commit 52694b69390f · 2017-03-21T13:29:08.000-07:00
diff --git a/tests/php/test-class-customize-snapshot-manager.php b/tests/php/test-class-customize-snapshot-manager.php
@@ -387,10 +387,13 @@ public function test_is_theme_active() {
 	public function test_add_snapshot_uuid_to_return_url() {
 		global $wp_version;
 		if ( version_compare( $wp_version, '4.4-beta', '>=' ) ) {
-			$_GET[ $this->front_param ] = $_REQUEST[ $this->front_param ] = self::UUID;
+			wp_set_current_user( $this->factory()->user->create( array( 'role' => 'administrator' ) ) );
+			$_GET[ $this->front_param ] = self::UUID;
+			$_REQUEST[ $this->front_param ] = self::UUID;
 			$manager = $this->get_snapshot_manager_instance( $this->plugin );
 			$manager->init();
 			$manager->ensure_customize_manager();
+			do_action( 'setup_theme' );
 			$this->assertNotContains( $this->front_param, $manager->customize_manager->get_return_url() );
 			$manager->add_snapshot_uuid_to_return_url();
 			$this->assertContains( $this->front_param, $manager->customize_manager->get_return_url() );
