woocommerce/tests/legacy/unit-tests/customer/crud.php

<?php

/**
 * Class CustomerCRUD.
 * @package WooCommerce\Tests\Customer
 */
class WC_Tests_CustomerCRUD extends WC_Unit_Test_Case {

	/**
	 * Test creating a new customer.
	 * @since 3.0.0
	 */
	public function test_create_customer() {
		$username = 'testusername-' . time();
		$customer = new WC_Customer();
		$customer->set_username( $username );
		$customer->set_password( 'test123' );
		$customer->set_email( 'test@woo.local' );
		$customer->save();
		$wp_user = new WP_User( $customer->get_id() );

		$this->assertEquals( $username, $customer->get_username() );
		$this->assertNotEquals( 0, $customer->get_id() );
		$this->assertEquals( strtotime( $wp_user->user_registered ), $customer->get_date_created()->getOffsetTimestamp() );
	}

	/**
	 * Test updating a customer.
	 * @since 3.0.0
	 */
	public function test_update_customer() {
		$customer    = WC_Helper_Customer::create_customer();
		$customer_id = $customer->get_id();
		$this->assertEquals( 'test@woo.local', $customer->get_email() );
		$this->assertEquals( 'Apt 1', $customer->get_billing_address_2() );
		$customer->set_email( 'test@wc.local' );
		$customer->set_first_name( 'Justin' );
		$customer->set_billing_address_2( 'Apt 5' );
		$customer->save();

		$customer = new WC_Customer( $customer_id ); // so we can read fresh copies from the DB
		$this->assertEquals( 'test@wc.local', $customer->get_email() );
		$this->assertEquals( 'Justin', $customer->get_first_name() );
		$this->assertEquals( 'Apt 5', $customer->get_billing_address_2() );
	}


	/**
	 * Test saving a customer.
	 * @since 3.0.0
	 */
	public function test_save_customer() {
		// test save() -> Create
		$customer = new WC_Customer();
		$customer->set_username( 'testusername-' . time() );
		$customer->set_password( 'test123' );
		$customer->set_email( 'test@woo.local' );
		$this->assertEquals( 0, $customer->get_id() );
		$customer->save();
		$customer_id = $customer->get_id();
		$wp_user     = new WP_User( $customer->get_id() );

		$this->assertNotEquals( 0, $customer->get_id() );

		// test save() -> Update
		$this->assertEquals( 'test@woo.local', $customer->get_email() );
		$customer->set_email( 'test@wc.local' );
		$customer->save();

		$customer = new WC_Customer( $customer_id );
		$this->assertEquals( 'test@wc.local', $customer->get_email() );
	}

	/**
	 * Test deleting a customer.
	 * @since 3.0.0
	 */
	public function test_delete_customer() {
		$customer    = WC_Helper_Customer::create_customer();
		$customer_id = $customer->get_id();
		$this->assertNotEquals( 0, $customer->get_id() );
		$customer->delete();
		$this->assertEquals( 0, $customer->get_id() );
	}

	/**
	 * Test reading a customer.
	 * @since 3.0.0
	 */
	public function test_read_customer() {
		$username = 'user-' . time();
		$customer = new WC_Customer();
		$customer->set_username( $username );
		$customer->set_email( 'test@woo.local' );
		$customer->set_password( 'hunter2' );
		$customer->set_first_name( 'Billy' );
		$customer->set_last_name( 'Bob' );
		$customer->set_display_name( 'Billy Bob' );
		$customer->save();
		$customer_id   = $customer->get_id();
		$customer_read = new WC_Customer( $customer_id );

		$this->assertEquals( $customer_id, $customer_read->get_id() );
		$this->assertEquals( 'test@woo.local', $customer_read->get_email() );
		$this->assertEquals( 'Billy', $customer_read->get_first_name() );
		$this->assertEquals( 'Bob', $customer_read->get_last_name() );
		$this->assertEquals( 'Billy Bob', $customer_read->get_display_name() );
		$this->assertEquals( $username, $customer_read->get_username() );
	}

