woocommerce/tests/unit-tests/api/product-variations.php

<?php
/**
 * Tests for Variations API.
 *
 * @package WooCommerce\Tests\API
 * @since 3.0.0
 */

class Product_Variations_API extends WC_REST_Unit_Test_Case {

	/**
	 * Setup our test server, endpoints, and user info.
	 */
	public function setUp() {
		parent::setUp();
		$this->endpoint = new WC_REST_Product_Variations_Controller();
		$this->user = $this->factory->user->create( array(
			'role' => 'administrator',
		) );
	}

	/**
	 * Test route registration.
	 *
	 * @since 3.0.0
	 */
	public function test_register_routes() {
		$routes = $this->server->get_routes();
		$this->assertArrayHasKey( '/wc/v2/products/(?P<product_id>[\d]+)/variations', $routes );
		$this->assertArrayHasKey( '/wc/v2/products/(?P<product_id>[\d]+)/variations/(?P<id>[\d]+)', $routes );
		$this->assertArrayHasKey( '/wc/v2/products/(?P<product_id>[\d]+)/variations/batch', $routes );
	}

	/**
	 * Test getting variations.
	 *
	 * @since 3.0.0
	 */
	public function test_get_variations() {
		wp_set_current_user( $this->user );
		$product    = WC_Helper_Product::create_variation_product();
		$response   = $this->server->dispatch( new WP_REST_Request( 'GET', '/wc/v2/products/' . $product->get_id() . '/variations' ) );
		$variations = $response->get_data();
		$this->assertEquals( 200, $response->get_status() );
		$this->assertEquals( 2, count( $variations ) );
		$this->assertEquals( 'DUMMY SKU VARIABLE LARGE', $variations[0]['sku'] );
		$this->assertEquals( 'size', $variations[0]['attributes'][0]['name'] );
		$product->delete( true );
	}

	/**
	 * Test getting variations without permission.
	 *
	 * @since 3.0.0
	 */
	public function test_get_variations_without_permission() {
		wp_set_current_user( 0 );
		$product    = WC_Helper_Product::create_variation_product();
		$response   = $this->server->dispatch( new WP_REST_Request( 'GET', '/wc/v2/products/' . $product->get_id() . '/variations' ) );
		$this->assertEquals( 401, $response->get_status() );
		$product->delete( true );
	}

	/**
	 * Test getting a single variation.
	 *
	 * @since 3.0.0
	 */
	public function test_get_variation() {
		wp_set_current_user( $this->user );
		$product      = WC_Helper_Product::create_variation_product();
		$children     = $product->get_children();
		$variation_id = $children[0];

		$response   = $this->server->dispatch( new WP_REST_Request( 'GET', '/wc/v2/products/' . $product->get_id() . '/variations/' . $variation_id ) );
		$variation  = $response->get_data();

		$this->assertEquals( 200, $response->get_status() );
		$this->assertEquals( $variation_id, $variation['id'] );
		$this->assertEquals( 'size', $variation['attributes'][0]['name'] );
		$product->delete( true );
	}

	/**
	 * Test getting single variation without permission.
	 *
	 * @since 3.0.0
	 */
	public function test_get_variation_without_permission() {
		wp_set_current_user( 0 );
		$product      = WC_Helper_Product::create_variation_product();
		$children     = $product->get_children();
		$variation_id = $children[0];
		$response   = $this->server->dispatch( new WP_REST_Request( 'GET', '/wc/v2/products/' . $product->get_id() . '/variations/' . $variation_id ) );
		$this->assertEquals( 401, $response->get_status() );
		$product->delete( true );
	}

