feat: add hijack test

Author: egor-romanov

.gitignore

@@ -109,4 +109,7 @@ docs/
 
 # Test reports
 allure-results
-allure-report
+allure-report
+
+# IDE specific
+.vscode

test/functions/hijack/index.ts

@@ -0,0 +1,23 @@
+import { serve } from 'https://deno.land/[email protected]/http/server.ts'
+
+serve((req) => {
+  const p = Deno.upgradeHttp(req);
+  
+  (
+    // Run this async IIFE concurrently, first packet won't arrive
+    // until we return HTTP101 response.
+    async () => {
+      const [conn, firstPacket] = await p
+      const decoder = new TextDecoder()
+      const text = decoder.decode(firstPacket)
+      console.log(text)
+      // Hello
+      const uint8Array = new Uint8Array([72, 101, 108, 108, 111])
+      conn.write(uint8Array)
+      conn.close()
+    }
+  )()
+
+  // HTTP101 - Switching Protocols
+  return new Response(null, { status: 101 })
+})

test/relay/container.ts

@@ -79,6 +79,7 @@ export async function runRelay(
     'run',
     '--allow-all',
     '--watch',
+    '--unstable',
     `/home/deno/${slug}/index.ts`,
   ])
 
@@ -95,7 +96,7 @@ export async function runRelay(
           },
         }
       )
-      if (healthCheck.ok) {
+      if (healthCheck.ok || healthCheck.status === 101) {
         log(`function started to serve: ${slug + '-' + id}`)
         return { container: startedRelay, id, execCache, execRun }
       }

test/spec/hijack.spec.ts

@@ -0,0 +1,46 @@
+import { Blob } from 'buffer'
+import querystring from 'querystring'
+
+import 'jest'
+import { nanoid } from 'nanoid'
+import { sign } from 'jsonwebtoken'
+import { ContentType } from 'allure-js-commons'
+
+import { FunctionsClient } from '../../src/index'
+
+import { Relay, runRelay } from '../relay/container'
+import { attach, log } from '../utils/jest-custom-reporter'
+import { MirrorResponse } from '../models/mirrorResponse'
+
+describe('hijack connection', () => {
+  let relay: Relay
+  const jwtSecret = nanoid(10)
+  const apiKey = sign({ name: 'anon' }, jwtSecret)
+  const func = 'hijack'
+
+  beforeAll(async () => {
+    relay = await runRelay(func, jwtSecret)
+  })
+
+  afterAll(async () => {
+    relay && relay.container && (await relay.container.stop())
+  })
+
+  test('invoke func', async () => {
+    /**
+     * @feature hijack
+     */
+    log('create FunctionsClient')
+    const fclient = new FunctionsClient(`http://localhost:${relay.container.getMappedPort(8081)}`, {
+      Authorization: `Bearer ${apiKey}`,
+    })
+
+    log('invoke func')
+    const { data, error } = await fclient.invoke(func, {
+      responseType: 'text',
+    })
+
+    log('assert error to be "Connection Upgrade is not supported"')
+    expect(error).not.toBeNull()
+  })
+})