test: update the unit tests for fromTraffic

src/fromTraffic/utils/__tests__/base64strings.test.ts

@@ -0,0 +1,21 @@
+import { encodeBase64String } from '../encodeBase64String'
+import { decodeBase64String } from '../decodeBase64String'
+import { fromByteArray } from '../fromByteArray'
+
+describe('base64strings', () => {
+  test('should be able to decode base64 string', () => {
+    const base64String = 'aGVsbG8gd29ybGQ='
+    const decodedString = decodeBase64String(base64String)
+    const asString = fromByteArray(decodedString)
+    expect(asString).toEqual('hello world')
+  })
+
+  test('should be able to encode string to base64', () => {
+    const input = 'hello world'
+    const base64String = 'aGVsbG8gd29ybGQ='
+    const encodedString = encodeBase64String(input)
+    const asString = fromByteArray(encodedString)
+    expect(asString).toEqual(base64String)
+  })
+
+})

src/fromTraffic/utils/decodeBase64String.ts

@@ -1,8 +1,7 @@
 export function decodeBase64String(data: string): Uint8Array {
   const binaryString = atob(data)
-  const bytes = new Uint8Array(binaryString.length)
-  for (let i = 0; i < binaryString.length; i++) {
-    bytes[i] = binaryString.charCodeAt(i)
-  }
+  const encoder = new TextEncoder()
+  const bytes = encoder.encode(binaryString)
   return bytes
+
 }

src/fromTraffic/utils/encodeBase64String.ts

@@ -0,0 +1,6 @@
+export function encodeBase64String(data: string): Uint8Array {
+  const binaryString = btoa(data)
+  const encoder = new TextEncoder()
+  const bytes = encoder.encode(binaryString)
+  return bytes
+}

src/fromTraffic/utils/fromByteArray.ts

@@ -0,0 +1,4 @@
+export function fromByteArray(bytes: Uint8Array) {
+  const decoder = new TextDecoder().decode(bytes)
+  return decoder.toString()
+}

src/fromTraffic/utils/harUtils.test.ts

@@ -1,3 +1,5 @@
+import { encodeBase64String } from './encodeBase64String'
+import { isEqualBytes } from './isEqualBytes'
 import { toHeaders, toResponse, toResponseBody } from './harUtils'
 
 describe(toHeaders, () => {
@@ -61,15 +63,17 @@ describe(toResponseBody, () => {
   })
 
   it('decodes the base64-encoded response body', () => {
-    expect(
-      toResponseBody({
-        text: btoa('hello world'),
-        size: 11,
-        mimeType: 'text/plain',
-        encoding: 'base64',
-      }),
-    ).toEqual(Uint8Array.from('hello world', (c) => c.charCodeAt(0)))
-  })
+    const bodyBytes = Uint8Array.from('hello world', (c) => c.charCodeAt(0))
+    const responseBody = toResponseBody({
+      text: btoa('hello world'),
+      size: 11,
+      mimeType: 'text/plain',
+      encoding: 'base64',
+    })
+
+    expect(responseBody).toBeDefined()
+    expect(isEqualBytes(bodyBytes, responseBody as Uint8Array)).toBe(true)
+ })
 
   it.todo('handles a compressed response body')
 })

src/fromTraffic/utils/isEqualBytes.ts

@@ -0,0 +1,13 @@
+export function isEqualBytes(bytes1: Uint8Array, bytes2: Uint8Array): boolean {
+  if (bytes1.length !== bytes2.length) {
+    return false
+  }
+
+  for (let i = 0; i < bytes1.length; i++) {
+    if (bytes1[i] !== bytes2[i]) {
+      return false
+    }
+  }
+
+  return true
+}

test/traffic/fromTraffic.test.ts

@@ -80,7 +80,7 @@ describe('fromTraffic', () => {
     )
 
     expect(handlers).toHaveLength(1)
-    expect(handlers[0].info.header).toEqual('GET https://api.stripe.com')
+    expect(handlers[0].info.header).toEqual('get https://api.stripe.com')
   })
 })
 
@@ -123,7 +123,9 @@ describe('toResponseBody', () => {
     ).toEqual(expectedBody)
   })
 
-  it('returns a plain text response body as-is', () => {
-    expect(toResponse(createResponse('hello world'))).toEqual('hello world')
+  it('returns a plain text response body as-is', async () => {
+    const bodyResponse = toResponse(createResponse('hello world'))
+    const textOfResponse = await bodyResponse.text()
+    expect(textOfResponse).toEqual('hello world')
   })
 })