- update ACfFieldTestCase to provide support for testing querying against previews

- update TextFieldTest.php to test querying against a preview

Author: jasonbahl

tests/_support/WPUnit/AcfFieldTestCase.php

@@ -75,6 +75,13 @@ public function get_data_to_store() {
 		return 'text value...';
 	}
 
+	/**
+	 * @return mixed
+	 */
+	public function get_preview_data_to_store() {
+		return 'preview - ' . $this->get_data_to_store();
+	}
+
 	/**
 	 * Override this in the testing class to test the field against blocks
 	 *
@@ -123,6 +130,16 @@ public function get_expected_value() {
 		return 'null';
 	}
 
+	/**
+	 * Test class should override this with the expected value
+	 *
+	 * @return mixed
+	 */
+	public function get_expected_preview_value() {
+		return 'null';
+	}
+
+
 	/**
 	 * @return string
 	 */
@@ -1026,6 +1043,111 @@ public function testQueryFieldOnPostReturnsExpectedValue() {
 
 	}
 
+	public function testQueryFieldOnPostAsPreviewReturnsExpectedValue() {
+
+		// disable the block editor
+		add_filter('use_block_editor_for_post', '__return_false');
+
+		$field_key = $this->register_acf_field();
+
+		// Save data to the post
+		update_field( $field_key, $this->get_data_to_store(), $this->published_post->ID );
+
+		$fragment = $this->get_query_fragment();
+
+		if ( 'null' === $fragment ) {
+			$this->markTestIncomplete( 'get_query_fragment() not defined' );
+		}
+
+		$expected_value = $this->get_expected_value();
+
+		if ( 'null' === $expected_value ) {
+			$this->markTestIncomplete( 'get_expected_value() not defined' );
+		}
+
+		$expected_preview_value = $this->get_expected_preview_value();
+
+		if ( 'null' === $expected_preview_value ) {
+			$this->markTestIncomplete( 'get_expected_preview_value() not defined' );
+		}
+
+		$preview_data_to_store = $this->get_preview_data_to_store();
+
+		if ( 'null' === $preview_data_to_store ) {
+			$this->markTestIncomplete( 'get_preview_data_to_store() not defined' );
+		}
+
+		$query = '
+		query AcfFieldOnPost ($id: ID! $asPreview:Boolean) {
+		  post( id: $id idType: DATABASE_ID asPreview: $asPreview) {
+		    databaseId
+		    __typename
+		    title
+		    ...on WithAcfAcfTestGroup {
+		      acfTestGroup {
+		        ...AcfTestGroupFragment
+		      }
+		    }
+		  }
+		}
+		' . $fragment;
+
+		$actual = $this->graphql([
+			'query' => $query,
+			'variables' => [
+				'id' => $this->published_post->ID,
+				'asPreview' => false,
+			]
+		]);
+
+		// querying as preview without any preview revision will
+		// return the same result as querying as published
+		self::assertQuerySuccessful( $actual, [
+			$this->expectedField( 'post.databaseId', $this->published_post->ID ),
+			$this->expectedField( 'post.__typename', 'Post' ),
+			$this->expectedField( 'post.acfTestGroup.' . $this->get_formatted_field_name(), $this->get_expected_value() )
+		]);
+
+		// set the current user as admin
+		wp_set_current_user( $this->admin->ID );
+
+		// create a preview revision
+		$preview_id = wp_create_post_autosave( [
+			'post_ID' => $this->published_post->ID,
+			'post_type' => 'post',
+			'post_title' => 'Preview Title',
+		] );
+
+		if ( is_wp_error( $preview_id ) ) {
+			// Handle error; autosave creation failed
+			codecept_debug( 'Error creating autosave: ' . $preview_id->get_error_message() );
+		}
+
+		// update the preview revision with the preview data
+		update_field( $field_key, $this->get_preview_data_to_store(), $preview_id );
+
+		$actual = $this->graphql([
+			'query' => $query,
+			'variables' => [
+				'id' => $this->published_post->ID,
+				'asPreview' => true,
+			]
+		]);
+
+		self::assertQuerySuccessful( $actual, [
+			$this->expectedField( 'post.databaseId', $preview_id ),
+			$this->expectedField( 'post.__typename', 'Post' ),
+			$this->expectedField( 'post.title', 'Preview Title' ),
+			$this->expectedField( 'post.acfTestGroup.' . $this->get_formatted_field_name(), $this->get_expected_preview_value() )
+		]);
+
+		wp_delete_post( $preview_id, true );
+
+		// re-enable the block editor
+		add_filter('use_block_editor_for_post', '__return_true');
+
+	}
+
 	/**
 	 * @todo: implement the below tests
 	 */

tests/wpunit/FieldTypes/TextFieldTest.php

@@ -72,6 +72,10 @@ public function get_expected_value() {
 		return $this->get_data_to_store();
 	}
 
+	public function get_expected_preview_value() {
+		return $this->get_preview_data_to_store();
+	}
+
 	public function get_expected_block_fragment_response() {
 		return $this->get_block_data_to_store();
 	}