<?php

/**
 * Order coupon tests.
 * @package WooCommerce\Tests\Orders
 */
class WC_Tests_Order_Coupons extends WC_Unit_Test_Case {

	/**
	 * Track ids.
	 *
	 * @var array
	 */
	protected $objects = array();

	/**
	 * Setup an order.
	 */
	public function setUp() {
		$this->delete_objects();

		$_SERVER['REMOTE_ADDR'] = '127.0.0.1';

		update_option( 'woocommerce_default_customer_address', 'base' );
		update_option( 'woocommerce_tax_based_on', 'base' );

		$product = WC_Helper_Product::create_simple_product();
		update_post_meta( $product->get_id(), '_regular_price', '1000' );
		update_post_meta( $product->get_id(), '_price', '1000' );
		$product = wc_get_product( $product->get_id() );

		$coupon = new WC_Coupon;
		$coupon->set_code( 'test-coupon-1' );
		$coupon->set_amount( 1.00 );
		$coupon->set_discount_type( 'fixed_cart' );
		$coupon->save();

		$coupon2 = new WC_Coupon;
		$coupon2->set_code( 'test-coupon-2' );
		$coupon2->set_amount( 20 );
		$coupon2->set_discount_type( 'percent' );
		$coupon2->save();

		$order = wc_create_order( array(
			'status'        => 'pending',
			'customer_id'   => 1,
			'customer_note' => '',
			'total'         => '',
		) );

		// Add order products
		$product_item  = new WC_Order_Item_Product();
		$coupon_item_1 = new WC_Order_Item_Coupon();
		$coupon_item_2 = new WC_Order_Item_Coupon();

		if ( get_option( 'woocommerce_prices_include_tax', 'no' ) === 'yes' && get_option( 'woocommerce_calc_taxes', 'no' ) === 'yes' ) {
			$product_item->set_props( array(
				'product'  => $product,
				'quantity' => 1,
				'subtotal' => 909.09, // Ex tax.
				'total'    => 726.36,
			) );
			$coupon_item_1->set_props( array(
				'code'         => 'test-coupon-1',
				'discount'     => 0.91,
				'discount_tax' => 0.09,
			) );
			$coupon_item_2->set_props( array(
				'code'         => 'this-is-a-virtal-coupon',
				'discount'     => 181.82,
				'discount_tax' => 18.18,
			) );
		} else {
			$product_item->set_props( array(
				'product'  => $product,
				'quantity' => 1,
				'subtotal' => 1000, // Ex tax.
				'total'    => 799,
			) );
			$coupon_item_1->set_props( array(
				'code'         => 'test-coupon-1',
				'discount'     => 1,
				'discount_tax' => get_option( 'woocommerce_calc_taxes', 'no' ) === 'yes' ? 0.1 : 0,
			) );
			$coupon_item_2->set_props( array(
				'code'         => 'this-is-a-virtal-coupon',
				'discount'     => 200,
				'discount_tax' => get_option( 'woocommerce_calc_taxes', 'no' ) === 'yes' ? 20 : 0,
			) );
		}

		$product_item->save();
		$coupon_item_1->save();
		$coupon_item_2->save();

		$order->add_item( $product_item );
		$order->add_item( $coupon_item_1 );
		$order->add_item( $coupon_item_2 );

		$this->objects['coupons'][]      = $coupon;
		$this->objects['coupons'][]      = $coupon2;
		$this->objects['products'][]     = $product;
		$this->objects['order']          = $order;
		$this->objects['tax_rate_ids'][] = WC_Tax::_insert_tax_rate( array(
			'tax_rate_country'  => '',
			'tax_rate_state'    => '',
			'tax_rate'          => '10.0000',
			'tax_rate_name'     => 'VAT',
			'tax_rate_priority' => '1',
			'tax_rate_compound' => '0',
			'tax_rate_shipping' => '1',
			'tax_rate_order'    => '1',
			'tax_rate_class'    => '',
		) );

		$order->calculate_totals( true );
		$order->save();
	}

	/**
	 * Clean up after test.
	 */
	public function tearDown() {
		update_option( 'woocommerce_calc_taxes', 'no' );
		update_option( 'woocommerce_prices_include_tax', 'no' );
		$this->delete_objects();
	}

