<?php
/**
 * Reports customers tests.
 *
 * @package WooCommerce\Admin\Tests\Customers
 */

use \Automattic\WooCommerce\Admin\API\Reports\Customers\Stats\DataStore;

/**
 * Class WC_Tests_Reports_Customers
 */
class WC_Tests_Reports_Customer extends WC_Unit_Test_Case {

	/**
	 * Test order count caluclation for customer.
	 *
	 * @covers \Automattic\WooCommerce\Admin\API\Reports\Customers\DataStore::get_order_count
	 */
	public function test_customer_order_count() {
		WC_Helper_Reports::reset_stats_dbs();

		// Create a customer.
		$customer = WC_Helper_Customer::create_customer();

		// Create product.
		$product = new WC_Product_Simple();
		$product->set_name( 'Test Product' );
		$product->set_regular_price( 25 );
		$product->save();

		WC_Helper_Queue::run_all_pending();

		$customer_id = DataStore::get_customer_id_by_user_id( $customer->get_id() ); // This is the customer ID from lookup table.

		// Create 3 orders.
		foreach ( range( 1, 3 ) as $i ) {
			$order = WC_Helper_Order::create_order( $customer->get_id(), $product );
			$order->save();
		}

		WC_Helper_Queue::run_all_pending();

		// Customer should have 3 orders.
		$this->assertSame( 3, DataStore::get_order_count( $customer_id ) );

		// Failure from bad customer IDs.
		$this->assertSame( null, DataStore::get_order_count( 0 ) );
		$this->assertSame( null, DataStore::get_order_count( 'ABC' ) );
		$this->assertSame( null, DataStore::get_order_count( false ) );
		$this->assertSame( null, DataStore::get_order_count( null ) );
	}

	/**
	 * Test customer lookup tables are cleaned after deleting an order.
	 *
	 * A customer record should only be deleted if the customer has no other orders.
	 *
	 * @covers \Automattic\WooCommerce\Admin\API\Reports\Customers\DataStore::sync_on_order_delete
	 */
	public function test_order_deletion_removes_customer() {
		WC_Helper_Reports::reset_stats_dbs();

		// Create a customer.
		$customer = WC_Helper_Customer::create_customer();

		// Create products.
		$product1 = new WC_Product_Simple();
		$product1->set_name( 'Test Product 1' );
		$product1->set_regular_price( 1 );
		$product1->save();

		$product2 = new WC_Product_Simple();
		$product2->set_name( 'Test Product 2' );
		$product2->set_regular_price( 2 );
		$product2->save();

		WC_Helper_Queue::run_all_pending();

		// Create the first order.
		$order1 = WC_Helper_Order::create_order( $customer->get_id(), $product1 );
		$order1->save();

		// Create the second order.
		$order2 = WC_Helper_Order::create_order( $customer->get_id(), $product2 );
		$order2->save();

		WC_Helper_Queue::run_all_pending();

		$customer_id = DataStore::get_customer_id_by_user_id( $customer->get_id() ); // This is the customer ID from lookup table.

		// Customer should remain in lookup table after first order deleted.
		$order1->delete( true );
		$this->assertCount( 1, $this->get_customer_record( $customer_id ), 'customer remains' );

		// Customer should be removed in lookup table after both orders are deleted.
		$order2->delete( true );
		$this->assertCount( 0, $this->get_customer_record( $customer_id ), 'customer removed' );
	}

	/**
	 * Get a customer's record from the database.
	 *
	 * @param int $customer_id Analytics Customer ID (not WP User ID).
	 */
	private function get_customer_record( $customer_id ) {
		global $wpdb;

		$results = $wpdb->get_results(
			$wpdb->prepare(
				"SELECT * FROM {$wpdb->prefix}wc_customer_lookup WHERE customer_id = %d",
				$customer_id
			)
		);

		return $results;
	}
}