Fix test

omise · May 27, 2024 · f42ee9e · f42ee9e
1 parent b6d3d07
commit f42ee9e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/includes/gateway/class-omise-payment-truemoney.php b/includes/gateway/class-omise-payment-truemoney.php
@@ -33,6 +33,7 @@ public function __construct()
 		$this->title                = $this->get_option( 'title' );
 		$this->description          = $this->get_option( 'description' );
 		$this->restricted_countries = array( 'TH' );
+		$this->source_type          = null;
 
 		add_action( 'woocommerce_update_options_payment_gateways_' . $this->id, array( $this, 'process_admin_options' ) );
 		add_action( 'woocommerce_api_' . $this->id . '_callback', 'Omise_Callback::execute' );
@@ -85,7 +86,6 @@ public function is_wallet()
 		// the Capability API is not being called from WC blocks.
 		if (!$this->source_type) {
 			$this->source_type = $this->get_source();
-
 		}
 
 		return self::WALLET === $this->source_type;

diff --git a/tests/unit/includes/gateway/class-omise-payment-truemoney-test.php b/tests/unit/includes/gateway/class-omise-payment-truemoney-test.php
@@ -25,7 +25,6 @@ public function setUp(): void
 
     public function test_get_charge_request()
     {
-        $this->omise_capability_mock->shouldReceive('retrieve')->once();
         // set source type to truemoney wallet
         $obj = new Omise_Payment_Truemoney();
         $obj->source_type = 'truemoney';