Add test for Customize_Snapshot_Manager::should_import_and_preview_snapshot()

westonruter · westonruter · commit be2a9cd3ee6d · 2016-08-09T00:05:25.000-07:00
diff --git a/php/class-customize-snapshot-manager.php b/php/class-customize-snapshot-manager.php
@@ -292,8 +292,8 @@ public function is_theme_active() {
 	public function should_import_and_preview_snapshot( Customize_Snapshot $snapshot ) {
 		global $pagenow;
 
-		// Ignore if in the admin.
-		if ( is_admin() && ! ( defined( 'DOING_AJAX' ) && DOING_AJAX ) && 'customize.php' !== $pagenow ) {
+		// Ignore if in the admin, but not Admin Ajax or Customizer.
+		if ( is_admin() && ! in_array( $pagenow, array( 'admin-ajax.php', 'customize.php' ), true ) ) {
 			return false;
 		}
 
@@ -316,7 +316,7 @@ public function should_import_and_preview_snapshot( Customize_Snapshot $snapshot
 		 * Note that wp.customize.Snapshots.extendPreviewerQuery() will extend the
 		 * previewer data to include the current snapshot UUID.
 		 */
-		if ( count( $this->customize_manager->unsanitized_post_values() ) > 0 ) {
+		if ( $this->customize_manager && count( $this->customize_manager->unsanitized_post_values() ) > 0 ) {
 			return false;
 		}
 
diff --git a/tests/php/test-class-customize-snapshot-manager.php b/tests/php/test-class-customize-snapshot-manager.php
@@ -377,7 +377,55 @@ public function test_is_theme_active() {
 	 * @covers Customize_Snapshot_Manager::should_import_and_preview_snapshot()
 	 */
 	public function test_should_import_and_preview_snapshot() {
-		$this->markTestIncomplete();
+		global $pagenow, $wp_customize;
+		$_REQUEST['customize_snapshot_uuid'] = self::UUID;
+		$manager = $this->plugin->customize_snapshot_manager;
+		$post_id = $manager->post_type->save( array(
+			'uuid' => self::UUID,
+			'data' => array( 'blogname' => array( 'value' => 'Foo' ) ),
+		) );
+		$snapshot = new Customize_Snapshot( $manager, self::UUID );
+
+		// Not if admin.
+		set_current_screen( 'posts' );
+		$pagenow = 'posts.php'; // WPCS: global override ok.
+		$this->assertTrue( is_admin() );
+		$this->assertFalse( $manager->should_import_and_preview_snapshot( $snapshot ) );
+
+		// Not if theme switch error.
+		set_current_screen( 'customize' );
+		$pagenow = 'customize.php'; // WPCS: global override ok.
+		update_post_meta( $post_id, '_snapshot_theme', 'Foo' );
+		$this->assertFalse( $manager->should_import_and_preview_snapshot( $snapshot ) );
+		delete_post_meta( $post_id, '_snapshot_theme' );
+
+		// Not if customize_save.
+		$_REQUEST['action'] = 'customize_save';
+		$this->assertFalse( $manager->should_import_and_preview_snapshot( $snapshot ) );
+		unset( $_REQUEST['action'] );
+
+		// Not if published snapshot.
+		$manager->post_type->save( array(
+			'uuid' => self::UUID,
+			'status' => 'publish',
+		) );
+		$this->assertFalse( $manager->should_import_and_preview_snapshot( $snapshot ) );
+		$manager->post_type->save( array(
+			'uuid' => self::UUID,
+			'status' => 'draft',
+		) );
+
+		// Not if unsanitized post values is not empty.
+		$manager->customize_manager = new \WP_Customize_Manager();
+		$wp_customize = $manager->customize_manager; // WPCS: global override ok.
+		$wp_customize->set_post_value( 'name', 'value' );
+		$this->assertNotEmpty( $manager->customize_manager->unsanitized_post_values() );
+		$this->assertFalse( $manager->should_import_and_preview_snapshot( $snapshot ) );
+
+		// OK.
+		$manager->customize_manager = new \WP_Customize_Manager();
+		$wp_customize = $manager->customize_manager; // WPCS: global override ok.
+		$this->assertTrue( $manager->should_import_and_preview_snapshot( $snapshot ) );
 	}
 
 	/**
