feat: add blob arrayBuf and queryString tests

Author: egor-romanov

test/functions/mirror/index.ts

@@ -2,21 +2,35 @@ import { serve } from 'https://deno.land/std/http/server.ts'
 
 serve(async (request: Request) => {
   let body
-  switch (request.headers.get('content-type')) {
-    case 'application/json': {
+  let contentType = 'application/json'
+  switch (request.headers.get('response-type')) {
+    case 'json': {
       body = await request.json()
       break
     }
-    case 'application/x-www-form-urlencoded': {
+    case 'form': {
       const formBody = await request.formData()
       body = []
       for (const e of formBody.entries()) {
         body.push(e)
       }
       break
     }
+    case 'blob': {
+      const data = await request.blob()
+      body = await data.text()
+      contentType = 'application/octet-stream'
+      break
+    }
+    case 'arrayBuffer': {
+      const data = await request.arrayBuffer()
+      body = new TextDecoder().decode(data || new Uint8Array())
+      contentType = 'application/octet-stream'
+      break
+    }
     default: {
       body = await request.text()
+      contentType = 'text/plain'
       break
     }
   }
@@ -31,10 +45,16 @@ serve(async (request: Request) => {
     body: body ?? 'empty',
   }
 
-  return new Response(JSON.stringify(resp), {
+  let responseData
+  if (request.headers.get('response-type') === 'blob') {
+    responseData = new Blob([JSON.stringify(resp)], { type: 'application/json' })
+  } else {
+    responseData = JSON.stringify(resp)
+  }
+  return new Response(responseData, {
     status: 200,
     headers: {
-      'content-type': 'application/json',
+      'content-type': contentType,
     },
   })
 })

test/spec/params.spec.ts

@@ -1,3 +1,6 @@
+import { Blob } from 'buffer'
+import querystring from 'querystring'
+
 import 'jest'
 import { nanoid } from 'nanoid'
 import { sign } from 'jsonwebtoken'
@@ -7,6 +10,7 @@ import { FunctionsClient } from '../../src/index'
 
 import { Relay, runRelay } from '../relay/container'
 import { attach, log } from '../utils/jest-custom-reporter'
+import { str2ab } from '../utils/binaries'
 import { MirrorResponse } from '../models/mirrorResponse'
 
 describe('params reached to function', () => {
@@ -158,6 +162,7 @@ describe('params reached to function', () => {
       body: form,
       headers: {
         'content-type': 'application/x-www-form-urlencoded',
+        'response-type': 'form',
       },
     })
 
@@ -204,6 +209,7 @@ describe('params reached to function', () => {
       body: JSON.stringify(body),
       headers: {
         'content-type': 'application/json',
+        'response-type': 'json',
       },
     })
 
@@ -223,4 +229,138 @@ describe('params reached to function', () => {
     )
     expect(data).toEqual(expected)
   })
+
+  test('invoke mirror with body arrayBuffer', async () => {
+    /**
+     * @feature body
+     */
+    log('create FunctionsClient')
+    const fclient = new FunctionsClient(`http://localhost:${relay.container.getMappedPort(8081)}`)
+    attach('setAuth', apiKey, ContentType.TEXT)
+    fclient.setAuth(apiKey)
+
+    log('invoke mirror')
+    const body = {
+      one: nanoid(10),
+      two: nanoid(5),
+      three: nanoid(),
+      num: 11,
+      flag: false,
+    }
+    const arrayBuffer = str2ab(JSON.stringify(body))
+    const { data, error } = await fclient.invoke<ArrayBuffer>('mirror', {
+      responseType: 'arrayBuffer',
+      body: arrayBuffer,
+      headers: {
+        'content-type': 'application/octet-stream',
+        'response-type': 'arrayBuffer',
+      },
+    })
+    const dataJSON = JSON.parse(new TextDecoder().decode(data ?? Buffer.from('').buffer))
+    dataJSON.body = JSON.parse(dataJSON.body.replace(/\0/g, ''))
+
+    log('assert no error')
+    expect(error).toBeNull()
+
+    const expected = {
+      url: 'http://localhost:8000/mirror',
+      method: 'POST',
+      headers: dataJSON?.headers ?? [],
+      body: body,
+    }
+    attach(
+      'check data from function',
+      `expected: ${JSON.stringify(expected)}\n actual: ${JSON.stringify(dataJSON)}`,
+      ContentType.TEXT
+    )
+    expect(dataJSON).toEqual(expected)
+  })
+
+  test('invoke mirror with body blob', async () => {
+    /**
+     * @feature body
+     */
+    log('create FunctionsClient')
+    const fclient = new FunctionsClient(`http://localhost:${relay.container.getMappedPort(8081)}`)
+    attach('setAuth', apiKey, ContentType.TEXT)
+    fclient.setAuth(apiKey)
+
+    log('invoke mirror')
+    const body = {
+      one: nanoid(10),
+      two: nanoid(5),
+      three: nanoid(),
+      num: 11,
+      flag: false,
+    }
+    const bodyEncoded = str2ab(JSON.stringify(body))
+    const { data, error } = await fclient.invoke<Blob>('mirror', {
+      responseType: 'blob',
+      body: bodyEncoded,
+      headers: {
+        'content-type': 'application/octet-stream',
+        'response-type': 'blob',
+      },
+    })
+    const dataJSON = JSON.parse((await data?.text()) ?? '')
+    dataJSON.body = JSON.parse(dataJSON.body.replace(/\0/g, ''))
+
+    log('assert no error')
+    expect(error).toBeNull()
+
+    const expected = {
+      url: 'http://localhost:8000/mirror',
+      method: 'POST',
+      headers: dataJSON?.headers ?? [],
+      body: body,
+    }
+    attach(
+      'check data from function',
+      `expected: ${JSON.stringify(expected)}\n actual: ${JSON.stringify(dataJSON)}`,
+      ContentType.TEXT
+    )
+    expect(dataJSON).toEqual(expected)
+  })
+
+  test('invoke mirror with url params', async () => {
+    /**
+     * @feature body
+     */
+    log('create FunctionsClient')
+    const fclient = new FunctionsClient(`http://localhost:${relay.container.getMappedPort(8081)}`)
+    attach('setAuth', apiKey, ContentType.TEXT)
+    fclient.setAuth(apiKey)
+
+    log('invoke mirror')
+    const body = {
+      one: nanoid(10),
+      two: nanoid(5),
+      three: nanoid(),
+      num: '11',
+      flag: 'false',
+    }
+    const queryParams = new URLSearchParams(body)
+    const { data, error } = await fclient.invoke<MirrorResponse>(
+      `mirror?${queryParams.toString()}`,
+      {
+        responseType: 'json',
+      }
+    )
+
+    log('assert no error')
+    expect(error).toBeNull()
+
+    const expected = {
+      url: `http://localhost:8000/mirror?${queryParams.toString()}`,
+      method: 'POST',
+      headers: data?.headers ?? [],
+      body: '',
+    }
+    attach(
+      'check data from function',
+      `expected: ${JSON.stringify(expected)}\n actual: ${JSON.stringify(data)}`,
+      ContentType.TEXT
+    )
+    expect(data).toEqual(expected)
+  })
 })

test/utils/binaries.ts

@@ -0,0 +1,27 @@
+/**
+ * Convert a buffer to a string
+ * @param buf - the buffer to convert to a string
+ * @returns The string representation of the buffer.
+ */
+export function ab2str(buf: ArrayBufferLike) {
+  let str = ''
+
+  new Uint16Array(buf).forEach(function (byte: number) {
+    str += String.fromCharCode(byte)
+  })
+  return str
+}
+
+/**
+ * Convert a string to an array buffer
+ * @param str - The string to be converted into an ArrayBuffer.
+ * @returns The array buffer that contains the string.
+ */
+export function str2ab(str: string) {
+  var buf = new ArrayBuffer(str.length * 2) // 2 bytes for each char
+  var bufView = new Uint16Array(buf)
+  for (var i = 0, strLen = str.length; i < strLen; i++) {
+    bufView[i] = str.charCodeAt(i)
+  }
+  return buf
+}