Ticket #39732: 39732.patch

src/wp-admin/includes/comment.php

@@ -82 +82 @@
                 $_POST['comment_date'] = "$aa-$mm-$jj $hh:$mn:$ss";
         }
 
-        wp_update_comment( $_POST );
+        $result = wp_update_comment( $_POST );
+        
+        // Check for WP_Error
+        if ( is_wp_error( $result ) )
+                wp_die ( $result->get_error_message() );
 }
 
 /**

src/wp-includes/rest-api/endpoints/class-wp-rest-comments-controller.php

@@ -730 +730 @@
                                 return new WP_Error( 'rest_comment_failed_edit', __( 'Updating comment failed.' ), array( 'status' => 500 ) );
                         }
 
+                        // Check for WP_Error
+                        if ( is_wp_error( $updated ) ) {
+                                return $updated;
+                        }
+
                         if ( isset( $request['status'] ) ) {
                                 $this->handle_status_param( $request['status'], $id );
                         }

src/wp-includes/comment.php

@@ -2130 +2130 @@
  * @global wpdb $wpdb WordPress database abstraction object.
  *
  * @param array $commentarr Contains information on the comment.
- * @return int Comment was updated if value is 1, or was not updated if value is 0.
+ * @return int|WP_Error Comment was updated if value is 1, or was not updated if value is 0 or WP_Error.
  */
 function wp_update_comment($commentarr) {
         global $wpdb;
@@ -2194 +2194 @@
          */
         $data = apply_filters( 'wp_update_comment_data', $data, $comment, $commentarr );
 
+        // Check for WP_Error
+        if ( is_wp_error( $data ) ) {
+                return $data;
+        }
+
         $keys = array( 'comment_post_ID', 'comment_content', 'comment_author', 'comment_author_email', 'comment_approved', 'comment_karma', 'comment_author_url', 'comment_date', 'comment_date_gmt', 'comment_type', 'comment_parent', 'user_id', 'comment_agent', 'comment_author_IP' );
         $data = wp_array_slice_assoc( $data, $keys );
 

tests/phpunit/tests/comment.php

@@ -762 +762 @@
 
                 $this->assertSame( '1', $comment->comment_approved );
         }
+
+        /**
+         * @ticket 39732
+         */
+        public function test_wp_update_comment_is_wp_error () {
+                $comment_id = self::factory()->comment->create( array( 'comment_post_ID' => self::$post_id ) );
+
+                add_filter( 'wp_update_comment_data', array( $this, '_wp_update_comment_data_filter' ), 10, 3 );
+                $result = wp_update_comment( array( 'comment_ID' => $comment_id, 'comment_type' => 'pingback' ) );
+                $this->assertInstanceOf( 'WP_Error', $result );
+                remove_filter( 'wp_update_comment_data', array( $this, '_wp_update_comment_data_filter' ), 10, 3 );
+        }
+
+        /**
+         *  Block comments from being updated by returning WP_Error
+         */
+        public function _wp_update_comment_data_filter ( $data, $comment, $commentarr ) {
+                return new WP_Error( 'comment_wrong', __( 'wp_update_comment_data filter fails for this comment.' ), 500 );
+        }
 }

tests/phpunit/tests/xmlrpc/wp/editComment.php

@@ -69 +69 @@
 
                 $this->assertEquals( 'trash', get_comment( $comment_id )->comment_approved );
         }
