wp-graphql-woocommerce/tests/_support/Factory/ShippingZoneFactory.php

<?php
/**
 * Factory class for the WooCommerce's shipping method data objects.
 *
 * @since v0.10.0
 * @package Tests\WPGraphQL\WooCommerce\Factory
 */

namespace Tests\WPGraphQL\WooCommerce\Factory;

use Tests\WPGraphQL\WooCommerce\Utils\Dummy;

/**
 * Shipping method factory class for testing.
 */
class ShippingZoneFactory extends \WP_UnitTest_Factory_For_Thing {
	public function __construct( $factory = null ) {
		parent::__construct( $factory );

		$this->default_generation_definitions = array(
			'zone_name' => '',
		);
		$this->dummy                          = Dummy::instance();
	}

	public function create_object( $args ) {
		if ( is_wp_error( $args ) ) {
			codecept_debug( $args );
		}

		$object = new \WC_Shipping_Zone();

		foreach ( $args as $key => $value ) {
			if ( is_callable( array( $object, "set_{$key}" ) ) ) {
				$object->{"set_{$key}"}( $value );
			}
		}

		return $object->save();
	}

	public function update_object( $object, $fields ) {
		if ( ! $object instanceof \WC_Customer && 0 !== absint( $object ) ) {
			$object = $this->get_object_by_id( $object );
		}
	}

	public function get_object_by_id( $id ) {
		if ( class_exists( '\WC_Shipping_Zones' ) ) {
			$zone = \WC_Shipping_Zones::get_zone( $id );
			return $zone;
		}

		return false;
	}

	public function getAllZones() {
		if ( class_exists( '\WC_Shipping_Zones' ) ) {
			$all_zones = \WC_Shipping_Zones::get_zones();
			return $all_zones;
		}

		return false;
	}

	public function createLegacyFlatRate( $args = array() ) {
		$flat_rate_settings = array_merge(
			array(
				'enabled'      => 'yes',
				'title'        => 'Flat rate',
				'availability' => 'all',
				'countries'    => '',
				'tax_status'   => 'taxable',
				'cost'         => '10',
			),
			$args
		);
		update_option( 'woocommerce_flat_rate_settings', $flat_rate_settings );
		update_option( 'woocommerce_flat_rate', array() );
		\WC_Cache_Helper::get_transient_version( 'shipping', true );
		\WC()->shipping()->load_shipping_methods();

		return 'legacy_flat_rate';
	}

	public function createLegacyFreeShipping( $args = array() ) {
		$free_shipping_settings = array_merge(
			array(
				'enabled'      => 'yes',
				'title'        => 'Free shipping',
				'availability' => 'all',
				'countries'    => '',
			),
			$args
		);
		update_option( 'woocommerce_free_shipping_settings', $free_shipping_settings );
		update_option( 'woocommerce_free_shipping', array() );
		\WC_Cache_Helper::get_transient_version( 'shipping', true );
		WC()->shipping()->load_shipping_methods();

		return 'legacy_free_shipping';
	}
}