- update WPGraphQLAcfTestCase to create 2 images for use in testing (1 as the published image, 1 as the preview image, for example)

- update ImageFieldTest to test querying the image field on a post and querying the image field on a preview

Author: jasonbahl

tests/_support/WPUnit/WPGraphQLAcfTestCase.php

@@ -78,12 +78,22 @@ class WPGraphQLAcfTestCase extends \Tests\WPGraphQL\TestCase\WPGraphQLTestCase {
 	public $test_image;
 
 	/**
-	 * @var \WP_Post
+	 * @var string
+	 */
+	public $test_image_2;
+
+	/**
+	 * @var int
 	 */
 	public $imageId;
 
 	/**
-	 * @var \WP_Post
+	 * @var int
+	 */
+	public $imageId_2;
+
+	/**
+	 * @var int
 	 */
 	public $fileId;
 
@@ -103,6 +113,7 @@ public function setUp(): void {
 		$this->is_acf_pro = in_array( 'advanced-custom-fields-pro/acf.php', $active_plugins, true );
 
 		$this->test_image = dirname( __FILE__, 3 ) . '/_data/images/test.png';
+		$this->test_image_2 = dirname( __FILE__, 3 ) . '/_data/images/test2.png';
 
 		// create users for use within tests
 		$this->admin = self::factory()->user->create_and_get( [ 'role' => 'administrator' ] );
@@ -165,6 +176,7 @@ public function setUp(): void {
 		$this->fileId = self::factory()->attachment->create_upload_object( $this->test_image, 0 );
 
 		$this->imageId = self::factory()->attachment->create_upload_object( $this->test_image, $this->published_post->ID );
+		$this->imageId_2 = self::factory()->attachment->create_upload_object( $this->test_image_2, $this->published_post->ID );
 
 		$this->menu_location_name = 'test-location';
 		add_theme_support( 'nav_menus' );

tests/wpunit/FieldTypes/ImageFieldTest.php

@@ -9,7 +9,6 @@ public function setUp(): void {
 		parent::setUp();
 	}
 
-
 	/**
 	 * @return void
 	 */
@@ -29,6 +28,44 @@ public function get_expected_field_resolve_kind(): ?string {
 		return 'OBJECT';
 	}
 
+	public function get_query_fragment(): string {
+		return '
+		fragment AcfTestGroupFragment on AcfTestGroup {
+		  testImage {
+		    node {
+		      __typename
+		      databaseId
+		    }
+		  }
+		}';
+	}
+
+	public function get_data_to_store() {
+		return $this->imageId;
+	}
+
+	public function get_expected_value() {
+		return [
+			'node' => [
+				'__typename' => 'MediaItem',
+				'databaseId' => $this->imageId
+			]
+		];
+	}
+
+	public function get_preview_data_to_store() {
+		return $this->imageId_2;
+	}
+
+	public function get_expected_preview_value() {
+		return [
+			'node' => [
+				'__typename' => 'MediaItem',
+				'databaseId' => $this->imageId_2
+			]
+		];;
+	}
+
 	public function get_block_query_fragment() {
 		return '
 		fragment BlockQueryFragment on AcfTestGroup {
@@ -46,6 +83,7 @@ public function get_block_data_to_store() {
 		return $this->imageId;
 	}
 
+
 	public function get_expected_block_fragment_response() {
 		return [
 			'node' => [