make axios config method on testEnv class

Author: AbhiPrasad

packages/node-integration-tests/utils/index.ts

@@ -121,6 +121,8 @@ async function makeRequest(
 }
 
 export class TestEnv {
+  private _axiosConfig: AxiosRequestConfig = undefined;
+
   public constructor(public readonly server: http.Server, public readonly url: string) {
     this.server = server;
     this.url = url;
@@ -165,10 +167,7 @@ export class TestEnv {
    * @param {DataCollectorOptions} options
    * @returns The intercepted envelopes.
    */
-  public async getMultipleEnvelopeRequest(
-    options: DataCollectorOptions,
-    axiosConfig?: AxiosRequestConfig,
-  ): Promise<Record<string, unknown>[][]> {
+  public async getMultipleEnvelopeRequest(options: DataCollectorOptions): Promise<Record<string, unknown>[][]> {
     const envelopeTypeArray =
       typeof options.envelopeType === 'string'
         ? [options.envelopeType]
@@ -180,7 +179,7 @@ export class TestEnv {
       envelopeTypeArray,
     );
 
-    void makeRequest(options.method, options.url || this.url, axiosConfig);
+    void makeRequest(options.method, options.url || this.url, this._axiosConfig);
     return resProm;
   }
 
@@ -190,11 +189,8 @@ export class TestEnv {
    * @param {DataCollectorOptions} options
    * @returns The extracted envelope.
    */
-  public async getEnvelopeRequest(
-    options?: DataCollectorOptions,
-    axiosConfig?: AxiosRequestConfig,
-  ): Promise<Array<Record<string, unknown>>> {
-    return (await this.getMultipleEnvelopeRequest({ ...options, count: 1 }, axiosConfig))[0];
+  public async getEnvelopeRequest(options?: DataCollectorOptions): Promise<Array<Record<string, unknown>>> {
+    return (await this.getMultipleEnvelopeRequest({ ...options, count: 1 }))[0];
   }
 
   /**
@@ -256,4 +252,8 @@ export class TestEnv {
         .reply(200);
     });
   }
+
+  public setAxiosConfig(axiosConfig: AxiosRequestConfig): void {
+    this._axiosConfig = axiosConfig;
+  }
 }

packages/remix/test/integration/test/server/loader.test.ts

@@ -162,15 +162,14 @@ describe.each(['builtin', 'express'])('Remix API Loaders with adapter = %s', ada
     const env = await RemixTestEnv.init(adapter);
     const url = `${env.url}/loader-json-response/3`;
 
-    const envelope = await env.getEnvelopeRequest(
-      { url, envelopeType: 'transaction' },
-      {
-        headers: {
-          'sentry-trace': '12312012123120121231201212312012-1121201211212012-1',
-          baggage: 'sentry-version=1.0,sentry-environment=production,sentry-trace_id=12312012123120121231201212312012',
-        },
+    // send sentry-trace and baggage headers to loader
+    env.setAxiosConfig({
+      headers: {
+        'sentry-trace': '12312012123120121231201212312012-1121201211212012-1',
+        baggage: 'sentry-version=1.0,sentry-environment=production,sentry-trace_id=12312012123120121231201212312012',
       },
-    );
+    });
+    const envelope = await env.getEnvelopeRequest({ url, envelopeType: 'transaction' });
 
     expect(envelope[0].trace).toMatchObject({
       trace_id: '12312012123120121231201212312012',