Ticket #38817: 38817.2.diff

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

@@ -1041 +1041 @@
                         $prepared_comment['comment_author_IP'] = $request['author_ip'];
                 }
 
-                if ( isset( $request['author_user_agent'] ) ) {
+                if ( ! empty( $request['author_user_agent'] ) ) {
                         $prepared_comment['comment_agent'] = $request['author_user_agent'];
+                } elseif ( $request->get_header( 'user_agent' ) ) {
+                        $prepared_comment['comment_agent'] = $request->get_header( 'user_agent' );
                 }
 
                 if ( isset( $request['type'] ) ) {

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

@@ -1246 +1246 @@
                 $this->assertEquals( 'Mozilla/5.0 (Windows NT 6.1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/41.0.2228.0 Safari/537.36', $data['author_user_agent'] );
         }
 
+        public function test_create_comment_user_agent_header() {
+                wp_set_current_user( self::$admin_id );
+
+                $params = array(
+                        'post'         => self::$post_id,
+                        'author_name'  => 'Homer Jay Simpson',
+                        'author_email' => 'chunkylover53@aol.com',
+                        'author_url'   => 'http://compuglobalhypermeganet.com',
+                        'content'      => 'Here\’s to alcohol: the cause of, and solution to, all of life\’s problems.',
+                );
+
+                $request = new WP_REST_Request( 'POST', '/wp/v2/comments' );
+                $request->add_header( 'content-type', 'application/json' );
+                $request->add_header( 'user_agent', 'Mozilla/4.0 (compatible; MSIE 5.5; AOL 4.0; Windows 95)' );
+                $request->set_body( wp_json_encode( $params ) );
+
+                $response = $this->server->dispatch( $request );
+                $this->assertEquals( 201, $response->get_status() );
+
+                $data = $response->get_data();
+
+                $new_comment = get_comment( $data['id'] );
+                $this->assertEquals( 'Mozilla/4.0 (compatible; MSIE 5.5; AOL 4.0; Windows 95)', $new_comment->comment_agent );
+        }
+
         public function test_create_comment_invalid_author_IP() {
                 wp_set_current_user( self::$admin_id );