Removed test code. Updated tests.

Removed test code.
Updated test to do the action rest_api_init and then check if post and page are registered for preview links actions.

Author: colinmurphy

plugins/faustwp/includes/replacement/callbacks.php

@@ -338,27 +338,6 @@ function register_preview_link_hooks_for_all_draft_post_types() {
 	}
 }
 
-add_filter( 'rest_post_dispatch', __NAMESPACE__ . '\\rest_post_dispatch', 10, 3 );
-
-/**
- * Adds the preview link to rest responses.
- *
- * @param WP_HTTP_Response $response The rest response object.
- * @param WP_REST_Server   $post Post object.
- * @param WP_REST_Request  $request The request object.
- *
- * @return WP_REST_Response The rest response object.
- */
-function rest_post_dispatch( $response, $post, $request ) {
-
-	if ( isset( $post->post_status ) && 'draft' === $post->post_status ) {
-		$response->data['link'] = get_preview_post_link( $post->ID );
-	}
-
-	return $response;
-}
-
-
 /**
  * Adds the preview link to rest responses.
  *

plugins/faustwp/tests/integration/ReplacementCallbacksTests.php

@@ -48,11 +48,10 @@ public function test_preview_post_link_filter() {
 		$this->assertSame( 1000, has_action( 'preview_post_link', 'WPE\FaustWP\Replacement\post_preview_link' ) );
 	}
 
-	public function test_preview_rest_prepare_post_filter() {
+	public function test_rest_api_init_filter() {
+		$this->assertSame( 10, has_action( 'rest_api_init', 'WPE\FaustWP\Replacement\register_preview_link_hooks_for_all_draft_post_types' ) );
+		do_action( 'rest_api_init' );
 		$this->assertSame( 10, has_action( 'rest_prepare_post', 'WPE\FaustWP\Replacement\preview_link_in_rest_response' ) );
-	}
-
-	public function test_preview_rest_prepare_page_filter() {
 		$this->assertSame( 10, has_action( 'rest_prepare_page', 'WPE\FaustWP\Replacement\preview_link_in_rest_response' ) );
 	}