more constants and test fix

Automattic · Nov 8, 2024 · a1921e8 · a1921e8
1 parent 4ae3dbb
commit a1921e8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/includes/constants/class-country-test-cards.php b/includes/constants/class-country-test-cards.php
@@ -90,6 +90,6 @@ class Country_Test_Cards extends Base_Constant {
 	 * @return string Test card number
 	 */
 	public static function get_test_card_for_country( string $country_code ) {
-		return self::$country_test_cards[ $country_code ] ?? self::$country_test_cards['US'];
+		return self::$country_test_cards[ $country_code ] ?? self::$country_test_cards[ Country_Code::UNITED_STATES ];
 	}
 }
diff --git a/tests/unit/test-class-wc-payments-checkout.php b/tests/unit/test-class-wc-payments-checkout.php
@@ -25,7 +25,7 @@
  *
  * @package WooCommerce\Payments\Tests
  */
-class WC_Payments_Checkout_Test extends WP_UnitTestCase {
+class Timur_Test extends WP_UnitTestCase {
 
 	/**
 	 * Holds the object, which will be tested.
@@ -404,6 +404,10 @@ public function test_link_payment_method_provided_when_card_enabled() {
 		 * @dataProvider non_reusable_payment_method_provider
 		 */
 	public function test_no_save_option_for_non_reusable_payment_method( $payment_method_id, $payment_method_class ) {
+		$this->mock_wcpay_account
+			->method( 'get_account_country' )
+			->willReturn( 'US' );
+
 		$this->mock_wcpay_gateway
 			->expects( $this->any() )
 			->method( 'get_payment_method_ids_enabled_at_checkout' )