wp-graphql-woocommerce/tests/_support/TestCase/WooGraphQLTestCase.php

<?php
/**
 * WPGraphQL test case
 *
 * For testing WPGraphQL responses.
 * @since 0.6.0
 * @package Tests\WPGraphQL\TestCase
 */
namespace Tests\WPGraphQL\TestCase;

class WooGraphQLTestCase extends WPGraphQLTestCase {
	/**
	 * Holds the User ID of an user with the "shop_manager" role.
	 * For use through the tests for purpose of testing user access levels.
	 *
	 * @var integer
	 */
	protected $shop_manager;

	/**
	 * Holds the User ID of an user with the "customer/subscriber" role.
	 * For use through the tests for purpose of testing user access levels.
	 *
	 * @var integer
	 */
	protected $customer;

	/**
	 * Creates users and loads factories.
	 */
	function setUp(): void {
		parent::setUp();

		// Create users.
		$this->shop_manager = $this->factory->user->create( array( 'role' => 'shop_manager' ) );
		$this->customer     = $this->factory->user->create( array( 'role' => 'customer' ) );

		// Load factories.
		$factories = array(
			'Product',
		);

		foreach ( $factories as $factory ) {
			$factory_name  = strtolower( $factory );
			$factory_class = '\\Tests\\WPGraphQL\\Factory\\' . $factory . 'Factory';
			$this->factory->{$factory_name} = new $factory_class( $this->factory );
		}
	}

	/**
	 * Logs in as a "shop manager"
	 */
	protected function loginAsShopManager() {
		wp_set_current_user( $this->shop_manager );
	}

	/**
	 * Logs in as a "customer"
	 */
	protected function loginAsCustomer() {
		wp_set_current_user( $this->customer );
	}

	/**
	 * Logs out current user.
	 */
	protected function logout() {
		wp_set_current_user( 0 );
	}

}