Prevent fatal error in set_current_gateway with no WC session

hjoelr · vedanshujain · commit d6771dc681cf · 2020-01-13T14:18:49.000+05:30
diff --git a/includes/class-wc-payment-gateways.php b/includes/class-wc-payment-gateways.php
@@ -183,10 +183,13 @@ public function set_current_gateway( $gateways ) {
 		}
 
 		$current_gateway = false;
-		$current         = WC()->session->get( 'chosen_payment_method' );
 
-		if ( $current && isset( $gateways[ $current ] ) ) {
-			$current_gateway = $gateways[ $current ];
+		if ( WC()->session ) {
+			$current = WC()->session->get( 'chosen_payment_method' );
+
+			if ( $current && isset( $gateways[ $current ] ) ) {
+				$current_gateway = $gateways[ $current ];
+			}
 		}
 
 		if ( ! $current_gateway ) {

Original file line number	Diff line number	Diff line change
`@@ -183,10 +183,13 @@ public function set_current_gateway( $gateways ) {`
`183`	`183`	`}`
`184`	`184`
`185`	`185`	`$current_gateway = false;`
`186`		`- $current = WC()->session->get( 'chosen_payment_method' );`
`187`	`186`
`188`		`- if ( $current && isset( $gateways[ $current ] ) ) {`
`189`		`- $current_gateway = $gateways[ $current ];`
	`187`	`+ if ( WC()->session ) {`
	`188`	`+ $current = WC()->session->get( 'chosen_payment_method' );`
	`189`	`+`
	`190`	`+ if ( $current && isset( $gateways[ $current ] ) ) {`
	`191`	`+ $current_gateway = $gateways[ $current ];`
	`192`	`+ }`
`190`	`193`	`}`
`191`	`194`
`192`	`195`	`if ( ! $current_gateway ) {`