	/**
	 * Test deleting a single variation.
	 *
	 * @since 3.0.0
	 */
	public function test_delete_variation() {
		wp_set_current_user( $this->user );
		$product      = WC_Helper_Product::create_variation_product();
		$children     = $product->get_children();
		$variation_id = $children[0];

		$request = new WP_REST_Request( 'DELETE', '/wc/v2/products/' . $product->get_id() . '/variations/' . $variation_id );
		$request->set_param( 'force', true );
		$response = $this->server->dispatch( $request );
		$this->assertEquals( 200, $response->get_status() );

		$response   = $this->server->dispatch( new WP_REST_Request( 'GET', '/wc/v2/products/' . $product->get_id() . '/variations' ) );
		$variations = $response->get_data();
		$this->assertEquals( 1, count( $variations ) );
		$product->delete( true );
	}

	/**
	 * Test deleting a single variation without permission.
	 *
	 * @since 3.0.0
	 */
	public function test_delete_variation_without_permission() {
		wp_set_current_user( 0 );
		$product      = WC_Helper_Product::create_variation_product();
		$children     = $product->get_children();
		$variation_id = $children[0];

		$request = new WP_REST_Request( 'DELETE', '/wc/v2/products/' . $product->get_id() . '/variations/' . $variation_id );
		$request->set_param( 'force', true );
		$response = $this->server->dispatch( $request );
		$this->assertEquals( 401, $response->get_status() );
		$product->delete( true );
	}

	/**
	 * Test deleting a single variation with an invalid ID.
	 *
	 * @since 3.0.0
	 */
	public function test_delete_variation_with_invalid_id() {
		wp_set_current_user( 0 );
		$product = WC_Helper_Product::create_variation_product();
		$request = new WP_REST_Request( 'DELETE', '/wc/v2/products/' . $product->get_id() . '/variations/0' );
		$request->set_param( 'force', true );
		$response = $this->server->dispatch( $request );
		$this->assertEquals( 404, $response->get_status() );
		$product->delete( true );
	}

	/**
	 * Test editing a single variation.
	 *
	 * @since 3.0.0
	 */
	public function test_update_variation() {
		wp_set_current_user( $this->user );
		$product      = WC_Helper_Product::create_variation_product();
		$children     = $product->get_children();
		$variation_id = $children[0];

		$response  = $this->server->dispatch( new WP_REST_Request( 'GET', '/wc/v2/products/' . $product->get_id() . '/variations/' . $variation_id ) );
		$variation = $response->get_data();

		$this->assertEquals( 'DUMMY SKU VARIABLE SMALL', $variation['sku'] );
		$this->assertEquals( 10, $variation['regular_price'] );
		$this->assertEmpty( $variation['sale_price'] );
		$this->assertEquals( 'small', $variation['attributes'][0]['option'] );

		$request = new WP_REST_Request( 'PUT', '/wc/v2/products/' . $product->get_id() . '/variations/' . $variation_id );
		$request->set_body_params( array(
			'sku'         => 'FIXED-SKU',
			'sale_price'  => '8',
			'description' => 'O_O',
			'image'       => array( 'position' => 0, 'src' => 'https://cldup.com/Dr1Bczxq4q.png', 'alt' => 'test upload image' ),
			'attributes'  => array( array( 'name' => 'pa_size', 'option' => 'medium' ) ),
		) );
		$response  = $this->server->dispatch( $request );
		$variation = $response->get_data();

		$this->assertContains( 'O_O', $variation['description'] );
		$this->assertEquals( '8', $variation['price'] );
		$this->assertEquals( '8', $variation['sale_price'] );
		$this->assertEquals( '10', $variation['regular_price'] );
		$this->assertEquals( 'FIXED-SKU', $variation['sku'] );
		$this->assertEquals( 'medium', $variation['attributes'][0]['option'] );
		$this->assertContains( 'Dr1Bczxq4q', $variation['image']['src'] );
		$this->assertContains( 'test upload image', $variation['image']['alt'] );
		$product->delete( true );
	}

