woocommerce/tests/unit-tests/order/class-wc-tests-orders.php

<?php
/**
 * Class WC_Tests_Order file.
 *
 * @package WooCommerce|Tests|Order
 */

/**
 * Class WC_Tests_Order.
 */
class WC_Tests_Order extends WC_Unit_Test_Case {

	/**
	 * Test for total when round at subtotal is enabled.
	 *
	 * @link https://github.com/woocommerce/woocommerce/issues/24695
	 */
	public function test_order_calculate_total_rounding_24695() {
		update_option( 'woocommerce_prices_include_tax', 'yes' );
		update_option( 'woocommerce_calc_taxes', 'yes' );
		update_option( 'woocommerce_tax_round_at_subtotal', 'yes' );

		$tax_rate = array(
			'tax_rate_country'  => '',
			'tax_rate_state'    => '',
			'tax_rate'          => '7.0000',
			'tax_rate_name'     => 'CGST',
			'tax_rate_priority' => '1',
			'tax_rate_compound' => '0',
			'tax_rate_shipping' => '0',
			'tax_rate_order'    => '1',
			'tax_rate_class'    => 'tax_1',
		);
		WC_Tax::_insert_tax_rate( $tax_rate );

		$product1 = WC_Helper_Product::create_simple_product();
		$product1->set_regular_price( 2 );
		$product1->save();

		$product2 = WC_Helper_Product::create_simple_product();
		$product2->set_regular_price( 2.5 );
		$product2->save();

		$order = new WC_Order();
		$order->add_product( $product1, 1 );
		$order->add_product( $product2, 4 );
		$order->save();

		$order->calculate_totals( true );

		$this->assertEquals( 12, $order->get_total() );
		$this->assertEquals( 0.79, $order->get_total_tax() );
	}

	/**
	 * Test shipping is added and rounded correctly when addedd to total.
	 *
	 * @throws WC_Data_Exception When lines cannot be added to order.
	 */
	public function test_order_rounding_with_shipping_25748() {
		update_option( 'woocommerce_prices_include_tax', 'no' );
		update_option( 'woocommerce_calc_taxes', 'yes' );
		update_option( 'woocommerce_tax_round_at_subtotal', 'yes' );

		$tax_rate = array(
			'tax_rate_country'  => '',
			'tax_rate_state'    => '',
			'tax_rate'          => '21.0000',
			'tax_rate_name'     => 'CGST',
			'tax_rate_priority' => '1',
			'tax_rate_compound' => '0',
			'tax_rate_shipping' => '1',
			'tax_rate_order'    => '1',
			'tax_rate_class'    => 'tax_1',
		);
		WC_Tax::_insert_tax_rate( $tax_rate );

		$product = WC_Helper_Product::create_simple_product();
		$product->set_regular_price( 23.85 );
		$product->save();

		$shipping_rate   = new WC_Shipping_Rate( 'flat_rate_shipping', 'Flat rate shipping', '9.5', array(), 'flat_rate' );
		$shipping_item   = new WC_Order_Item_Shipping();
		$shipping_item->set_props(
			array(
				'method_title' => $shipping_rate->label,
				'method_id'    => $shipping_rate->id,
				'total'        => wc_format_decimal( $shipping_rate->cost ),
				'taxes'        => $shipping_rate->taxes,
			)
		);

		foreach ( $shipping_rate->get_meta_data() as $key => $value ) {
			$shipping_item->add_meta_data( $key, $value, true );
		}

		$order = new WC_Order();
		$order->add_product( $product, 1 );
		$order->add_item( $shipping_item );

		$order->calculate_totals( true );

		$this->assertEquals( 7, $order->get_total_tax() );
		$this->assertEquals( 40.35, $order->get_total() );
	}

	/**
	 * Testing rounding when lines are copied over to order.
	 *
	 * @throws Exception When lines cannot be added to order.
	 */
	public function test_order_rounding_addition_25641() {
		update_option( 'woocommerce_prices_include_tax', 'no' );
		update_option( 'woocommerce_calc_taxes', 'yes' );
		update_option( 'woocommerce_tax_round_at_subtotal', 'yes' );

		$tax_rate1 = array(
			'tax_rate_country'  => '',
			'tax_rate_state'    => '',
			'tax_rate'          => '7.0000',
			'tax_rate_name'     => 'CGST',
			'tax_rate_priority' => '1',
			'tax_rate_compound' => '0',
			'tax_rate_shipping' => '1',
			'tax_rate_order'    => '1',
		);
		WC_Tax::_insert_tax_rate( $tax_rate1 );

		$tax_rate2 = array(
			'tax_rate_country'  => '',
			'tax_rate_state'    => '',
			'tax_rate'          => '2.2500',
			'tax_rate_name'     => 'SGST',
			'tax_rate_priority' => '2',
			'tax_rate_compound' => '0',
			'tax_rate_shipping' => '1',
			'tax_rate_order'    => '1',
		);
		WC_Tax::_insert_tax_rate( $tax_rate2 );

		$product = WC_Helper_Product::create_simple_product();
		$product->set_regular_price( 22.99 );
		$product->save();

		$flat_rate_settings = array(
			'enabled'      => 'yes',
			'title'        => 'Flat rate',
			'availability' => 'all',
			'countries'    => '',
			'tax_status'   => 'taxable',
			'cost'         => '10',
		);
		update_option( 'woocommerce_flat_rate_settings', $flat_rate_settings );

		WC()->cart->empty_cart();
		WC()->customer->set_is_vat_exempt( false );

		WC()->cart->add_to_cart( $product->get_id(), 1 );
		WC()->session->set( 'chosen_shipping_method', array( 'flat_rate' ) );
		WC()->cart->calculate_totals();

		$checkout = WC_Checkout::instance();
		$order = new WC_Order();
		$checkout->set_data_from_cart( $order );
		$this->assertEquals( 3.05, $order->get_total_tax() );
		$this->assertEquals( 36.04, $order->get_total() );
	}

}