Add header support for dokan test case post request. (getdokan#2348)

* Add header support for dokan test case post request. * Add header support for dokan test case get request.
mralaminahamed · Aug 27, 2024 · 8e10f32 · 8e10f32
1 parent 9d6adbf
commit 8e10f32
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/tests/php/src/DokanTestCase.php b/tests/php/src/DokanTestCase.php
@@ -204,11 +204,15 @@ protected function get_route( string $route_name ): string {
      *
      * @param string $route The REST API route.
      * @param array $params Optional query parameters.
+     * @param array $headers Optional headers.
+     * @param array $header_override Optional heaader override status.
+     *
      * @return WP_REST_Response The response object.
      */
-    protected function get_request( string $route, array $params = [] ): WP_REST_Response {
+    protected function get_request( string $route, array $params = [], array $headers = [], bool $header_override = true ): WP_REST_Response {
         $request = new WP_REST_Request( 'GET', $this->get_route( $route ) );
         $request->set_query_params( $params );
+        $request->set_headers( $headers, $header_override );
 
         return $this->server->dispatch( $request );
     }
@@ -218,11 +222,14 @@ protected function get_request( string $route, array $params = [] ): WP_REST_Res
      *
      * @param string $route The REST API route.
      * @param array $params Optional body parameters.
+     * @param array $headers Optional headers.
+     * @param array $header_override Optional heaader override status.
      * @return WP_REST_Response The response object.
      */
-    protected function post_request( string $route, array $params = [] ): WP_REST_Response {
+    protected function post_request( string $route, array $params = [], array $headers = [], bool $header_override = true ): WP_REST_Response {
         $request = new WP_REST_Request( 'POST', $this->get_route( $route ) );
         $request->set_body_params( $params );
+        $request->set_headers( $headers, $header_override );
 
         return $this->server->dispatch( $request );
     }