	/**
	 * Test updating a single variation without permission.
	 *
	 * @since 3.0.0
	 */
	public function test_update_variation_without_permission() {
		wp_set_current_user( 0 );
		$product      = WC_Helper_Product::create_variation_product();
		$children     = $product->get_children();
		$variation_id = $children[0];

		$request = new WP_REST_Request( 'PUT', '/wc/v2/products/' . $product->get_id() . '/variations/' . $variation_id );
		$request->set_body_params( array(
			'sku'         => 'FIXED-SKU-NO-PERMISSION',
		) );
		$response = $this->server->dispatch( $request );
		$this->assertEquals( 401, $response->get_status() );
		$product->delete( true );
	}

	/**
	 * Test updating a single variation with an invalid ID.
	 *
	 * @since 3.0.0
	 */
	public function test_update_variation_with_invalid_id() {
		wp_set_current_user( $this->user );
		$product = WC_Helper_Product::create_variation_product();
		$request = new WP_REST_Request( 'PUT', '/wc/v2/products/' . $product->get_id() . '/variations/0' );
		$request->set_body_params( array(
			'sku'         => 'FIXED-SKU-NO-PERMISSION',
		) );
		$response = $this->server->dispatch( $request );
		$this->assertEquals( 400, $response->get_status() );
		$product->delete( true );
	}

	/**
	 * Test creating a single variation.
	 *
	 * @since 3.0.0
	 */
	public function test_create_variation() {
		wp_set_current_user( $this->user );
		$product = WC_Helper_Product::create_variation_product();

		$response   = $this->server->dispatch( new WP_REST_Request( 'GET', '/wc/v2/products/' . $product->get_id() . '/variations' ) );
		$variations = $response->get_data();
		$this->assertEquals( 2, count( $variations ) );

		$request = new WP_REST_Request( 'POST', '/wc/v2/products/' . $product->get_id() . '/variations' );
		$request->set_body_params( array(
			'sku'            => 'DUMMY SKU VARIABLE MEDIUM',
			'regular_price'  => '12',
			'description'    => 'A medium size.',
			'attributes'     => array( array( 'name' => 'pa_size', 'option' => 'medium' ) ),
		) );
		$response  = $this->server->dispatch( $request );
		$variation = $response->get_data();

		$this->assertContains( 'A medium size.', $variation['description'] );
		$this->assertEquals( '12', $variation['price'] );
		$this->assertEquals( '12', $variation['regular_price'] );
		$this->assertTrue( $variation['purchasable'] );
		$this->assertEquals( 'DUMMY SKU VARIABLE MEDIUM', $variation['sku'] );
		$this->assertEquals( 'medium', $variation['attributes'][0]['option'] );

		$response   = $this->server->dispatch( new WP_REST_Request( 'GET', '/wc/v2/products/' . $product->get_id() . '/variations' ) );
		$variations = $response->get_data();
		$this->assertEquals( 3, count( $variations ) );
		$product->delete( true );
	}

	/**
	 * Test creating a single variation without permission.
	 *
	 * @since 3.0.0
	 */
	public function test_create_variation_without_permission() {
		wp_set_current_user( 0 );
		$product      = WC_Helper_Product::create_variation_product();

		$request = new WP_REST_Request( 'POST', '/wc/v2/products/' . $product->get_id() . '/variations' );
		$request->set_body_params( array(
			'sku'            => 'DUMMY SKU VARIABLE MEDIUM',
			'regular_price'  => '12',
			'description'    => 'A medium size.',
			'attributes'     => array( array( 'name' => 'pa_size', 'option' => 'medium' ) ),
		) );
		$response = $this->server->dispatch( $request );
		$this->assertEquals( 401, $response->get_status() );
		$product->delete( true );
	}