	/**
	 * Tests backwards compat / legacy handling.
	 * @expectedDeprecated WC_Customer::get_default_country
	 * @expectedDeprecated WC_Customer::get_default_state
	 * @expectedDeprecated WC_Customer::is_paying_customer
	 * @expectedDeprecated WC_Customer::calculated_shipping
	 * @since 3.0.0
	 */
	public function test_customer_backwards_compat() {
		// Properties.
		// Accessing properties directly will throw some wanted deprected notices
		// So we need to let PHPUnit know we are expecting them and it's fine to continue
		$legacy_keys = array(
			'id',
			'country',
			'state',
			'postcode',
			'city',
			'address',
			'address_1',
			'address_2',
			'shipping_country',
			'shipping_state',
			'shipping_postcode',
			'shipping_city',
			'shipping_address',
			'shipping_address_1',
			'shipping_address_2',
			'is_vat_exempt',
			'calculated_shipping',
		);

		$this->expected_doing_it_wrong = array_merge( $this->expected_doing_it_wrong, $legacy_keys );

		$customer = WC_Helper_Customer::create_customer();

		$this->assertEquals( $customer->get_id(), $customer->id );
		$this->assertEquals( $customer->get_billing_country(), $customer->country );
		$this->assertEquals( $customer->get_billing_state(), $customer->state );
		$this->assertEquals( $customer->get_billing_postcode(), $customer->postcode );
		$this->assertEquals( $customer->get_billing_city(), $customer->city );
		$this->assertEquals( $customer->get_billing_address(), $customer->address );
		$this->assertEquals( $customer->get_billing_address(), $customer->address_1 );
		$this->assertEquals( $customer->get_billing_address_2(), $customer->address_2 );
		$this->assertEquals( $customer->get_shipping_country(), $customer->shipping_country );
		$this->assertEquals( $customer->get_shipping_state(), $customer->shipping_state );
		$this->assertEquals( $customer->get_shipping_postcode(), $customer->shipping_postcode );
		$this->assertEquals( $customer->get_shipping_city(), $customer->shipping_city );
		$this->assertEquals( $customer->get_shipping_address(), $customer->shipping_address );
		$this->assertEquals( $customer->get_shipping_address(), $customer->shipping_address_1 );
		$this->assertEquals( $customer->get_shipping_address_2(), $customer->shipping_address_2 );
		$this->assertEquals( $customer->get_is_vat_exempt(), $customer->is_vat_exempt );
		$this->assertEquals( $customer->has_calculated_shipping(), $customer->calculated_shipping );

		// Functions
		$this->assertEquals( $customer->get_is_vat_exempt(), $customer->is_vat_exempt() );
		$this->assertEquals( $customer->has_calculated_shipping(), $customer->has_calculated_shipping() );
		$default = wc_get_customer_default_location();
		$this->assertEquals( $default['country'], $customer->get_default_country() );
		$this->assertEquals( $default['state'], $customer->get_default_state() );
		$this->assertFalse( $customer->has_calculated_shipping() );
		$customer->calculated_shipping( true );
		$this->assertTrue( $customer->has_calculated_shipping() );
		$this->assertEquals( $customer->get_is_paying_customer(), $customer->is_paying_customer() );
	}

