Steve-Dee-Designs
/
woocommerce
mirror of https://github.com/woocommerce/woocommerce.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add test for #17330

This commit is contained in:
Mike Jolley 2017-10-24 17:53:51 +01:00
parent 6223ffb684
commit 8ca76e84cd
2 changed files with 60 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
includes/class-wc-discounts.php
View File

@ -45,12 +45,10 @@ class WC_Discounts {
* @param array $object Cart or order object.
*/
public function __construct( $object = array() ) {
$this->object = $object;
if ( is_a( $this->object, 'WC_Cart' ) ) {
$this->set_items_from_cart( $this->object );
} elseif ( is_a( $this->object, 'WC_Order' ) ) {
$this->set_items_from_order( $this->object );
if ( is_a( $object, 'WC_Cart' ) ) {
$this->set_items_from_cart( $object );
} elseif ( is_a( $object, 'WC_Order' ) ) {
$this->set_items_from_order( $object );
}
}
@ -67,6 +65,8 @@ class WC_Discounts {
return;
}
$this->object = $cart;
foreach ( $cart->get_cart() as $key => $cart_item ) {
$item = new stdClass();
$item->key = $key;
@ -87,12 +87,14 @@ class WC_Discounts {
* @param array $order Cart object.
*/
public function set_items_from_order( $order ) {
$this->items = $this->discounts = array();
$this->items = $this->discounts = array();
if ( ! is_a( $order, 'WC_Order' ) ) {
return;
}
$this->object = $order;
foreach ( $order->get_items() as $order_item ) {
$item = new stdClass();
$item->key = $order_item->get_id();

51
tests/unit-tests/discounts/discounts.php
View File

@ -1343,4 +1343,55 @@ class WC_Tests_Discounts extends WC_Unit_Test_Case {
$all_discounts = $discounts->get_discounts();
$this->assertEquals( 0, count( $all_discounts['freeshipping'] ), 'Free shipping coupon should not have any discounts.' );
}
public function filter_woocommerce_coupon_get_discount_amount( $discount ) {
return $discount / 2;
}
public function test_coupon_discount_amount_filter() {
$discounts = new WC_Discounts();
add_filter( 'woocommerce_coupon_get_discount_amount', array( $this, 'filter_woocommerce_coupon_get_discount_amount' ) );
$product = WC_Helper_Product::create_simple_product();
$product->set_regular_price( 100 );
$product->set_tax_status( 'taxable' );
$product->save();
WC()->cart->add_to_cart( $product->get_id(), 1 );
$product2 = WC_Helper_Product::create_simple_product();
$product2->set_regular_price( 100 );
$product2->set_tax_status( 'taxable' );
$product2->save();
WC()->cart->add_to_cart( $product2->get_id(), 1 );
$coupon = WC_Helper_Coupon::create_coupon( 'test' );
$coupon->set_props( array(
'code' => 'test',
'discount_type' => 'percent',
'amount' => '20',
) );
$discounts->set_items_from_cart( WC()->cart );
$discounts->apply_coupon( $coupon );
$all_discounts = $discounts->get_discounts();
$discount_total = 0;
foreach ( $all_discounts as $code_name => $discounts_by_coupon ) {
$discount_total += array_sum( $discounts_by_coupon );
}
$this->assertEquals( 20, $discount_total );
remove_filter( 'woocommerce_coupon_get_discount_amount', array( $this, 'filter_woocommerce_coupon_get_discount_amount' ) );
WC()->cart->empty_cart();
WC()->cart->remove_coupons();
$product->delete( true );
$product2->delete( true );
$coupon->delete( true );
// Temporarily necessary until https://github.com/woocommerce/woocommerce/pull/16767 is implemented.
wp_cache_delete( WC_Cache_Helper::get_cache_prefix( 'coupons' ) . 'coupon_id_from_code_test', 'coupons' );
}
}