wp-graphql-woocommerce/tests/wpunit/ShippingMethodQueriesTest.php

<?php

use GraphQLRelay\Relay;

class ShippingMethodQueriesTest extends \Codeception\TestCase\WPTestCase {
	private $shop_manager;
	private $customer;
	private $method;
	private $helper;

	public function setUp() {
		parent::setUp();

		$this->shop_manager = $this->factory->user->create( array( 'role' => 'shop_manager' ) );
		$this->customer     = $this->factory->user->create( array( 'role' => 'customer' ) );
		$this->helper       = $this->getModule('\Helper\Wpunit')->shipping_method();
		$this->method       = 'flat_rate';
	}

	public function tearDown() {
		// your tear down methods here
		// then
		parent::tearDown();
	}

	// tests
	public function testShippingMethodQueryAndArgs() {
		$id = Relay::toGlobalId( 'shipping_method', $this->method );

		$query = '
			query( $id: ID!, $idType: ShippingMethodIdTypeEnum ) {
				shippingMethod( id: $id, idType: $idType ) {
					id
					databaseId
					title
					description
				}
			}
		';

		/**
		 * Assertion One
		 *
		 * test "ID" ID type.
		 */
		$variables = array(
			'id'     => $id,
			'idType' => 'ID',
		);
		$actual = graphql(
			array(
				'query'     => $query,
				'variables' => $variables,
			 )
		);
		$expected = array( 'data' => array( 'shippingMethod' => $this->helper->print_query( $this->method ) ) );

		// use --debug flag to view.
		codecept_debug( $actual );

		$this->assertEquals( $expected, $actual );

		/**
		 * Assertion Two
		 *
		 * test "DATABASE_ID" ID type.
		 */
		$variables = array(
			'id'     => $this->method,
			'idType' => 'DATABASE_ID',
		);
		$actual = graphql(
			array(
				'query'     => $query,
				'variables' => $variables,
			 )
		);
		$expected = array( 'data' => array( 'shippingMethod' => $this->helper->print_query( $this->method ) ) );

		// use --debug flag to view.
		codecept_debug( $actual );

		$this->assertEquals( $expected, $actual );

	}

	public function testShippingMethodsQuery() {
		$wc_shipping = WC_Shipping::instance();
		$methods = array_values(
			array_map(
				function( $method ) {
					return array( 'id' => Relay::toGlobalId( 'shipping_method', $method->id ) );
				},
				$wc_shipping->get_shipping_methods()
			)
		);

		$query = '
			query shippingMethodsQuery {
				shippingMethods {
					nodes {
						id
					}
				}
			}
		';

		/**
		 * Assertion One
		 *
		 * tests query
		 */
		$actual = do_graphql_request( $query, 'shippingMethodQuery' );
		$expected = array( 'data' => array( 'shippingMethods' => array( 'nodes' => $methods ) ) );

		// use --debug flag to view.
		codecept_debug( $actual );

		$this->assertEquals( $expected, $actual );
	}
}