	/**
	 * delete objects.
	 */
	protected function delete_objects() {
		if ( isset( $this->objects['coupons'] ) ) {
			foreach ( $this->objects['coupons'] as $object ) {
				$object->delete( true );
			}
		}

		if ( isset( $this->objects['products'] ) ) {
			foreach ( $this->objects['products'] as $object ) {
				$object->delete( true );
			}
		}

		if ( isset( $this->objects['order'] ) ) {
			$this->objects['order']->delete( true );
		}

		if ( isset( $this->objects['tax_rate_ids'] ) ) {
			foreach ( $this->objects['tax_rate_ids'] as $tax_rate_id ) {
				WC_Tax::_delete_tax_rate( $tax_rate_id );
			}
		}

		$this->objects = array();
		wp_cache_flush();
	}

	/**
	 * Test: test_remove_coupon_from_order
	 */
	function test_remove_coupon_from_order() {
		update_option( 'woocommerce_prices_include_tax', 'yes' );
		update_option( 'woocommerce_calc_taxes', 'yes' );
		$this->setUp();

		$order_id = $this->objects['order']->get_id();
		$order    = wc_get_order( $order_id );

		// Check it's expected.
		$this->assertEquals( 'shop_order', $order->get_type() );
		$this->assertEquals( '799', $order->get_total(), $order->get_total() );

		// Remove the virtual coupon. Total should be 999.
		$order->remove_coupon( 'this-is-a-virtal-coupon' );
		$this->assertEquals( '999', $order->get_total(), $order->get_total() );

		// Remove the other coupon. Total should be 1000.
		$order->remove_coupon( 'test-coupon-1' );
		$this->assertEquals( '1000', $order->get_total(), $order->get_total() );

		// Reset.
		$this->setUp();

		$order_id = $this->objects['order']->get_id();
		$order    = wc_get_order( $order_id );

		// Check it's expected.
		$this->assertEquals( 'shop_order', $order->get_type() );
		$this->assertEquals( '799', $order->get_total(), $order->get_total() );

		// Do the above tests in reverse.
		$order->remove_coupon( 'test-coupon-1' );
		$this->assertEquals( '800', $order->get_total(), $order->get_total() );
		$order->remove_coupon( 'this-is-a-virtal-coupon' );
		$this->assertEquals( '1000', $order->get_total(), $order->get_total() );
	}

	/**
	 * Test: test_add_coupon_to_order
	 */
	function test_add_coupon_to_order() {
		update_option( 'woocommerce_prices_include_tax', 'yes' );
		update_option( 'woocommerce_calc_taxes', 'yes' );
		$this->setUp();

		$order_id = $this->objects['order']->get_id();
		$order    = wc_get_order( $order_id );

		$order->apply_coupon( 'test-coupon-2' );
		$this->assertEquals( '639.2', $order->get_total(), $order->get_total() );
	}

	/**
	 * Test: test_remove_coupon_from_order_ex_tax
	 */
	function test_remove_coupon_from_order_ex_tax() {
		update_option( 'woocommerce_prices_include_tax', 'no' );
		update_option( 'woocommerce_calc_taxes', 'yes' );
		$this->setUp();

		$order_id = $this->objects['order']->get_id();
		$order    = wc_get_order( $order_id );

		// Check it's expected.
		$this->assertEquals( 'shop_order', $order->get_type() );
		$this->assertEquals( '878.90', $order->get_total(), $order->get_total() );

		// Remove the virtual coupon. Total should be 999.
		$order->remove_coupon( 'this-is-a-virtal-coupon' );
		$this->assertEquals( '1098.90', $order->get_total(), $order->get_total() );

		// Remove the other coupon. Total should be 1000.
		$order->remove_coupon( 'test-coupon-1' );
		$this->assertEquals( '1100', $order->get_total(), $order->get_total() );

		// Reset.
		$this->setUp();

		$order_id = $this->objects['order']->get_id();
		$order    = wc_get_order( $order_id );

		// Check it's expected.
		$this->assertEquals( 'shop_order', $order->get_type() );
		$this->assertEquals( '878.90', $order->get_total(), $order->get_total() );

		// Do the above tests in reverse.
		$order->remove_coupon( 'test-coupon-1' );
		$this->assertEquals( '880', $order->get_total(), $order->get_total() );
		$order->remove_coupon( 'this-is-a-virtal-coupon' );
		$this->assertEquals( '1100', $order->get_total(), $order->get_total() );
	}

	/**
	 * Test: test_add_coupon_to_order_ex_tax
	 */
	function test_add_coupon_to_order_ex_tax() {
		update_option( 'woocommerce_prices_include_tax', 'no' );
		update_option( 'woocommerce_calc_taxes', 'yes' );
		$this->setUp();

		$order_id = $this->objects['order']->get_id();
		$order    = wc_get_order( $order_id );

		$order->apply_coupon( 'test-coupon-2' );
		$this->assertEquals( '703.12', $order->get_total(), $order->get_total() );
	}

