woocommerce/plugins/woocommerce-blocks/tests/php/StoreApi/Utilities/ReserveStock.php

<?php
/**
 * Utility Tests.
 */

namespace Automattic\WooCommerce\Blocks\Tests\StoreApi\Utilities;

use PHPUnit\Framework\TestCase;
use \WC_Helper_Order as OrderHelper;
use \WC_Helper_Product as ProductHelper;
use Automattic\WooCommerce\Blocks\StoreApi\Utilities\ReserveStock;

/**
 * ReserveStock Utility Tests.
 */
class ReserveStockTests extends TestCase {

	/**
	 * Test that stock is reserved for draft orders.
	 */
	public function test_reserve_stock_for_order() {
		$class = new ReserveStock();

		$product = ProductHelper::create_simple_product();
		$product->set_manage_stock( true );
		// this also saves product
		wc_update_product_stock( $product, 10 );

		$order = OrderHelper::create_order( 1, $product ); // Note this adds 4 to the order.
		$order->set_status( 'checkout-draft' );
		$order->save();

		$class->reserve_stock_for_order( $order );
		$this->assertEquals( 4, $this->get_reserved_stock_by_product_id( $product->get_stock_managed_by_id() ) );

		// Repeat.
		$order = OrderHelper::create_order( 1, $product );
		$order->set_status( 'checkout-draft' );
		$order->save();

		$class->reserve_stock_for_order( $order );
		$this->assertEquals( 8, $this->get_reserved_stock_by_product_id( $product->get_stock_managed_by_id() ) );
	}

	/**
	 * Test that trying to reserve stock too much throws an exception.
	 *
	 * @expectedException Automattic\WooCommerce\Blocks\StoreApi\Utilities\ReserveStockException
	 */
	public function test_reserve_stock_for_order_throws_exception() {
		$class = new ReserveStock();

		$product = ProductHelper::create_simple_product();
		$product->set_manage_stock( true );
		// this also saves product
		wc_update_product_stock( $product, 10 );

		$order = OrderHelper::create_order( 1, $product ); // Note this adds 4 to the order.
		$order->set_status( 'checkout-draft' );
		$order->save();

		$order2 = OrderHelper::create_order( 1, $product );
		$order2->set_status( 'checkout-draft' );
		$order2->save();

		$order3 = OrderHelper::create_order( 1, $product );
		$order3->set_status( 'checkout-draft' );
		$order3->save();

		$class->reserve_stock_for_order( $order );
		$class->reserve_stock_for_order( $order2 );
		$class->reserve_stock_for_order( $order3 );
	}

	/**
	 * Helper to get the count of reserved stock.
	 *
	 * @param integer $product_id
	 * @return integer
	 */
	protected function get_reserved_stock_by_product_id( $product_id ) {
		global $wpdb;
		return $wpdb->get_var(
			$wpdb->prepare(
				"SELECT SUM( stock_table.`stock_quantity` ) FROM $wpdb->wc_reserved_stock stock_table WHERE stock_table.`product_id` = %d",
				$product_id
			)
		);
	}
}