Add image support for productBrand query

Author: webwiz94

includes/class-type-registry.php

@@ -137,6 +137,7 @@ public function init() {
 		/**
 		 * Object fields.
 		 */
+		Type\WPObject\Product_Brand_Type::register_fields();
 		Type\WPObject\Product_Category_Type::register_fields();
 		Type\WPObject\Root_Query::register_fields();
 

includes/class-wp-graphql-woocommerce.php

@@ -290,6 +290,7 @@ private function includes() {
 			require $include_directory_path . 'type/object/class-product-attribute-object-type.php';
 			require $include_directory_path . 'type/object/class-product-attribute-term-object-type.php';
 			require $include_directory_path . 'type/object/class-product-attribute-types.php';
+			require $include_directory_path . 'type/object/class-product-brand-type.php';
 			require $include_directory_path . 'type/object/class-product-category-type.php';
 			require $include_directory_path . 'type/object/class-product-download-type.php';
 			require $include_directory_path . 'type/object/class-product-types.php';

includes/type/object/class-product-brand-type.php

@@ -0,0 +1,43 @@
+<?php
+/**
+ * WPObjectType - ProductBrand
+ *
+ * Registers fields for the ProductBrand type
+ *
+ * @package WPGraphQL\WooCommerce\Type\WPObject
+ * @since   1.0.3
+ */
+
+namespace WPGraphQL\WooCommerce\Type\WPObject;
+
+use WPGraphQL\AppContext;
+
+/**
+ * Class - Product_Brand_Type
+ */
+class Product_Brand_Type {
+	/**
+	 * Registers fields to ProductBrand.
+	 *
+	 * @return void
+	 */
+	public static function register_fields() {
+		register_graphql_fields(
+			'ProductBrand',
+			[
+				'image' => [
+					'type'        => 'MediaItem',
+					'description' => static function () {
+						return __( 'Product brand image', 'wp-graphql-woocommerce' );
+					},
+					'resolve'     => static function ( $source, array $args, AppContext $context ) {
+						$thumbnail_id = get_term_meta( $source->term_id, 'thumbnail_id', true );
+						return ! empty( $thumbnail_id )
+							? $context->get_loader( 'post' )->load_deferred( $thumbnail_id )
+							: null;
+					},
+				],
+			]
+		);
+	}
+}

tests/wpunit/ProductBrandQueriesTest.php

@@ -96,6 +96,58 @@ public function testSingleProductBrandBySlug() {
 		$this->assertQuerySuccessful( $response, $expected );
 	}
 
+	/**
+	 * Test that productBrand image resolves from WooCommerce term thumbnail meta.
+	 */
+	public function testProductBrandImage() {
+		$brand_id = $this->createProductBrand( 'brand-with-image' );
+		$image_id = $this->factory->post->create(
+			[
+				'post_author' => $this->shop_manager,
+				'post_status' => 'publish',
+				'post_title'  => 'Brand Image',
+				'post_type'   => 'attachment',
+			]
+		);
+
+		update_term_meta( $brand_id, 'thumbnail_id', $image_id );
+
+		$query = '
+			query ($id: ID!) {
+				productBrand(id: $id, idType: SLUG) {
+					databaseId
+					image {
+						id
+					}
+				}
+				productBrands(first: 100) {
+					nodes {
+						slug
+						image {
+							id
+						}
+					}
+				}
+			}
+		';
+
+		$variables = [ 'id' => 'brand-with-image' ];
+		$response  = $this->graphql( compact( 'query', 'variables' ) );
+		$expected  = [
+			$this->expectedField( 'productBrand.databaseId', $brand_id ),
+			$this->expectedField( 'productBrand.image.id', $this->toRelayId( 'post', $image_id ) ),
+			$this->expectedNode(
+				'productBrands.nodes',
+				[
+					$this->expectedField( 'slug', 'brand-with-image' ),
+					$this->expectedField( 'image.id', $this->toRelayId( 'post', $image_id ) ),
+				]
+			),
+		];
+
+		$this->assertQuerySuccessful( $response, $expected );
+	}
+
 	/**
 	 * Test the connection from productBrand to products.
 	 */