Fix php unit tests

Author: Ferdev

classes/class-wc-connect-nux.php

@@ -291,6 +291,17 @@ public static function get_banner_type_to_display( $status = array() ) {
 						$status['should_display_contextual_banner'] = true;
 					}
 
+					// Fallback for non-US stores if contextual banner flag is not set
+					if ( isset( $status['tos_accepted'] ) && $status['tos_accepted'] &&
+						( ! isset( $status['should_display_after_cxn_banner'] ) || ! $status['should_display_after_cxn_banner'] ) &&
+						( ! isset( $status['should_display_contextual_banner'] ) || ! $status['should_display_contextual_banner'] )
+					) {
+						$is_us_store = ( isset( $status['store_country'] ) && 'US' === $status['store_country'] );
+						if ( ! $is_us_store ) {
+							return 'after_cxn_non_us';
+						}
+					}
+
 					// Priority 3: Contextual banners (if standard "after connection" is done or was not applicable,
 					// TOS is accepted, and the contextual flag is set - either previously or by the block above).
 					if ( isset( $status['should_display_contextual_banner'] ) && $status['should_display_contextual_banner'] ) {

classes/class-wc-connect-options.php

@@ -51,6 +51,7 @@ public static function get_option_names( $type = 'compact' ) {
 				'predefined_packages',
 				'shipping_methods_migrated',
 				'should_display_nux_after_jp_cxn_banner',
+				'should_display_nux_contextual_banner',
 				'needs_tax_environment_setup',
 				'banner_ppec',
 			);