	/**
	 * Test generic getters & setters
	 * @since 3.0.0
	 */
	public function test_customer_setters_and_getters() {
		$time    = time();
		$setters = array(
			'username'            => 'test',
			'email'               => 'test@woo.local',
			'first_name'          => 'Bob',
			'last_name'           => 'tester',
			'display_name'        => 'Bob Tester',
			'role'                => 'customer',
			'date_created'        => $time,
			'date_modified'       => $time,
			'billing_postcode'    => 11010,
			'billing_city'        => 'New York',
			'billing_address'     => '123 Main St.',
			'billing_address_1'   => '123 Main St.',
			'billing_address_2'   => 'Apt 2',
			'billing_state'       => 'NY',
			'billing_country'     => 'US',
			'shipping_state'      => 'NY',
			'shipping_postcode'   => 11011,
			'shipping_city'       => 'New York',
			'shipping_address'    => '123 Main St.',
			'shipping_address_1'  => '123 Main St.',
			'shipping_address_2'  => 'Apt 2',
			'is_vat_exempt'       => true,
			'calculated_shipping' => true,
			'is_paying_customer'  => true,
		);

		$customer = new WC_Customer();

		foreach ( $setters as $method => $value ) {
			$customer->{"set_{$method}"}( $value );
		}

		$getters = array();

		foreach ( $setters as $method => $value ) {
			$getters[ $method ] = $customer->{"get_{$method}"}();
		}

		// Get timestamps from date_created and date_modified.
		$getters['date_created']  = $getters['date_created']->getOffsetTimestamp();
		$getters['date_modified'] = $getters['date_modified']->getOffsetTimestamp();

		$this->assertEquals( $setters, $getters );
	}

	/**
	 * Test getting a customer's last order ID and date
	 * @since 3.0.0
	 */
	public function test_customer_get_last_order_info() {
		$customer    = WC_Helper_Customer::create_customer();
		$customer_id = $customer->get_id();
		$order       = WC_Helper_Order::create_order( $customer_id );
		$customer    = new WC_Customer( $customer_id );
		$last_order  = $customer->get_last_order();
		$this->assertEquals( $order->get_id(), $last_order ? $last_order->get_id() : 0 );
		$this->assertEquals( $order->get_date_created(), $last_order ? $last_order->get_date_created() : 0 );
		$order->delete();
	}

	/**
	 * Test getting a customer's order count from DB.
	 * @since 3.0.0
	 */
	public function test_customer_get_order_count_read() {
		$customer    = WC_Helper_Customer::create_customer();
		$customer_id = $customer->get_id();
		WC_Helper_Order::create_order( $customer_id );
		WC_Helper_Order::create_order( $customer_id );
		WC_Helper_Order::create_order( $customer_id );
		$customer = new WC_Customer( $customer_id );
		$this->assertEquals( 3, $customer->get_order_count() );
	}

	/**
	 * Test getting a customer's total amount spent from DB.
	 * @since 3.0.0
	 */
	public function test_customer_get_total_spent_read() {
		$customer    = WC_Helper_Customer::create_customer();
		$customer_id = $customer->get_id();
		$order       = WC_Helper_Order::create_order( $customer_id );
		$customer    = new WC_Customer( $customer_id );
		$this->assertEquals( 0, $customer->get_total_spent() );
		$order->update_status( 'wc-completed' );
		$customer = new WC_Customer( $customer_id );
		$this->assertEquals( 50, $customer->get_total_spent() );
		$order->delete();
	}

	/**
	 * Test getting a customer's avatar URL.
	 * @since 3.0.0
	 */
	public function test_customer_get_avatar_url() {
		$customer = WC_Helper_Customer::create_customer();
		$this->assertContains( 'gravatar.com/avatar', $customer->get_avatar_url() );
		$this->assertContains( md5( 'test@woo.local' ), $customer->get_avatar_url() );
	}

	/**
	 * Test getting a customer's creation date from DB.
	 * @since 3.0.0
	 */
	public function test_customer_get_date_created_read() {
		$customer    = WC_Helper_Customer::create_customer();
		$customer_id = $customer->get_id();
		$user        = new WP_User( $customer_id );
		$this->assertEquals( strtotime( $user->data->user_registered ), $customer->get_date_created()->getOffsetTimestamp() );
	}

