chore: Add tests

Author: Nicholas Roscino

tests/formatters/networkFormatter.test.ts

@@ -7,8 +7,12 @@
 import assert from 'node:assert';
 import {describe, it} from 'node:test';
 
+import {ProtocolError} from 'puppeteer-core';
+
 import {
   getFormattedHeaderValue,
+  getFormattedRequestBody,
+  getFormattedResponseBody,
   getShortDescriptionForRequest,
 } from '../../src/formatters/networkFormatter.js';
 import {getMockRequest, getMockResponse} from '../utils.js';
@@ -34,7 +38,6 @@ describe('networkFormatter', () => {
 
       assert.equal(result, 'http://example.com GET [success - 200]');
     });
-
     it('shows correct status for request with response code in 100', async () => {
       const response = getMockResponse({
         status: 199,
@@ -53,7 +56,6 @@ describe('networkFormatter', () => {
 
       assert.equal(result, 'http://example.com GET [failed - 300]');
     });
-
     it('shows correct status for request that failed', async () => {
       const request = getMockRequest({
         failure() {
@@ -100,4 +102,137 @@ describe('networkFormatter', () => {
       assert.deepEqual(result, []);
     });
   });
+
+  describe('getFormattedRequestBody', () => {
+    it('shows data from fetchPostData if postData is undefined', async () => {
+      const request = getMockRequest({
+        hasPostData: true,
+        postData: undefined,
+        fetchPostData: Promise.resolve('test'),
+      });
+
+      const result = await getFormattedRequestBody('body', request, 200);
+
+      assert.strictEqual(result, 'body\ntest');
+    });
+    it('shows empty string when no postData available', async () => {
+      const request = getMockRequest({
+        hasPostData: false,
+      });
+
+      const result = await getFormattedRequestBody('body', request, 200);
+
+      assert.strictEqual(result, '');
+    });
+    it('shows request body when postData is available', async () => {
+      const request = getMockRequest({
+        postData: JSON.stringify({
+          request: 'body',
+        }),
+        hasPostData: true,
+      });
+
+      const result = await getFormattedRequestBody('body', request, 200);
+
+      assert.strictEqual(
+        result,
+        `body\n${JSON.stringify({
+          request: 'body',
+        })}`,
+      );
+    });
+    it('shows trunkated string correctly with postData', async () => {
+      const response = getMockRequest({
+        postData: 'some text that is longer than expected',
+        hasPostData: true,
+      });
+
+      const result = await getFormattedRequestBody('body', response, 20);
+
+      assert.strictEqual(result, 'body\nsome text that is lo... <truncated>');
+    });
+    it('shows trunkated string correctly with fetchPostData', async () => {
+      const response = getMockRequest({
+        fetchPostData: Promise.resolve(
+          'some text that is longer than expected',
+        ),
+        postData: undefined,
+        hasPostData: true,
+      });
+
+      const result = await getFormattedRequestBody('body', response, 20);
+
+      assert.strictEqual(result, 'body\nsome text that is lo... <truncated>');
+    });
+    it('shows nothing on exception', async () => {
+      const request = getMockRequest({
+        hasPostData: true,
+        postData: undefined,
+        fetchPostData: Promise.reject(new ProtocolError()),
+      });
+
+      const result = await getFormattedRequestBody('body', request, 200);
+
+      assert.strictEqual(result, '');
+    });
+  });
+
+  describe('getFormattedResponseBody', () => {
+    it('handles empty buffer correctly', async () => {
+      const response = getMockResponse();
+      response.buffer = () => {
+        return Promise.resolve(Buffer.from(''));
+      };
+
+      const result = await getFormattedResponseBody('body', response, 20);
+
+      assert.strictEqual(result, 'body\n<empty response>');
+    });
+    it('handles base64 text correctly', async () => {
+      const binaryBuffer = Buffer.from([
+        0xde, 0xad, 0xbe, 0xef, 0x00, 0x41, 0x42, 0x43,
+      ]);
+      const response = getMockResponse();
+      response.buffer = () => {
+        return Promise.resolve(binaryBuffer);
+      };
+
+      const result = await getFormattedResponseBody('body', response, 20);
+
+      assert.strictEqual(result, 'body\n<binary data>');
+    });
+    it('handles the text limit correctly', async () => {
+      const response = getMockResponse();
+      response.buffer = () => {
+        return Promise.resolve(
+          Buffer.from('some text that is longer than expected'),
+        );
+      };
+
+      const result = await getFormattedResponseBody('body', response, 20);
+
+      assert.strictEqual(result, 'body\nsome text that is lo... <truncated>');
+    });
+    it('handles the text format correctly', async () => {
+      const response = getMockResponse();
+      response.buffer = () => {
+        return Promise.resolve(Buffer.from(JSON.stringify({response: 'body'})));
+      };
+
+      const result = await getFormattedResponseBody('body', response, 200);
+
+      assert.strictEqual(result, `body\n${JSON.stringify({response: 'body'})}`);
+    });
+    it('handles error correctly', async () => {
+      const response = getMockResponse();
+      response.buffer = () => {
+        // CDP Error simulation
+        return Promise.reject(new ProtocolError());
+      };
+
+      const result = await getFormattedResponseBody('body', response, 200);
+
+      assert.strictEqual(result, '');
+    });
+  });
 });

tests/utils.ts

@@ -46,6 +46,9 @@ export function getMockRequest(
     response?: HTTPResponse;
     failure?: HTTPRequest['failure'];
     resourceType?: string;
+    hasPostData?: boolean;
+    postData?: string;
+    fetchPostData?: Promise<string>;
   } = {},
 ): HTTPRequest {
   return {
@@ -55,6 +58,15 @@ export function getMockRequest(
     method() {
       return options.method ?? 'GET';
     },
+    fetchPostData() {
+      return options.fetchPostData ?? Promise.reject();
+    },
+    hasPostData() {
+      return options.hasPostData ?? false;
+    },
+    postData() {
+      return options.postData;
+    },
     response() {
       return options.response ?? null;
     },