Added tests.

omise · Jan 8, 2024 · fe2a1a2 · fe2a1a2
1 parent 271b47a
commit fe2a1a2
Show file tree

Hide file tree

Showing 2 changed files with 95 additions and 16 deletions.
diff --git a/includes/gateway/class-omise-payment-truemoney.php b/includes/gateway/class-omise-payment-truemoney.php
@@ -75,9 +75,8 @@ public function init_form_fields()
 	public function payment_fields()
 	{
 		parent::payment_fields();
-
 		if (self::WALLET === $this->source_type) {
-			Omise_Util::render_view( 'templates/payment/form-truemoney.php', array() );
+			Omise_Util::render_view( 'templates/payment/form-truemoney.php', [] );
 		}
 	}
 
@@ -118,7 +117,7 @@ public function get_charge_request($order_id, $order)
 	 * Return the right ShopeePay backend depending on the platform and availability of
 	 * the backend in the capability
 	 */
-	private function get_source()
+	public function get_source()
 	{
 		$capabilities = Omise_Capabilities::retrieve();
 

diff --git a/tests/unit/includes/gateway/class-omise-payment-truemoney-test.php b/tests/unit/includes/gateway/class-omise-payment-truemoney-test.php
@@ -4,7 +4,7 @@
 
 class Omise_Payment_Truemoney_Test extends Omise_Offsite_Test
 {
-    private $obj;
+    private $mockOmiseCapability;
 
     public function setUp(): void
     {
@@ -20,44 +20,124 @@ public function setUp(): void
 			->with('123')
 			->andReturn(true);
         require_once __DIR__ . '/../../../../includes/gateway/class-omise-payment-truemoney.php';
-        require_once __DIR__ . '/../../../../includes/class-omise-capabilities.php';
-
-        $this->obj = new Omise_Payment_Truemoney();
+        $this->mockOmiseCapability = Mockery::mock('alias:Omise_Capabilities');
     }
 
-    public function testGetChargeRequest()
+    public function test_get_charge_request()
     {
+        $this->mockOmiseCapability->shouldReceive('retrieve')->once();
         // set source type to truemoney wallet
-        $this->obj->source_type = 'truemoney';
+        $obj = new Omise_Payment_Truemoney();
+        $obj->source_type = 'truemoney';
         $orderId = 'order_123';
         $expectedAmount = 999999;
         $expectedCurrency = 'thb';
         $orderMock = $this->getOrderMock($expectedAmount, $expectedCurrency);
 
         $_POST['omise_phone_number_default'] = true;
-        $result = $this->obj->get_charge_request($orderId, $orderMock);
+        $result = $obj->get_charge_request($orderId, $orderMock);
 
         $this->assertEquals($orderMock->get_billing_phone(), $result['source']['phone_number']);
     }
 
-    public function testGetChargeRequestWhenCustomerOverridesDefaultPhone()
+    public function test_get_charge_request_when_customer_overrides_default_phone()
     {
+        $this->mockOmiseCapability->shouldReceive('retrieve')->once();
         $orderId = 'order_123';
         $expectedAmount = 999999;
         $expectedCurrency = 'thb';
         $orderMock = $this->getOrderMock($expectedAmount, $expectedCurrency);
 
-        $_POST['omise_phone_number_default'] = false;
         $_POST['omise_phone_number'] = '1234567890';
-
-        $result = $this->obj->get_charge_request($orderId, $orderMock);
+
+        $obj = new Omise_Payment_Truemoney();
+        $result = $obj->get_charge_request($orderId, $orderMock);
 
         $this->assertEquals($this->sourceType, $result['source']['type']);
     }
 
-    public function testCharge()
+    public function test_charge()
     {
+        $this->mockOmiseCapability->shouldReceive('retrieve')->once();
         $_POST['omise_phone_number_default'] = true;
-        $this->getChargeTest($this->obj);
+        $obj = new Omise_Payment_Truemoney();
+        $this->getChargeTest($obj);
+    }
+
+    public function test_get_source_returns_jumpapp()
+    {
+        $this->mockOmiseCapability->shouldReceive('retrieve');
+        $obj = new Omise_Payment_Truemoney();
+        $source_type = $obj->get_source();
+        $this->assertEquals('truemoney_jumpapp', $source_type);
+    }
+
+    public function test_get_source_returns_wallet()
+    {
+        $this->mockOmiseCapability->shouldReceive('retrieve')
+            ->andReturn(new class() {
+                public function get_truemoney_backend($source_type) {
+                    if ('truemoney' === $source_type) {
+                        return (object)[
+                            'truemoney' => [
+                                'type' => 'truemoney',
+                                'currencies' => [
+                                    'thb'
+                                ],
+                                'amount' => [
+                                    'min' => 2000,
+                                    'max' => 500000000000
+                                ]
+                            ]
+                        ];
+                    }
+
+                    return null;
+                }
+            });
+
+        $obj = new Omise_Payment_Truemoney();
+        $source_type = $obj->get_source();
+        $this->assertEquals('truemoney', $source_type);
+    }
+
+    public function test_get_source_returns_jumpapp_when_both_are_enabled()
+    {
+        $this->mockOmiseCapability->shouldReceive('retrieve')
+            ->andReturn(new class() {
+                public function get_truemoney_backend($source_type) {
+                    if ('truemoney' === $source_type) {
+                        return (object)[
+                            'truemoney' => [
+                                'type' => 'truemoney',
+                                'currencies' => [
+                                    'thb'
+                                ],
+                                'amount' => [
+                                    'min' => 2000,
+                                    'max' => 500000000000
+                                ]
+                            ]
+                        ];
+                    }
+
+                    return (object)[
+                        'truemoney_jumpapp' => [
+                            'type' => 'truemoney_jumpapp',
+                            'currencies' => [
+                                'thb'
+                            ],
+                            'amount' => [
+                                'min' => 2000,
+                                'max' => 500000000000
+                            ]
+                        ]
+                    ];
+                }
+            });
+
+        $obj = new Omise_Payment_Truemoney();
+        $source_type = $obj->get_source();
+        $this->assertEquals('truemoney_jumpapp', $source_type);
     }
 }