<?php

abstract class WP_Test_REST_TestCase extends WP_UnitTestCase {

/**

* Asserts that the REST API response has the specified error.

*

* @since 4.4.0

* @since 6.6.0 Added the `$message` parameter.

*

* @param string|int $code Expected error code.

* @param WP_REST_Response|WP_Error $response REST API response.

* @param int $status Optional. Status code.

* @param string $message Optional. Message to display when the assertion fails.

*/

protected function assertErrorResponse( $code, $response, $status = null, $message = '' ) {

if ( $response instanceof WP_REST_Response ) {

$response = $response->as_error();

}

$this->assertWPError( $response, $message . ' Passed $response is not a WP_Error object.' );

$this->assertSame( $code, $response->get_error_code(), $message . ' The expected error code does not match.' );

if ( null !== $status ) {

$data = $response->get_error_data();

$this->assertArrayHasKey( 'status', $data, $message . ' Passed $response does not include a status code.' );

$this->assertSame( $status, $data['status'], $message . ' The expected status code does not match.' );

}

}

}