119 lines
3.4 KiB
PHP
119 lines
3.4 KiB
PHP
<?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;
|
|
}
|
|
}
|