71 lines
2.0 KiB
PHP
71 lines
2.0 KiB
PHP
<?php
|
|
|
|
class Spy_REST_Server extends WP_REST_Server {
|
|
|
|
public $sent_headers = array();
|
|
public $sent_body = '';
|
|
public $last_request = null;
|
|
public $override_by_default = false;
|
|
|
|
/**
|
|
* Get the raw $endpoints data from the server
|
|
*
|
|
* @return array
|
|
*/
|
|
public function get_raw_endpoint_data() {
|
|
return $this->endpoints;
|
|
}
|
|
|
|
/**
|
|
* Allow calling protected methods from tests
|
|
*
|
|
* @param string $method Method to call
|
|
* @param array $args Arguments to pass to the method
|
|
* @return mixed
|
|
*/
|
|
public function __call( $method, $args ) {
|
|
return call_user_func_array( array( $this, $method ), $args );
|
|
}
|
|
|
|
public function send_header( $header, $value ) {
|
|
$this->sent_headers[ $header ] = $value;
|
|
}
|
|
|
|
public function remove_header( $header ) {
|
|
unset( $this->sent_headers[ $header ] );
|
|
}
|
|
|
|
/**
|
|
* Override the dispatch method so we can get a handle on the request object.
|
|
*
|
|
* @param WP_REST_Request $request
|
|
* @return WP_REST_Response Response returned by the callback.
|
|
*/
|
|
public function dispatch( $request ) {
|
|
$this->last_request = $request;
|
|
return parent::dispatch( $request );
|
|
}
|
|
|
|
/**
|
|
* Override the register_route method so we can re-register routes internally if needed.
|
|
*
|
|
* @param string $namespace Namespace.
|
|
* @param string $route The REST route.
|
|
* @param array $route_args Route arguments.
|
|
* @param bool $override Optional. Whether the route should be overridden if it already exists.
|
|
* Default false. Also set $GLOBALS['wp_rest_server']->override_by_default = true
|
|
* to set overrides when you don't have access to the caller context.
|
|
*/
|
|
public function register_route( $namespace, $route, $route_args, $override = false ) {
|
|
parent::register_route( $namespace, $route, $route_args, $override || $this->override_by_default );
|
|
}
|
|
|
|
public function serve_request( $path = null ) {
|
|
|
|
ob_start();
|
|
$result = parent::serve_request( $path );
|
|
$this->sent_body = ob_get_clean();
|
|
return $result;
|
|
}
|
|
}
|