Cart* test refactored heavily to extend the WPGraphQLTestCase

Author: kidunot89

includes/type/object/class-cart-type.php

@@ -398,21 +398,21 @@ public static function register_cart_fee() {
 						'type'        => 'Boolean',
 						'description' => __( 'Is fee taxable?', 'wp-graphql-woocommerce' ),
 						'resolve'     => function( $source ) {
-							return ! empty( $source->taxable ) ? $source->taxable : null;
+							return ! is_null( $source->taxable ) ? $source->taxable : null;
 						},
 					),
 					'amount'   => array(
 						'type'        => 'Float',
 						'description' => __( 'Fee amount', 'wp-graphql-woocommerce' ),
 						'resolve'     => function( $source ) {
-							return ! empty( $source->amount ) ? $source->amount : null;
+							return ! is_null( $source->amount ) ? $source->amount : 0;
 						},
 					),
 					'total'    => array(
 						'type'        => 'Float',
 						'description' => __( 'Fee total', 'wp-graphql-woocommerce' ),
 						'resolve'     => function( $source ) {
-							return ! empty( $source->total ) ? $source->total : null;
+							return ! is_null( $source->total ) ? $source->total : 0;
 						},
 					),
 				),

tests/_support/Factory/CartFactory.php

@@ -2,7 +2,7 @@
 /**
  * Factory class for the WooCommerce's Cart data objects.
  *
- * @since v0.6.0
+ * @since v0.6.1
  * @package Tests\WPGraphQL\WooCommerce\Factory
  */
 

tests/_support/Factory/CouponFactory.php

@@ -0,0 +1,75 @@
+<?php
+/**
+ * Factory class for the WooCommerce's coupon data objects.
+ *
+ * @since v0.6.1
+ * @package Tests\WPGraphQL\WooCommerce\Factory
+ */
+
+namespace Tests\WPGraphQL\WooCommerce\Factory;
+
+use Tests\WPGraphQL\WooCommerce\Utils\Dummy;
+
+/**
+ * Coupon factory class for testing.
+ */
+class CouponFactory extends \WP_UnitTest_Factory_For_Thing {
+	function __construct( $factory = null ) {
+		parent::__construct( $factory );
+
+		$this->default_generation_definitions = array(
+			'coupon_class'  => '\WC_Coupon',
+		);
+	}
+
+	public function create_object( $args ) {
+		if ( is_wp_error( $args ) ) codecept_debug( $args );
+		$coupon_class = $args['coupon_class' ];
+		unset( $args['coupon_class'] );
+
+		$coupon = new $coupon_class();
+
+		$amount = Dummy::instance()->number( 0, 75 );
+		$coupon->set_props(
+			array_merge(
+				array(
+					'code'          => $amount . 'off',
+					'amount'        => floatval( $amount ),
+					'date_expires'  => null,
+					'discount_type' => 'percent',
+					'description'   => 'Test coupon',
+				),
+				$args
+			)
+		);
+
+		// Set meta data.
+		if ( ! empty( $args['meta_data'] ) ) {
+			$coupon->set_meta_data( $args['meta_data'] );
+		}
+
+		return $coupon->save();
+	}
+
+	public function update_object( $object, $fields ) {
+		if ( ! $object instanceof \WC_Coupon && 0 !== absint( $object ) ) {
+			$object = $this->get_object_by_id( $object );
+		}
+
+		foreach( $fields as $field => $field_value ) {
+			if ( ! is_callable( array( $object, "set_{$field}" ) ) ) {
+				throw new \Exception(
+					sprintf( '"%1$s" is not a valid %2$s coupon field.', $field, $object->get_type() )
+				);
+			}
+
+			$object->{"set_{$field}"}( $field_value );
+		}
+
+		$object->save();
+	}
+
+	public function get_object_by_id( $id ) {
+		return new \WC_Coupon( $id );
+	}
+}

tests/_support/Factory/ProductFactory.php

@@ -2,7 +2,7 @@
 /**
  * Factory class for the WooCommerce's Product data objects.
  *
- * @since v0.6.0
+ * @since v0.6.1
  * @package Tests\WPGraphQL\WooCommerce\Factory
  */
 
@@ -17,7 +17,6 @@ class ProductFactory extends \WP_UnitTest_Factory_For_Thing {
 	function __construct( $factory = null ) {
 		parent::__construct( $factory );
 
-		$name = Dummy::instance()->product();
 		$this->default_generation_definitions = array(
 			'product_class' => '\WC_Product_Simple',
 		);

tests/_support/Factory/ProductVariationFactory.php

@@ -2,7 +2,7 @@
 /**
  * Factory class for the WooCommerce's Product variation data objects.
  *
- * @since v0.6.0
+ * @since v0.6.1
  * @package Tests\WPGraphQL\WooCommerce\Factory
  */
 

tests/_support/TestCase/WooGraphQLTestCase.php

@@ -28,7 +28,7 @@ class WooGraphQLTestCase extends \Tests\WPGraphQL\TestCase\WPGraphQLTestCase {
 	/**
 	 * Creates users and loads factories.
 	 */
-	function setUp(): void {
+	public function setUp(): void {
 		parent::setUp();
 
 		// Create users.
@@ -40,6 +40,7 @@ function setUp(): void {
 			'Product',
 			'ProductVariation',
 			'Cart',
+			'Coupon',
 		);
 
 		foreach ( $factories as $factory ) {
@@ -49,6 +50,13 @@ function setUp(): void {
 		}
 	}
 
+	public function tearDown(): void {
+		\WC()->cart->empty_cart( true );
+
+		// then
+		parent::tearDown();
+	}
+
 	/**
 	 * Logs in as a "shop manager"
 	 */