export function 'isRedirectStatusCode' (#2810)

Author: miherlosev

src/index.ts

@@ -28,7 +28,7 @@ import RequestHookEventProvider from './request-pipeline/request-hooks/events/ev
 import { RequestInfo, ResponseInfo } from './request-pipeline/request-hooks/events/info';
 import BaseRequestHookEventFactory from './request-pipeline/request-hooks/events/factory/base';
 import BaseRequestPipelineContext from './request-pipeline/context/base';
-
+import isRedirectStatusCode from './utils/is-redirect-status-code';
 
 export default {
     Proxy,
@@ -61,4 +61,5 @@ export default {
     acceptCrossOrigin,
     getAssetPath,
     ResponseInfo,
+    isRedirectStatusCode,
 };

src/request-pipeline/context/index.ts

@@ -35,6 +35,7 @@ import RequestHookEventProvider from '../request-hooks/events/event-provider';
 import { PassThrough } from 'stream';
 import promisifyStream from '../../utils/promisify-stream';
 import { toReadableStream } from '../../utils/buffer';
+import isRedirectStatusCode from '../../utils/is-redirect-status-code';
 
 export interface DestInfo {
     url: string;
@@ -93,7 +94,6 @@ interface Socket extends net.Socket {
 
 export type DestinationResponse = IncomingMessage | FileStream | IncomingMessageLike | Http2Response;
 
-const REDIRECT_STATUS_CODES                  = [301, 302, 303, 307, 308];
 const CANNOT_BE_USED_WITH_WEB_SOCKET_ERR_MSG = 'The function cannot be used with a WebSocket request.';
 
 export default class RequestPipelineContext extends BaseRequestPipelineContext {
@@ -302,7 +302,7 @@ export default class RequestPipelineContext extends BaseRequestPipelineContext {
 
         const isRedirect              = this.destRes.headers[BUILTIN_HEADERS.location] &&
                                         this.destRes.statusCode &&
-                                        REDIRECT_STATUS_CODES.includes(this.destRes.statusCode) ||
+                                        isRedirectStatusCode(this.destRes.statusCode) ||
                                         false;
         const requireAssetsProcessing = (isCSS || isScript || isManifest) && this.destRes.statusCode !== 204;
         const isNotModified           = this.req.method === 'GET' && this.destRes.statusCode === 304 &&

src/utils/is-redirect-status-code.ts

@@ -0,0 +1,5 @@
+const REDIRECT_STATUS_CODES = [301, 302, 303, 307, 308];
+
+export default function (code: number): boolean {
+    return REDIRECT_STATUS_CODES.includes(code);
+}

ts-defs/index.d.ts

@@ -591,6 +591,9 @@ declare module 'testcafe-hammerhead' {
     /** Calculates the asset path depending on the run mode (production or development) **/
     function getAssetPath(originPath: string, developmentMode: boolean): string;
 
+    /** Return whether the HTTP status code is a redirect status code **/
+    function isRedirectStatusCode (code: number): boolean;
+
     /**  **/
     export class IncomingMessageLike {
         /** The headers of the instance **/