Changeset 26995 for trunk/tests/phpunit/tests/ajax/Autosave.php

Timestamp:

01/22/2014 04:55:37 AM (11 years ago)

Author:

azaozz

Message:

Autosave: refactor autosave.js, use heartbeat for transport and move all "Add/Edit Post" related functionality to post.js. See #25272.

File:

• trunk/tests/phpunit/tests/ajax/Autosave.php (modified) (3 diffs)

trunk/tests/phpunit/tests/ajax/Autosave.php

@@ -24 +24 @@
 
     /**
+     * user_id
+     * @var int
+     */
+    protected $user_id = 0;
+
+    /**
      * Set up the test fixture
      */
     public function setUp() {
         parent::setUp();
+        // Set a user so the $post has 'post_author'
+        $this->user_id = $this->factory->user->create( array( 'role' => 'administrator' ) );
+        wp_set_current_user( $this->user_id );
+
         $post_id = $this->factory->post->create( array( 'post_status' => 'draft' ) );
         $this->_post = get_post( $post_id );
+    }
+
+    /**
+     * Tear down the test fixture.
+     * Reset the current user
+     */
+    public function tearDown() {
+        parent::tearDown();
+        wp_set_current_user( 0 );
     }
 
@@ -37 +56 @@
      */
     public function test_autosave_post() {
-
-        // Become an admin
-        $this->_setRole( 'administrator' );
+        // The original post_author
+        wp_set_current_user( $this->user_id );
 
         // Set up the $_POST request
         $md5 = md5( uniqid() );
         $_POST = array(
-            'post_id'       => $this->_post->ID,
-            'autosavenonce' => wp_create_nonce( 'autosave' ),
-            'post_content'  => $this->_post->post_content . PHP_EOL . $md5,
-            'post_type'     => 'post',
-            'autosave'      => 1,
+            'action' => 'heartbeat',
+            '_nonce' => wp_create_nonce( 'heartbeat-nonce' ),
+            'data' => array(
+                'wp_autosave' => array(
+                    'post_id'       => $this->_post->ID,
+                    '_wpnonce'      => wp_create_nonce( 'update-post_' . $this->_post->ID ),
+                    'post_content'  => $this->_post->post_content . PHP_EOL . $md5,
+                    'post_type'     => 'post',
+                ),
+            ),
         );
 
         // Make the request
         try {
-            $this->_handleAjax( 'autosave' );
+            $this->_handleAjax( 'heartbeat' );
         } catch ( WPAjaxDieContinueException $e ) {
             unset( $e );
         }
 
-        // Get the response
-        $xml = simplexml_load_string( $this->_last_response, 'SimpleXMLElement', LIBXML_NOCDATA );
+        // Get the response, it is in heartbeat's response
+        $response = json_decode( $this->_last_response, true );
 
         // Ensure everything is correct
-        $this->assertEquals( $this->_post->ID, (int) $xml->response[0]->autosave['id'] );
-        $this->assertEquals( 'autosave_' . $this->_post->ID, (string) $xml->response['action']);
+        $this->assertNotEmpty( $response['wp_autosave'] );
+        $this->assertTrue( $response['wp_autosave']['success'] );
 
         // Check that the edit happened
-        $post = get_post( $this->_post->ID) ;
+        $post = get_post( $this->_post->ID );
         $this->assertGreaterThanOrEqual( 0, strpos( $post->post_content, $md5 ) );
+    }
+    
+    /**
+     * Test autosaving a locked post
+     * @return void
+     */
+    public function test_autosave_locked_post() {
+        // Lock the post to another user
+        $another_user_id = $this->factory->user->create( array( 'role' => 'editor' ) );
+        wp_set_current_user( $another_user_id );
+        wp_set_post_lock( $this->_post->ID );
+
+        wp_set_current_user( $this->user_id );
+
+        // Ensure post is locked
+        $this->assertEquals( $another_user_id, wp_check_post_lock( $this->_post->ID ) );
+
+        // Set up the $_POST request
+        $md5 = md5( uniqid() );
+        $_POST = array(
+            'action' => 'heartbeat',
+            '_nonce' => wp_create_nonce( 'heartbeat-nonce' ),
+            'data' => array(
+                'wp_autosave' => array(
+                    'post_id'       => $this->_post->ID,
+                    '_wpnonce'      => wp_create_nonce( 'update-post_' . $this->_post->ID ),
+                    'post_content'  => $this->_post->post_content . PHP_EOL . $md5,
+                    'post_type'     => 'post',
+                ),
+            ),
+        );
+
+        // Make the request
+        try {
+            $this->_handleAjax( 'heartbeat' );
+        } catch ( WPAjaxDieContinueException $e ) {
+            unset( $e );
+        }
+
+        $response = json_decode( $this->_last_response, true );
+
+        // Ensure everything is correct
+        $this->assertNotEmpty( $response['wp_autosave'] );
+        $this->assertTrue( $response['wp_autosave']['success'] );
+
+        // Check that the original post was NOT edited
+        $post = get_post( $this->_post->ID );
+        $this->assertFalse( strpos( $post->post_content, $md5 ) );
+
+        // Check if the autosave post was created
+        $autosave = wp_get_post_autosave( $this->_post->ID, get_current_user_id() );
+        $this->assertNotEmpty( $autosave );
+        $this->assertGreaterThanOrEqual( 0, strpos( $autosave->post_content, $md5 ) );
     }
 
@@ -76 +152 @@
     public function test_with_invalid_nonce( ) {
 
-        // Become an administrator
-        $this->_setRole( 'administrator' );
+        wp_set_current_user( $this->user_id );
 
         // Set up the $_POST request
         $_POST = array(
-            'post_id'       => $this->_post->ID,
-            'autosavenonce' => md5( uniqid() ),
-            'autosave'      => 1
+            'action' => 'heartbeat',
+            '_nonce' => wp_create_nonce( 'heartbeat-nonce' ),
+            'data' => array(
+                'wp_autosave' => array(
+                    'post_id'  => $this->_post->ID,
+                    '_wpnonce' => substr( md5( uniqid() ), 0, 10 ),
+                ),
+            ),
         );
 
         // Make the request
-        $this->setExpectedException( 'WPAjaxDieStopException', '-1' );
-        $this->_handleAjax( 'autosave' );
-    }
+        try {
+            $this->_handleAjax( 'heartbeat' );
+        } catch ( WPAjaxDieContinueException $e ) {
+            unset( $e );
+        }
 
-    /**
-     * Test with a bad post id
-     * @return void
-     */
-    public function test_with_invalid_post_id( ) {
+        $response = json_decode( $this->_last_response, true );
 
-        // Become an administrator
-        $this->_setRole( 'administrator' );
-
-        // Set up the $_POST request
-        $_POST = array(
-            'post_id'       => 0,
-            'autosavenonce' => wp_create_nonce( 'autosave' ),
-            'autosave'      => 1,
-            'post_type'     => 'post'
-        );
-
-        // Make the request
-        $this->setExpectedException( 'WPAjaxDieStopException', 'You are not allowed to edit this post.' );
-        $this->_handleAjax( 'autosave' );
+        $this->assertNotEmpty( $response['wp_autosave'] );
+        $this->assertFalse( $response['wp_autosave']['success'] );
     }
 }