	/**
	 * Test: test_remove_coupon_from_order_no_tax
	 */
	function test_remove_coupon_from_order_no_tax() {
		update_option( 'woocommerce_prices_include_tax', 'no' );
		update_option( 'woocommerce_calc_taxes', 'no' );
		$this->setUp();

		$order_id = $this->objects['order']->get_id();
		$order    = wc_get_order( $order_id );

		// Check it's expected.
		$this->assertEquals( 'shop_order', $order->get_type() );
		$this->assertEquals( '799', $order->get_total(), $order->get_total() );

		// Remove the virtual coupon. Total should be 999.
		$order->remove_coupon( 'this-is-a-virtal-coupon' );
		$this->assertEquals( '999', $order->get_total(), $order->get_total() );

		// Remove the other coupon. Total should be 1000.
		$order->remove_coupon( 'test-coupon-1' );
		$this->assertEquals( '1000', $order->get_total(), $order->get_total() );

		// Reset.
		$this->setUp();

		$order_id = $this->objects['order']->get_id();
		$order    = wc_get_order( $order_id );

		// Check it's expected.
		$this->assertEquals( 'shop_order', $order->get_type() );
		$this->assertEquals( '799', $order->get_total(), $order->get_total() );

		// Do the above tests in reverse.
		$order->remove_coupon( 'test-coupon-1' );
		$this->assertEquals( '800', $order->get_total(), $order->get_total() );
		$order->remove_coupon( 'this-is-a-virtal-coupon' );
		$this->assertEquals( '1000', $order->get_total(), $order->get_total() );
	}

	/**
	 * Test: test_add_coupon_to_order_no_tax
	 */
	function test_add_coupon_to_order_no_tax() {
		update_option( 'woocommerce_prices_include_tax', 'no' );
		update_option( 'woocommerce_calc_taxes', 'no' );
		$this->setUp();

		$order_id = $this->objects['order']->get_id();
		$order    = wc_get_order( $order_id );

		$order->apply_coupon( 'test-coupon-2' );
		$this->assertEquals( '639.2', $order->get_total(), $order->get_total() );
	}

	/**
	 * Test: test_remove_coupon_from_order_no_tax
	 */
	function test_remove_coupon_from_order_no_tax_inc_prices_on() {
		update_option( 'woocommerce_prices_include_tax', 'yes' );
		update_option( 'woocommerce_calc_taxes', 'no' );
		$this->setUp();

		$order_id = $this->objects['order']->get_id();
		$order    = wc_get_order( $order_id );

		// Check it's expected.
		$this->assertEquals( 'shop_order', $order->get_type() );
		$this->assertEquals( '799', $order->get_total(), $order->get_total() );

		// Remove the virtual coupon. Total should be 999.
		$order->remove_coupon( 'this-is-a-virtal-coupon' );
		$this->assertEquals( '999', $order->get_total(), $order->get_total() );

		// Remove the other coupon. Total should be 1000.
		$order->remove_coupon( 'test-coupon-1' );
		$this->assertEquals( '1000', $order->get_total(), $order->get_total() );

		// Reset.
		$this->setUp();

		$order_id = $this->objects['order']->get_id();
		$order    = wc_get_order( $order_id );

		// Check it's expected.
		$this->assertEquals( 'shop_order', $order->get_type() );
		$this->assertEquals( '799', $order->get_total(), $order->get_total() );

		// Do the above tests in reverse.
		$order->remove_coupon( 'test-coupon-1' );
		$this->assertEquals( '800', $order->get_total(), $order->get_total() );
		$order->remove_coupon( 'this-is-a-virtal-coupon' );
		$this->assertEquals( '1000', $order->get_total(), $order->get_total() );
	}

	/**
	 * Test: test_add_coupon_to_order_no_tax
	 */
	function test_add_coupon_to_order_no_tax_inc_prices_on() {
		update_option( 'woocommerce_prices_include_tax', 'yes' );
		update_option( 'woocommerce_calc_taxes', 'no' );
		$this->setUp();

		$order_id = $this->objects['order']->get_id();
		$order    = wc_get_order( $order_id );

		$order->apply_coupon( 'test-coupon-2' );
		$this->assertEquals( '639.2', $order->get_total(), $order->get_total() );
	}
}