	/**
	 * Test getting a customer's modification date from DB.
	 * @since 3.0.0
	 */
	public function test_customer_get_date_modified_read() {
		$customer    = WC_Helper_Customer::create_customer();
		$customer_id = $customer->get_id();
		$last        = get_user_meta( $customer_id, 'last_update', true );
		sleep( 1 );
		$this->assertEquals( $last, $customer->get_date_modified()->getOffsetTimestamp() );
		$customer->set_billing_address( '1234 Some St.' );
		$customer->save();
		$update = get_user_meta( $customer_id, 'last_update', true );
		$this->assertEquals( $update, $customer->get_date_modified()->getOffsetTimestamp() );
		$this->assertNotEquals( $update, $last );
	}

	/**
	 * Test getting a customer's taxable address.
	 * @since 3.0.0
	 */
	public function test_customer_get_taxable_address() {
		$customer    = WC_Helper_Customer::create_customer();
		$customer_id = $customer->get_id();
		$customer->set_shipping_postcode( '11111' );
		$customer->set_shipping_city( 'Test' );
		$customer->save();
		$customer = new WC_Customer( $customer_id );

		update_option( 'woocommerce_tax_based_on', 'shipping' );
		$taxable = $customer->get_taxable_address();
		$this->assertEquals( 'US', $taxable[0] );
		$this->assertEquals( 'PA', $taxable[1] );
		$this->assertEquals( '11111', $taxable[2] );
		$this->assertEquals( 'Test', $taxable[3] );

		update_option( 'woocommerce_tax_based_on', 'billing' );
		$taxable = $customer->get_taxable_address();
		$this->assertEquals( 'US', $taxable[0] );
		$this->assertEquals( 'PA', $taxable[1] );
		$this->assertEquals( '19123', $taxable[2] );
		$this->assertEquals( 'Philadelphia', $taxable[3] );

		update_option( 'woocommerce_tax_based_on', 'base' );
		$taxable = $customer->get_taxable_address();
		$this->assertEquals( WC()->countries->get_base_country(), $taxable[0] );
		$this->assertEquals( WC()->countries->get_base_state(), $taxable[1] );
		$this->assertEquals( WC()->countries->get_base_postcode(), $taxable[2] );
		$this->assertEquals( WC()->countries->get_base_city(), $taxable[3] );
	}

	/**
	 * Test getting a customer's downloadable products.
	 * @since 3.0.0
	 */
	public function test_customer_get_downloadable_products() {
		$customer    = WC_Helper_Customer::create_customer();
		$customer_id = $customer->get_id();
		$this->assertEquals( wc_get_customer_available_downloads( $customer_id ), $customer->get_downloadable_products() );
	}

	/**
	 * Test setting a password on update - making sure it actually changes the users password.
	 * @since 3.0.0
	 */
	public function test_customer_password() {
		$customer    = WC_Helper_Customer::create_customer();
		$customer_id = $customer->get_id();

		$user = get_user_by( 'id', $customer_id );
		$this->assertTrue( wp_check_password( 'hunter2', $user->data->user_pass, $user->ID ) );

		$customer->set_password( 'hunter3' );
		$customer->save();

		$user = get_user_by( 'id', $customer_id );
		$this->assertTrue( wp_check_password( 'hunter3', $user->data->user_pass, $user->ID ) );
	}

	/**
	 * Test setting a customer's address to the base address.
	 * @since 3.0.0
	 */
	public function test_customer_set_address_to_base() {
		$customer = WC_Helper_Customer::create_customer();
		$customer->set_billing_address_to_base();
		$base = wc_get_customer_default_location();
		$this->assertEquals( $base['country'], $customer->get_billing_country() );
		$this->assertEquals( $base['state'], $customer->get_billing_state() );
		$this->assertEmpty( $customer->get_billing_postcode() );
		$this->assertEmpty( $customer->get_billing_city() );
	}