+
+        /**
+         * @ticket 39732
+         */
+        public function test__wp_update_comment_data_filter () {
+                $author_id = $this->make_user_by_role( 'author' );
+                $post_id = self::factory()->post->create( array(
+                        'post_title' => 'Post test by author',
+                        'post_author' => $author_id
+                ) );
+
+                $comment_id = wp_insert_comment(array(
+                        'comment_post_ID' => $post_id,
+                        'comment_author' => 'Commenter 1',
+                        'comment_author_url' => "http://example.com/1/",
+                        'comment_approved' => 1,
+                ));
+
+                add_filter( 'wp_update_comment_data', array( $this, '_wp_update_comment_data_filter' ), 10, 3 );
+                $result = $this->myxmlrpcserver->wp_editComment( array( 1, 'author', 'author', $comment_id, array( 'status' => 'hold' ) ) );
+
+                $this->assertInstanceOf( 'IXR_Error', $result );
+                $this->assertEquals( 500, $result->code );
+                $this->assertEquals( __( 'wp_update_comment_data filter fails for this comment.' ), $result->message ); 
+        }
+
+        /**
+         *  Block comments from being updated by returning WP_Error
+         */
+        public function _wp_update_comment_data_filter ( $data, $comment, $commentarr ) {
+                return new WP_Error( 'comment_wrong', __( 'wp_update_comment_data filter fails for this comment.' ), 500 );
+        }
 }

tests/phpunit/tests/ajax/EditComment.php

@@ -191 +191 @@
                 $this->setExpectedException( 'WPAjaxDieStopException', '-1' );
                 $this->_handleAjax( 'edit-comment' );
         }
+
+        /**
+         * @ticket 39732
+         */
+        public function test_wp_update_comment_data_is_wp_error () {
+                
+                // Become an administrator
+                $this->_setRole( 'administrator' );
+
+                // Get a comment
+                $comments = get_comments( array(
+                        'post_id' => $this->_comment_post->ID
+                ) );
+                $comment = array_pop( $comments );
+
+                // Set up a default request
+                $_POST['_ajax_nonce-replyto-comment'] = wp_create_nonce( 'replyto-comment' );
+                $_POST['comment_ID']                  = $comment->comment_ID;
+                $_POST['content']                     = 'Lorem ipsum dolor sit amet, consectetur adipiscing elit.';
+
+                // Simulate filter check error
+                add_filter( 'wp_update_comment_data', array( $this, '_wp_update_comment_data_filter' ), 10, 3 );
+
+                // Make the request
+                $this->setExpectedException( 'WPAjaxDieStopException', 'wp_update_comment_data filter fails for this comment.' );
+                $this->_handleAjax( 'edit-comment' );
+                
+                remove_filter( 'wp_update_comment_data', array( $this, '_wp_update_comment_data_filter' ), 10, 3 );
+                
+        }
+        
+        /**
+         *  Block comments from being updated by returning WP_Error
+         */
+        public function _wp_update_comment_data_filter ( $data, $comment, $commentarr ) {
+                return new WP_Error( 'comment_wrong', __( 'wp_update_comment_data filter fails for this comment.' ), 500 );
+        }
 }

tests/phpunit/tests/rest-api/rest-comments-controller.php

@@ -2403 +2403 @@
                 $this->assertErrorResponse( 'comment_content_column_length', $response, 400 );
         }
 
+        /**
+         * @ticket 39732
+         */
+        public function test_update_comment_is_wp_error() {
+                wp_set_current_user( self::$admin_id );
+
+                $params = array(
+                        'content'     => 'This isn\'t a saxophone. It\'s an umbrella.',
+                );
+
+                add_filter( 'wp_update_comment_data', array( $this, '_wp_update_comment_data_filter' ), 10, 3 );
+
+                $request = new WP_REST_Request( 'PUT', sprintf( '/wp/v2/comments/%d', self::$approved_id ) );
+
+                $request->add_header( 'content-type', 'application/json' );
+                $request->set_body( wp_json_encode( $params ) );
+                $response = $this->server->dispatch( $request );
+
+                $this->assertErrorResponse( 'comment_wrong', $response, 500 );
+        }
+
+        /**
+         *  Block comments from being updated by returning WP_Error
+         */
+        public function _wp_update_comment_data_filter ( $data, $comment, $commentarr ) {
+                return new WP_Error( 'comment_wrong', __( 'wp_update_comment_data filter fails for this comment.' ), array( 'status' => 500 ) );
+        }
+
         public function verify_comment_roundtrip( $input = array(), $expected_output = array() ) {
                 // Create the comment
                 $request = new WP_REST_Request( 'POST', '/wp/v2/comments' );