Add tests for WC_Order_Data_Store_Custom_Table::create()

Author: stevegrunwell

tests/test-data-store.php

@@ -8,6 +8,15 @@
 
 class DataStoreTest extends TestCase {
 
+	/**
+	 * Fire the necessary actions to bootstrap WordPress.
+	 *
+	 * @before
+	 */
+	public function init() {
+		do_action( 'init' );
+	}
+
 	/**
 	 * Remove any closures that have been assigned to filters.
 	 *
@@ -17,6 +26,20 @@ public function remove_filter_callbacks() {
 		remove_all_filters( 'woocommerce_shop_order_search_fields' );
 	}
 
+	public function test_create() {
+		$instance  = new WC_Order_Data_Store_Custom_Table();
+		$order     = $this->factory()->order->create_and_get();
+		$order_key = 'my_custom_order_key';
+
+		add_filter( 'woocommerce_generate_order_key', function () use ( $order_key ) {
+			return $order_key;
+		} );
+
+		$instance->create( $order );
+
+		$this->assertEquals( 'wc_' . $order_key, $order->get_order_key() );
+	}
+
 	public function test_get_order_count() {
 		$orders = $this->factory()->order->create_many( 5, array(
 			'post_status' => 'wc-pending',
@@ -107,7 +130,7 @@ public function test_search_orders_only_checks_post_meta_if_specified() {
 	 * @dataProvider order_type_provider()
 	 */
 	public function test_get_order_type( $order_type ) {
-		$order = $this->factory()->order->create_and_get( array(
+		$order = $this->factory()->order->create( array(
 			'post_type' => $order_type,
 		) );
 

tests/test-tools/class-wp-unittest-factory-for-order.php

@@ -21,4 +21,8 @@ function __construct( $factory = null ) {
 			'post_type'     => 'shop_order',
 		);
 	}
+
+	function get_object_by_id( $order_id ) {
+		return wc_get_order( $order_id );
+	}
 }

tests/test-tools/class-wp-unittest-factory-for-product.php

@@ -22,4 +22,8 @@ function __construct( $factory = null ) {
 			'post_type'    => 'post',
 		);
 	}
+
+	function get_object_by_id( $product_id ) {
+		return wc_get_product( $product_id );
+	}
 }