	/**
	 * Test setting a customer's shipping address to the base address.
	 * @since 3.0.0
	 */
	public function test_customer_set_shipping_address_to_base() {
		$customer = WC_Helper_Customer::create_customer();
		$customer->set_shipping_address_to_base();
		$base = wc_get_customer_default_location();
		$this->assertEquals( $base['country'], $customer->get_shipping_country() );
		$this->assertEquals( $base['state'], $customer->get_shipping_state() );
		$this->assertEmpty( $customer->get_shipping_postcode() );
		$this->assertEmpty( $customer->get_shipping_city() );
	}

	/**
	 * Test setting a customer's location (multiple address fields at once)
	 * @since 3.0.0
	 */
	public function test_customer_set_location() {
		$customer = WC_Helper_Customer::create_customer();
		$customer->set_billing_location( 'US', 'OH', '12345', 'Cleveland' );
		$this->assertEquals( 'US', $customer->get_billing_country() );
		$this->assertEquals( 'OH', $customer->get_billing_state() );
		$this->assertEquals( '12345', $customer->get_billing_postcode() );
		$this->assertEquals( 'Cleveland', $customer->get_billing_city() );
	}

	/**
	 * Test setting a customer's shipping location (multiple address fields at once)
	 * @since 3.0.0
	 */
	public function test_customer_set_shipping_location() {
		$customer = WC_Helper_Customer::create_customer();
		$customer->set_shipping_location( 'US', 'OH', '12345', 'Cleveland' );
		$this->assertEquals( 'US', $customer->get_shipping_country() );
		$this->assertEquals( 'OH', $customer->get_shipping_state() );
		$this->assertEquals( '12345', $customer->get_shipping_postcode() );
		$this->assertEquals( 'Cleveland', $customer->get_shipping_city() );
	}

	/**
	 * Test is_customer_outside_base.
	 * @since 3.0.0
	 */
	public function test_customer_is_customer_outside_base() {
		$customer = WC_Helper_Customer::create_customer();
		$this->assertTrue( $customer->is_customer_outside_base() );
		update_option( 'woocommerce_tax_based_on', 'base' );
		$customer->set_billing_address_to_base();
		$this->assertFalse( $customer->is_customer_outside_base() );
	}

	/**
	 * Test WC_Customer's session handling code.
	 * @since 3.0.0
	 */
	public function test_customer_sessions() {
		$session = WC_Helper_Customer::create_mock_customer(); // set into session....

		$this->assertEquals( '19123', $session->get_billing_postcode() );
		$this->assertEquals( '123 South Street', $session->get_billing_address() );
		$this->assertEquals( 'Philadelphia', $session->get_billing_city() );

		$session->set_billing_address( '124 South Street' );
		$session->save();

		$session = new WC_Customer( 0, true );
		$this->assertEquals( '124 South Street', $session->get_billing_address() );

		$session = new WC_Customer( 0, true );
		$session->set_billing_postcode( '32191' );
		$session->save();

		// should still be session ID, not a created row, since we are working with guests/sessions
		$this->assertFalse( $session->get_id() > 0 );
		$this->assertEquals( '32191', $session->get_billing_postcode() );
	}

	/**
	 * Test getting meta.
	 * @since 3.0.0
	 */
	public function test_get_meta() {
		$customer    = WC_Helper_Customer::create_customer();
		$customer_id = $customer->get_id();
		$meta_value  = time() . '-custom-value';
		add_user_meta( $customer_id, 'test_field', $meta_value, true );
		$customer = new WC_Customer( $customer_id );
		$fields   = $customer->get_meta_data();
		$this->assertEquals( $meta_value, $customer->get_meta( 'test_field' ) );
	}

	/**
	 * Test setting meta.
	 * @since 3.0.0
	 */
	public function test_set_meta() {
		$customer    = WC_Helper_Customer::create_customer();
		$customer_id = $customer->get_id();
		$meta_value  = time() . '-custom-value';
		$customer->add_meta_data( 'my-field', $meta_value, true );
		$customer->save();
		$customer = new WC_Customer( $customer_id );
		$this->assertEquals( $meta_value, $customer->get_meta( 'my-field' ) );
	}
}