woocommerce/tests/unit-tests/totals/totals.php

232 lines
7.4 KiB
PHP
Raw Normal View History

2017-07-23 11:05:11 +00:00
<?php
/**
* Tests for the totals class.
*
2017-07-23 11:05:11 +00:00
* @package WooCommerce\Tests\Discounts
*/
/**
* WC_Tests_Totals
*/
2017-07-23 11:05:11 +00:00
class WC_Tests_Totals extends WC_Unit_Test_Case {
/**
* Totals class for getter tests.
*
* @var object
*/
2017-07-23 11:05:11 +00:00
protected $totals;
/**
* ID tracking for cleanup.
*
* @var array
*/
protected $ids = array();
2017-07-23 11:05:11 +00:00
/**
* Setup the cart for totals calculation.
*/
public function setUp() {
parent::setUp();
2017-07-25 21:44:54 +00:00
$this->ids = array();
2018-04-20 17:53:06 +00:00
$tax_rate = array(
2017-07-23 11:05:11 +00:00
'tax_rate_country' => '',
'tax_rate_state' => '',
'tax_rate' => '20.0000',
'tax_rate_name' => 'VAT',
'tax_rate_priority' => '1',
'tax_rate_compound' => '0',
'tax_rate_shipping' => '1',
'tax_rate_order' => '1',
'tax_rate_class' => '',
);
$tax_rate_id = WC_Tax::_insert_tax_rate( $tax_rate );
2017-07-28 16:43:05 +00:00
2017-07-23 11:05:11 +00:00
update_option( 'woocommerce_calc_taxes', 'yes' );
2017-07-28 16:43:05 +00:00
update_option( 'woocommerce_default_customer_address', 'base' );
update_option( 'woocommerce_tax_based_on', 'base' );
2017-07-23 11:05:11 +00:00
$product = WC_Helper_Product::create_simple_product();
$product2 = WC_Helper_Product::create_simple_product();
// Variations with parent tax class.
2018-04-20 17:53:06 +00:00
$product3 = WC_Helper_Product::create_variation_product();
$variations = $product3->get_available_variations();
// Update product3 so that each variation has the parent tax class.
foreach ( $variations as $variation ) {
$child = wc_get_product( $variation['variation_id'] );
$child->set_tax_class( 'parent' );
$child->save();
}
2017-07-23 11:05:11 +00:00
2017-07-25 13:05:49 +00:00
WC_Helper_Shipping::create_simple_flat_rate();
WC()->session->set( 'chosen_shipping_methods', array( 'flat_rate' ) );
2018-04-20 17:53:06 +00:00
$coupon = new WC_Coupon();
2017-07-23 11:05:11 +00:00
$coupon->set_code( 'test-coupon-10' );
$coupon->set_amount( 10 );
$coupon->set_discount_type( 'percent' );
$coupon->save();
$this->ids['tax_rate_ids'][] = $tax_rate_id;
$this->ids['products'][] = $product;
$this->ids['products'][] = $product2;
$this->ids['products'][] = $product3;
$this->ids['coupons'][] = $coupon;
2017-07-23 11:05:11 +00:00
WC()->cart->add_to_cart( $product->get_id(), 1 );
WC()->cart->add_to_cart( $product2->get_id(), 2 );
$variations = $product3->get_available_variations();
$variation = array_shift( $variations );
WC()->cart->add_to_cart( $product3->get_id(), 1, $variation['variation_id'], array( 'Size' => ucfirst( $variation['attributes']['attribute_pa_size'] ) ) );
2017-07-24 16:21:08 +00:00
WC()->cart->add_discount( $coupon->get_code() );
2017-07-23 11:05:11 +00:00
2017-07-25 13:05:49 +00:00
add_action( 'woocommerce_cart_calculate_fees', array( $this, 'add_cart_fees_callback' ) );
2017-07-23 11:05:11 +00:00
}
2017-07-25 13:05:49 +00:00
/**
* Add fees when the fees API is called.
*/
public function add_cart_fees_callback() {
WC()->cart->add_fee( 'test fee', 10, true );
WC()->cart->add_fee( 'test fee 2', 20, true );
WC()->cart->add_fee( 'test fee non-taxable', 10, false );
2017-07-25 13:05:49 +00:00
}
2017-07-23 11:05:11 +00:00
/**
* Clean up after test.
*/
public function tearDown() {
WC()->cart->empty_cart();
2017-07-25 13:05:49 +00:00
WC()->session->set( 'chosen_shipping_methods', array() );
WC_Helper_Shipping::delete_simple_flat_rate();
remove_action( 'woocommerce_cart_calculate_fees', array( $this, 'add_cart_fees_callback' ) );
2017-07-23 11:05:11 +00:00
2018-09-06 15:41:24 +00:00
WC()->customer->set_is_vat_exempt( false );
parent::tearDown();
2017-07-23 11:05:11 +00:00
}
/**
* Test get and set items.
*/
public function test_get_totals() {
2018-09-06 15:41:24 +00:00
WC()->customer->set_is_vat_exempt( false );
$this->totals = new WC_Cart_Totals( WC()->cart );
2018-04-20 17:53:06 +00:00
$this->assertEquals(
array(
'fees_total' => 40.00,
'fees_total_tax' => 6.00,
'items_subtotal' => 40.00,
'items_subtotal_tax' => 8.00,
'items_total' => 36.00,
'items_total_tax' => 7.20,
'total' => 101.20,
'shipping_total' => 10,
'shipping_tax_total' => 2,
'discounts_total' => 4.00,
'discounts_tax_total' => 0.80,
),
$this->totals->get_totals()
);
2017-07-23 11:05:11 +00:00
}
2017-07-25 21:44:54 +00:00
2018-09-06 15:41:24 +00:00
/**
* Test get totals for tax exempt customer.
*/
public function test_get_totals_tax_exempt_customer() {
WC()->customer->set_is_vat_exempt( true );
$this->totals = new WC_Cart_Totals( WC()->cart );
$this->assertEquals(
array(
'fees_total' => 40.00,
'fees_total_tax' => 0.00,
'items_subtotal' => 40.00,
'items_subtotal_tax' => 0.00,
'items_total' => 36.00,
'items_total_tax' => 0.00,
'total' => 86.00,
'shipping_total' => 10,
'shipping_tax_total' => 0.00,
'discounts_total' => 4.00,
'discounts_tax_total' => 0.00,
),
$this->totals->get_totals()
);
}
2017-07-25 21:44:54 +00:00
/**
* Test that cart totals get updated.
*/
public function test_cart_totals() {
2018-09-06 15:41:24 +00:00
WC()->customer->set_is_vat_exempt( false );
$this->totals = new WC_Cart_Totals( WC()->cart );
2017-07-25 21:44:54 +00:00
$cart = WC()->cart;
2017-08-18 14:05:01 +00:00
$this->assertEquals( 40.00, $cart->get_fee_total() );
$this->assertEquals( 36.00, $cart->get_cart_contents_total() );
$this->assertEquals( 101.20, $cart->get_total( 'edit' ) );
$this->assertEquals( 40.00, $cart->get_subtotal() );
$this->assertEquals( 8.00, $cart->get_subtotal_tax() );
$this->assertEquals( 7.20, $cart->get_cart_contents_tax() );
$this->assertEquals( 7.20, array_sum( $cart->get_cart_contents_taxes() ) );
2017-08-18 14:05:01 +00:00
$this->assertEquals( 6.00, $cart->get_fee_tax() );
$this->assertEquals( 6.00, array_sum( $cart->get_fee_taxes() ) );
$this->assertEquals( 4, $cart->get_discount_total() );
$this->assertEquals( 0.80, $cart->get_discount_tax() );
2017-08-18 14:05:01 +00:00
$this->assertEquals( 10, $cart->get_shipping_total() );
$this->assertEquals( 2, $cart->get_shipping_tax() );
$this->assertEquals( 36.00, $cart->cart_contents_total );
$this->assertEquals( 101.20, $cart->total );
$this->assertEquals( 48.00, $cart->subtotal );
$this->assertEquals( 40.00, $cart->subtotal_ex_tax );
$this->assertEquals( 13.20, $cart->tax_total );
$this->assertEquals( 4, $cart->discount_cart );
$this->assertEquals( 0.80, $cart->discount_cart_tax );
2017-07-25 21:44:54 +00:00
$this->assertEquals( 40.00, $cart->fee_total );
$this->assertEquals( 10, $cart->shipping_total );
$this->assertEquals( 2, $cart->shipping_tax_total );
}
2018-09-06 15:41:24 +00:00
/**
* Test that cart totals get updated for tax exempt customer.
*/
public function test_cart_totals_tax_exempt_customer() {
WC()->customer->set_is_vat_exempt( true );
$this->totals = new WC_Cart_Totals( WC()->cart );
$cart = WC()->cart;
$this->assertEquals( 40.00, $cart->get_fee_total() );
$this->assertEquals( 36.00, $cart->get_cart_contents_total() );
$this->assertEquals( 86.00, $cart->get_total( 'edit' ) );
$this->assertEquals( 40.00, $cart->get_subtotal() );
$this->assertEquals( 0.00, $cart->get_subtotal_tax() );
$this->assertEquals( 0.00, $cart->get_cart_contents_tax() );
$this->assertEquals( 0.00, array_sum( $cart->get_cart_contents_taxes() ) );
$this->assertEquals( 0.00, $cart->get_fee_tax() );
$this->assertEquals( 0.00, array_sum( $cart->get_fee_taxes() ) );
$this->assertEquals( 4, $cart->get_discount_total() );
$this->assertEquals( 0.00, $cart->get_discount_tax() );
$this->assertEquals( 10, $cart->get_shipping_total() );
$this->assertEquals( 0.00, $cart->get_shipping_tax() );
$this->assertEquals( 36.00, $cart->cart_contents_total );
$this->assertEquals( 86.00, $cart->total );
$this->assertEquals( 40.00, $cart->subtotal );
$this->assertEquals( 40.00, $cart->subtotal_ex_tax );
$this->assertEquals( 0.00, $cart->tax_total );
$this->assertEquals( 4, $cart->discount_cart );
$this->assertEquals( 0.00, $cart->discount_cart_tax );
$this->assertEquals( 40.00, $cart->fee_total );
$this->assertEquals( 10, $cart->shipping_total );
$this->assertEquals( 0.00, $cart->shipping_tax_total );
}
2017-07-23 11:05:11 +00:00
}