Check wp_insert_user for WP_Error

minidora · Feb 3, 2013 · 5b80c94 · 5b80c94
1 parent 3680be6
commit 5b80c94
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/classes/class-wc-checkout.php b/classes/class-wc-checkout.php
@@ -620,8 +620,9 @@ public function process_checkout() {
 
 		                $this->customer_id = wp_insert_user( apply_filters( 'woocommerce_new_customer_data', $new_customer_data ) );
 
-		                if ( ! $this->customer_id )
+		                if ( is_wp_error( $this->customer_id ) ) {
 		                	throw new MyException( '<strong>' . __( 'ERROR', 'woocommerce' ) . '</strong>: ' . __( 'Couldn&#8217;t register you&hellip; please contact us if you continue to have problems.', 'woocommerce' ) );
+						}
 
                         // Set the global user object
                         $current_user = get_user_by ( 'id', $this->customer_id );

diff --git a/woocommerce-functions.php b/woocommerce-functions.php
@@ -748,7 +748,7 @@ function woocommerce_process_registration() {
 
             $user_id = wp_insert_user( apply_filters( 'woocommerce_new_customer_data', $new_customer_data ) );
 
-            if ( ! $user_id ) {
+            if ( is_wp_error($user_id) ) {
             	$woocommerce->add_error( '<strong>' . __( 'ERROR', 'woocommerce' ) . '</strong>: ' . __( 'Couldn&#8217;t register you&hellip; please contact us if you continue to have problems.', 'woocommerce' ) );
                 return;
             }