	/**
	 * Test batch managing product variations.
	 */
	public function test_product_variations_batch() {
		wp_set_current_user( $this->user );
		$product  = WC_Helper_Product::create_variation_product();
		$children = $product->get_children();
		$request  = new WP_REST_Request( 'POST', '/wc/v2/products/' . $product->get_id() . '/variations/batch' );
		$request->set_body_params( array(
			'update' => array(
				array(
					'id'          => $children[0],
					'description' => 'Updated description.',
					'image'       => array( 'position' => 0, 'src' => 'https://cldup.com/Dr1Bczxq4q.png', 'alt' => 'test upload image' ),
				),
			),
			'delete' => array(
				$children[1],
			),
			'create' => array(
				array(
					'sku'            => 'DUMMY SKU VARIABLE MEDIUM',
					'regular_price'  => '12',
					'description'    => 'A medium size.',
					'attributes'     => array( array( 'name' => 'pa_size', 'option' => 'medium' ) ),
				),
			),
		) );
		$response = $this->server->dispatch( $request );
		$data     = $response->get_data();

		$this->assertContains( 'Updated description.', $data['update'][0]['description'] );
		$this->assertEquals( 'DUMMY SKU VARIABLE MEDIUM', $data['create'][0]['sku'] );
		$this->assertEquals( 'medium', $data['create'][0]['attributes'][0]['option'] );
		$this->assertEquals( $children[1], $data['delete'][0]['id'] );

		$request = new WP_REST_Request( 'GET', '/wc/v2/products/' . $product->get_id() . '/variations' );
		$response = $this->server->dispatch( $request );
		$data = $response->get_data();

		$this->assertEquals( 2, count( $data ) );
		$product->delete( true );
	}

	/**
	 * Test variation schema.
	 *
	 * @since 3.0.0
	 */
	public function test_variation_schema() {
		wp_set_current_user( $this->user );
		$product = WC_Helper_Product::create_simple_product();
		$request = new WP_REST_Request( 'OPTIONS', '/wc/v2/products/' . $product->get_id() . '/variations' );
		$response = $this->server->dispatch( $request );
		$data = $response->get_data();
		$properties = $data['schema']['properties'];

		$this->assertEquals( 37, count( $properties ) );
		$this->assertArrayHasKey( 'id', $properties );
		$this->assertArrayHasKey( 'date_created', $properties );
		$this->assertArrayHasKey( 'date_modified', $properties );
		$this->assertArrayHasKey( 'description', $properties );
		$this->assertArrayHasKey( 'permalink', $properties );
		$this->assertArrayHasKey( 'sku', $properties );
		$this->assertArrayHasKey( 'price', $properties );
		$this->assertArrayHasKey( 'regular_price', $properties );
		$this->assertArrayHasKey( 'sale_price', $properties );
		$this->assertArrayHasKey( 'date_on_sale_from', $properties );
		$this->assertArrayHasKey( 'date_on_sale_to', $properties );
		$this->assertArrayHasKey( 'on_sale', $properties );
		$this->assertArrayHasKey( 'visible', $properties );
		$this->assertArrayHasKey( 'purchasable', $properties );
		$this->assertArrayHasKey( 'virtual', $properties );
		$this->assertArrayHasKey( 'downloadable', $properties );
		$this->assertArrayHasKey( 'downloads', $properties );
		$this->assertArrayHasKey( 'download_limit', $properties );
		$this->assertArrayHasKey( 'download_expiry', $properties );
		$this->assertArrayHasKey( 'tax_status', $properties );
		$this->assertArrayHasKey( 'tax_class', $properties );
		$this->assertArrayHasKey( 'manage_stock', $properties );
		$this->assertArrayHasKey( 'stock_quantity', $properties );
		$this->assertArrayHasKey( 'in_stock', $properties );
		$this->assertArrayHasKey( 'backorders', $properties );
		$this->assertArrayHasKey( 'backorders_allowed', $properties );
		$this->assertArrayHasKey( 'backordered', $properties );
		$this->assertArrayHasKey( 'weight', $properties );
		$this->assertArrayHasKey( 'dimensions', $properties );
		$this->assertArrayHasKey( 'shipping_class', $properties );
		$this->assertArrayHasKey( 'shipping_class_id', $properties );
		$this->assertArrayHasKey( 'image', $properties );
		$this->assertArrayHasKey( 'attributes', $properties );
		$this->assertArrayHasKey( 'menu_order', $properties );
		$this->assertArrayHasKey( 'meta_data', $properties );
		$product->delete( true );
	}

}