120 lines
3.2 KiB
PHP
120 lines
3.2 KiB
PHP
<?php
|
|
/**
|
|
* Data REST API Test
|
|
*
|
|
* @package WooCommerce Admin\Tests\API
|
|
*/
|
|
|
|
/**
|
|
* WC Tests API Data
|
|
*/
|
|
class WC_Tests_API_Data extends WC_REST_Unit_Test_Case {
|
|
|
|
/**
|
|
* Endpoints.
|
|
*
|
|
* @var string
|
|
*/
|
|
protected $endpoint = '/wc/v4/data';
|
|
|
|
/**
|
|
* Setup test data. Called before every test.
|
|
*/
|
|
public function setUp() {
|
|
parent::setUp();
|
|
|
|
$this->user = $this->factory->user->create(
|
|
array(
|
|
'role' => 'administrator',
|
|
)
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Test that the list of data endpoints includes download-ips.
|
|
*/
|
|
public function test_get_items_contains_download_ips() {
|
|
wp_set_current_user( $this->user );
|
|
|
|
$request = new WP_REST_Request( 'GET', $this->endpoint );
|
|
$response = $this->server->dispatch( $request );
|
|
$data = $response->get_data();
|
|
|
|
$this->assertEquals( 200, $response->get_status() );
|
|
$this->assertEquals( 4, count( $data ) );
|
|
$this->assertEquals( 'download-ips', $data[3]['slug'] );
|
|
}
|
|
|
|
/**
|
|
* Test download-ips match searching.
|
|
*/
|
|
public function test_download_ips() {
|
|
wp_set_current_user( $this->user );
|
|
WC_Helper_Reports::reset_stats_dbs();
|
|
|
|
$prod_download = new WC_Product_Download();
|
|
$prod_download->set_file( plugin_dir_url( __FILE__ ) . '/assets/images/help.png' );
|
|
$prod_download->set_id( 1 );
|
|
|
|
$product = new WC_Product_Simple();
|
|
$product->set_name( 'Test Product' );
|
|
$product->set_downloadable( 'yes' );
|
|
$product->set_downloads( array( $prod_download ) );
|
|
$product->set_regular_price( 25 );
|
|
$product->save();
|
|
|
|
$order = WC_Helper_Order::create_order( 1, $product );
|
|
$order->set_status( 'completed' );
|
|
$order->set_total( 100 );
|
|
$order->save();
|
|
|
|
$download = new WC_Customer_Download();
|
|
$download->set_user_id( $this->user );
|
|
$download->set_order_id( $order->get_id() );
|
|
$download->set_product_id( $product->get_id() );
|
|
$download->set_download_id( $prod_download->get_id() );
|
|
$download->save();
|
|
|
|
$object = new WC_Customer_Download_Log();
|
|
$object->set_permission_id( $download->get_id() );
|
|
$object->set_user_id( $this->user );
|
|
$object->set_user_ip_address( '1.2.3.4' );
|
|
$id = $object->save();
|
|
|
|
$object = new WC_Customer_Download_Log();
|
|
$object->set_permission_id( $download->get_id() );
|
|
$object->set_user_id( $this->user );
|
|
$object->set_user_ip_address( '54.2.1.3' );
|
|
$id = $object->save();
|
|
|
|
// Save a second log for the same IP -- only one result for this IP should be returned.
|
|
$object = new WC_Customer_Download_Log();
|
|
$object->set_permission_id( $download->get_id() );
|
|
$object->set_user_id( $this->user );
|
|
$object->set_user_ip_address( '54.2.1.3' );
|
|
$id = $object->save();
|
|
|
|
$object = new WC_Customer_Download_Log();
|
|
$object->set_permission_id( $download->get_id() );
|
|
$object->set_user_id( $this->user );
|
|
$object->set_user_ip_address( '54.5.1.7' );
|
|
$id = $object->save();
|
|
|
|
$request = new WP_REST_Request( 'GET', $this->endpoint . '/download-ips' );
|
|
$request->set_query_params(
|
|
array(
|
|
'match' => '54',
|
|
)
|
|
);
|
|
|
|
$response = $this->server->dispatch( $request );
|
|
$addresses = $response->get_data();
|
|
|
|
$this->assertEquals( 200, $response->get_status() );
|
|
$this->assertEquals( 2, count( $addresses ) );
|
|
|
|
$this->assertEquals( '54.2.1.3', $addresses[0]['user_ip_address'] );
|
|
$this->assertEquals( '54.5.1.7', $addresses[1]['user_ip_address'] );
|
|
}
|
|
}
|