wp-multisite-waas/tests/WP_Ultimo/Models/Membership_Test.php

<?php

namespace WP_Ultimo\Models;

use WP_Ultimo\Faker;

/**
 * Unit tests for the Membership class.
 */
class Membership_Test extends \WP_UnitTestCase {

	/**
	 * Membership instance.
	 *
	 * @var Membership
	 */
	protected $membership;

	/**
	 * Set up the test environment.
	 */
	public function setUp(): void {
		parent::setUp();

		// Create a new Membership instance for each test.
		$this->membership = new Membership();

		// Set a default customer ID.
		$this->membership->set_customer_id( 123 );
	}

	/**
	 * Test if the customer is allowed access to the membership.
	 */
	public function test_is_customer_allowed() {
		// Admins with 'manage_network' capability should always return true.
		$admin_user_id = $this->factory()->user->create( array( 'role' => 'administrator' ) );
		grant_super_admin( $admin_user_id );
		wp_set_current_user( $admin_user_id );
		$this->assertTrue( $this->membership->is_customer_allowed(), 'Failed asserting that admin is allowed.' );

		// Regular customers are allowed if IDs match.
		$customer_id = 123;
		$this->assertTrue(
			$this->membership->is_customer_allowed( $customer_id ),
			'Failed asserting that customer with matching ID is allowed.'
		);

		// Regular customers are denied if IDs do not match.
		$wrong_customer_id = 456;
		wp_set_current_user( $wrong_customer_id );
		$this->assertFalse(
			$this->membership->is_customer_allowed( $wrong_customer_id ),
			'Failed asserting that customer with non-matching ID is denied.'
		);
	}

	/**
	 * Test adding a product to the membership.
	 */
	public function test_add_product() {
		// Add a product with a specific ID and quantity.
		$quantity = 2;
		$faker    = new Faker();
		$faker->generate_fake_products();
		/** @var Product $product */
		$product    = $faker->get_fake_data_generated( 'products' )[0];
		$product_id = $product->get_id();

		$this->membership->add_product( $product_id, $quantity );

		// Verify that the product is added with the correct quantity.
		$addon_products = $this->membership->get_addon_ids();
		$this->assertContains( $product_id, $addon_products, 'Failed asserting that product ID was added.' );
		$this->assertEquals(
			$quantity,
			$this->membership->get_addon_products()[0]['quantity'],
			'Failed asserting that the product quantity is correct.'
		);

		// Add more of the same product and check the updated quantity.
		$additional_quantity = 3;
		$this->membership->add_product( $product_id, $additional_quantity );

		$this->assertEquals(
			$quantity + $additional_quantity,
			$this->membership->get_addon_products()[0]['quantity'],
			'Failed asserting that the quantity was updated correctly for the same product.'
		);
	}

	/**
	 * Test removing a product from the membership.
	 */
	public function test_remove_product() {
		// Add a product with a specific quantity.
		$quantity = 5;
		$faker    = new Faker();
		$faker->generate_fake_products();
		/** @var Product $product */
		$product    = $faker->get_fake_data_generated( 'products' )[0];
		$product_id = $product->get_id();

		$this->membership->add_product( $product_id, $quantity );

		// Remove some of the product's quantity.
		$remove_quantity = 3;
		$this->membership->remove_product( $product_id, $remove_quantity );

		// Verify the updated quantity.
		$this->assertEquals(
			$quantity - $remove_quantity,
			$this->membership->get_addon_products()[0]['quantity'],
			'Failed asserting that the quantity was reduced correctly.'
		);

		// Remove the remaining quantity and verify it is removed.
		$this->membership->remove_product( $product_id, $quantity );
		$addon_products = $this->membership->get_addon_ids();
		$this->assertNotContains( $product_id, $addon_products, 'Failed asserting that the product was removed.' );
	}

	/**
	 * Test get_remaining_days_in_cycle() method.
	 */
	public function test_get_remaining_days_in_cycle() {
		$this->membership->set_amount( 12.99 );
		// Case 1: Non-recurring membership should return 10000.
		$this->membership->set_recurring( false );
		$this->assertEquals(
			10000,
			$this->membership->get_remaining_days_in_cycle(),
			'Failed asserting that non-recurring membership returns 10000.'
		);

		// Case 2: Invalid expiration date should return 0.
		$this->membership->set_recurring( true );
		$this->membership->set_date_expiration( 'invalid-date' ); // Setting an invalid date.
		$this->assertEquals(
			0,
			$this->membership->get_remaining_days_in_cycle(),
			'Failed asserting that an invalid expiration date returns 0.'
		);

		// Case 3: No expiration date should return 0.
		$this->membership->set_date_expiration( '' );
		$this->assertEquals(
			0,
			$this->membership->get_remaining_days_in_cycle(),
			'Failed asserting that no expiration date returns 0.'
		);

		// Case 4: Expiration date is in the future and remaining days are calculated properly.
		$today = new \DateTime( 'now', new \DateTimeZone( 'UTC' ) );
		$today->add( new \DateInterval( 'P10D' ) );
		$this->membership->set_date_expiration( $today->format( 'Y-m-d H:i:s' ) );
		$remaining_days = $this->membership->get_remaining_days_in_cycle();
		$this->assertEquals(
			10,
			$remaining_days,
			'Failed asserting that 10 days remain when expiration date is 10 days in the future.'
		);

		// Case 5: Expiration date is in the past, should return 0.
		$this->membership->set_date_expiration( date( 'Y-m-d H:i:s', strtotime( '-5 days' ) ) );
		$remaining_days = $this->membership->get_remaining_days_in_cycle();
		$this->assertEquals(
			0,
			$remaining_days,
			'Failed asserting that remaining days return 0 when the expiration date is in the past.'
		);
	}
}