2017-12-14 10:45:00 +00:00
|
|
|
<?php
|
2018-01-22 12:59:09 +00:00
|
|
|
/**
|
|
|
|
* WC_Customer_Download tests file.
|
|
|
|
*
|
|
|
|
* @package WooCommerce\Tests\Customer
|
|
|
|
*/
|
2017-12-14 10:45:00 +00:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Class WC_Customer_Download.
|
2018-01-22 12:59:09 +00:00
|
|
|
*
|
2017-12-14 10:45:00 +00:00
|
|
|
* @since 3.3.0
|
|
|
|
* @package WooCommerce\Tests\Customer
|
|
|
|
*/
|
|
|
|
class WC_Tests_Customer_Download extends WC_Unit_Test_Case {
|
|
|
|
|
2018-01-22 12:59:09 +00:00
|
|
|
/**
|
|
|
|
* Test WC_Customer_Download_Data_Store::delete()
|
|
|
|
*/
|
2017-12-14 10:45:00 +00:00
|
|
|
public function test_delete() {
|
|
|
|
$customer_id = wc_create_new_customer( 'test@example.com', 'testuser', 'testpassword' );
|
2018-01-22 12:59:09 +00:00
|
|
|
$download = new WC_Customer_Download();
|
2017-12-14 10:45:00 +00:00
|
|
|
$download->set_user_id( $customer_id );
|
|
|
|
$download->set_order_id( 1 );
|
|
|
|
$download->save();
|
|
|
|
$data_store = WC_Data_Store::load( 'customer-download' );
|
|
|
|
$data_store->delete( $download );
|
|
|
|
$this->assertEquals( 0, $download->get_id() );
|
|
|
|
}
|
|
|
|
|
2018-01-22 12:59:09 +00:00
|
|
|
/**
|
|
|
|
* Test WC_Customer_Download_Data_Store::delete_by_id()
|
|
|
|
*/
|
2017-12-14 10:45:00 +00:00
|
|
|
public function test_delete_by_id() {
|
|
|
|
$customer_id = wc_create_new_customer( 'test@example.com', 'testuser', 'testpassword' );
|
2018-01-22 12:59:09 +00:00
|
|
|
$download = new WC_Customer_Download();
|
2017-12-14 10:45:00 +00:00
|
|
|
$download->set_user_id( $customer_id );
|
|
|
|
$download->set_order_id( 1 );
|
|
|
|
$download->save();
|
|
|
|
$data_store = WC_Data_Store::load( 'customer-download' );
|
|
|
|
$data_store->delete_by_id( $download->get_id() );
|
|
|
|
$this->assertEquals( 0, $data_store->get_id() );
|
|
|
|
}
|
|
|
|
|
2018-01-22 12:59:09 +00:00
|
|
|
/**
|
|
|
|
* Test WC_Customer_Download_Data_Store::delete_by_download_id()
|
|
|
|
*/
|
2017-12-14 10:45:00 +00:00
|
|
|
public function test_delete_by_download_id() {
|
|
|
|
$customer_id = wc_create_new_customer( 'test@example.com', 'testuser', 'testpassword' );
|
2018-01-22 12:59:09 +00:00
|
|
|
$download = new WC_Customer_Download();
|
2017-12-14 10:45:00 +00:00
|
|
|
$download->set_user_id( $customer_id );
|
|
|
|
$download->set_order_id( 1 );
|
|
|
|
$download->save();
|
|
|
|
$download_id = $download->get_download_id();
|
2018-01-22 12:59:09 +00:00
|
|
|
$data_store = WC_Data_Store::load( 'customer-download' );
|
|
|
|
$downloads = $data_store->get_downloads_for_customer( $customer_id );
|
2017-12-14 10:45:00 +00:00
|
|
|
$this->assertInstanceOf( 'StdClass', $downloads[0] );
|
|
|
|
$data_store->delete_by_download_id( $download_id );
|
|
|
|
$downloads = $data_store->get_downloads_for_customer( $customer_id );
|
|
|
|
$this->assertEquals( array(), $downloads );
|
|
|
|
}
|
|
|
|
|
2018-01-22 12:59:09 +00:00
|
|
|
/**
|
|
|
|
* Test WC_Customer_Download_Data_Store::get_downloads()
|
|
|
|
*/
|
2017-12-14 10:45:00 +00:00
|
|
|
public function test_get_downloads() {
|
|
|
|
$customer_id = wc_create_new_customer( 'test@example.com', 'testuser', 'testpassword' );
|
2018-01-22 12:59:09 +00:00
|
|
|
$download_1 = new WC_Customer_Download();
|
2017-12-14 10:45:00 +00:00
|
|
|
$download_1->set_user_id( $customer_id );
|
|
|
|
$download_1->set_user_email( 'test@example.com' );
|
|
|
|
$download_1->set_order_id( 1 );
|
|
|
|
$download_1->save();
|
|
|
|
|
2018-01-22 12:59:09 +00:00
|
|
|
$download_2 = new WC_Customer_Download();
|
2017-12-14 10:45:00 +00:00
|
|
|
$download_2->set_user_id( $customer_id );
|
|
|
|
$download_2->set_user_email( 'test@example.com' );
|
|
|
|
$download_2->set_order_id( 1 );
|
|
|
|
$download_2->save();
|
|
|
|
|
|
|
|
$data_store = WC_Data_Store::load( 'customer-download' );
|
2018-01-22 12:59:09 +00:00
|
|
|
$downloads = $data_store->get_downloads( array( 'user_email' => 'test@example.com' ) );
|
2017-12-14 10:45:00 +00:00
|
|
|
$this->assertEquals( 2, count( $downloads ) );
|
2018-01-22 13:01:00 +00:00
|
|
|
$this->assertTrue( $downloads[0] instanceof WC_Customer_Download );
|
|
|
|
$this->assertTrue( $downloads[1] instanceof WC_Customer_Download );
|
|
|
|
|
2017-12-14 10:45:00 +00:00
|
|
|
$downloads = $data_store->get_downloads( array( 'user_email' => 'test2@example.com' ) );
|
|
|
|
$this->assertEquals( array(), $downloads );
|
|
|
|
|
2018-01-22 13:01:00 +00:00
|
|
|
$expected_result = array( $download_1->get_id(), $download_2->get_id() );
|
|
|
|
$downloads = $data_store->get_downloads(
|
|
|
|
array(
|
|
|
|
'user_email' => 'test@example.com',
|
|
|
|
'return' => 'ids',
|
|
|
|
)
|
|
|
|
);
|
|
|
|
$this->assertEquals( $expected_result, $downloads );
|
2017-12-14 10:45:00 +00:00
|
|
|
}
|
|
|
|
}
|