outbound-serializer.test.ts

import { describe, it, expect } from 'vitest'
import { serialize } from '../src/serializer'
import BufferList from './testing/buffer-list'

describe('serializer', () => {
  it('builds startup message', () => {
    const actual = serialize.startup({
      user: 'brian',
      database: 'bang',
    })
    expect(actual).toEqual(
      new BufferList()
        .addInt16(3)
        .addInt16(0)
        .addCString('user')
        .addCString('brian')
        .addCString('database')
        .addCString('bang')
        .addCString('client_encoding')
        .addCString('UTF8')
        .addCString('')
        .join(true),
    )
  })

  it('builds password message', () => {
    const actual = serialize.password('!')
    expect(actual).toEqual(new BufferList().addCString('!').join(true, 'p'))
  })

  it('builds request ssl message', () => {
    const actual = serialize.requestSsl()
    const expected = new BufferList().addInt32(80877103).join(true)
    expect(actual).toEqual(expected)
  })

  it('builds SASLInitialResponseMessage message', () => {
    const actual = serialize.sendSASLInitialResponseMessage('mech', 'data')
    expect(actual).toEqual(
      new BufferList()
        .addCString('mech')
        .addInt32(4)
        .addString('data')
        .join(true, 'p'),
    )
  })

  it('builds SCRAMClientFinalMessage message', () => {
    const actual = serialize.sendSCRAMClientFinalMessage('data')
    expect(actual).toEqual(new BufferList().addString('data').join(true, 'p'))
  })

  it('builds query message', () => {
    const txt = 'select * from boom'
    const actual = serialize.query(txt)
    expect(actual).toEqual(new BufferList().addCString(txt).join(true, 'Q'))
  })

  describe('parse message', () => {
    it('builds parse message', () => {
      const actual = serialize.parse({ text: '!' })
      const expected = new BufferList()
        .addCString('')
        .addCString('!')
        .addInt16(0)
        .join(true, 'P')
      expect(actual).toEqual(expected)
    })

    it('builds parse message with named query', () => {
      const actual = serialize.parse({
        name: 'boom',
        text: 'select * from boom',
        types: [],
      })
      const expected = new BufferList()
        .addCString('boom')
        .addCString('select * from boom')
        .addInt16(0)
        .join(true, 'P')
      expect(actual).toEqual(expected)
    })

    it('with multiple parameters', () => {
      const actual = serialize.parse({
        name: 'force',
        text: 'select * from bang where name = $1',
        types: [1, 2, 3, 4],
      })
      const expected = new BufferList()
        .addCString('force')
        .addCString('select * from bang where name = $1')
        .addInt16(4)
        .addInt32(1)
        .addInt32(2)
        .addInt32(3)
        .addInt32(4)
        .join(true, 'P')
      expect(actual).toEqual(expected)
    })
  })

  describe('bind messages', () => {
    it('with no values', () => {
      const actual = serialize.bind()

      const expectedBuffer = new BufferList()
        .addCString('')
        .addCString('')
        .addInt16(0)
        .addInt16(0)
        .addInt16(0)
        .join(true, 'B')
      expect(actual).toEqual(expectedBuffer)
    })

    it('with named statement, portal, and values', () => {
      const actual = serialize.bind({
        portal: 'bang',
        statement: 'woo',
        values: ['1', 'hi', null, 'zing'],
      })
      const expectedBuffer = new BufferList()
        .addCString('bang') // portal name
        .addCString('woo') // statement name
        .addInt16(4)
        .addInt16(0)
        .addInt16(0)
        .addInt16(0)
        .addInt16(0)
        .addInt16(4)
        .addInt32(1)
        .add(new TextEncoder().encode('1'))
        .addInt32(2)
        .add(new TextEncoder().encode('hi'))
        .addInt32(-1)
        .addInt32(4)
        .add(new TextEncoder().encode('zing'))
        .addInt16(0)
        .join(true, 'B')
      expect(actual).toEqual(expectedBuffer)
    })
  })

  it('with custom valueMapper', () => {
    const actual = serialize.bind({
      portal: 'bang',
      statement: 'woo',
      values: ['1', 'hi', null, 'zing'],
      valueMapper: () => null,
    })
    const expectedBuffer = new BufferList()
      .addCString('bang') // portal name
      .addCString('woo') // statement name
      .addInt16(4)
      .addInt16(0)
      .addInt16(0)
      .addInt16(0)
      .addInt16(0)
      .addInt16(4)
      .addInt32(-1)
      .addInt32(-1)
      .addInt32(-1)
      .addInt32(-1)
      .addInt16(0)
      .join(true, 'B')
    expect(actual).toEqual(expectedBuffer)
  })

  it('with named statement, portal, and buffer value', () => {
    const actual = serialize.bind({
      portal: 'bang',
      statement: 'woo',
      values: ['1', 'hi', null, new TextEncoder().encode('zing')],
    })
    const expectedBuffer = new BufferList()
      .addCString('bang') // portal name
      .addCString('woo') // statement name
      .addInt16(4) // value count
      .addInt16(0) // string
      .addInt16(0) // string
      .addInt16(0) // string
      .addInt16(1) // binary
      .addInt16(4)
      .addInt32(1)
      .add(new TextEncoder().encode('1'))
      .addInt32(2)
      .add(new TextEncoder().encode('hi'))
      .addInt32(-1)
      .addInt32(4)
      .add(new TextEncoder().encode('zing'))
      .addInt16(0)
      .join(true, 'B')
    expect(actual).toEqual(expectedBuffer)
  })

  describe('builds execute message', () => {
    it('for unamed portal with no row limit', () => {
      const actual = serialize.execute()
      const expectedBuffer = new BufferList()
        .addCString('')
        .addInt32(0)
        .join(true, 'E')
      expect(actual).toEqual(expectedBuffer)
    })

    it('for named portal with row limit', () => {
      const actual = serialize.execute({
        portal: 'my favorite portal',
        rows: 100,
      })
      const expectedBuffer = new BufferList()
        .addCString('my favorite portal')
        .addInt32(100)
        .join(true, 'E')
      expect(actual).toEqual(expectedBuffer)
    })
  })

  it('builds flush command', () => {
    const actual = serialize.flush()
    const expected = new BufferList().join(true, 'H')
    expect(actual).toEqual(expected)
  })

  it('builds sync command', () => {
    const actual = serialize.sync()
    const expected = new BufferList().join(true, 'S')
    expect(actual).toEqual(expected)
  })

  it('builds end command', () => {
    const actual = serialize.end()
    const expected = new Uint8Array([0x58, 0, 0, 0, 4])
    expect(actual).toEqual(expected)
  })

  describe('builds describe command', () => {
    it('describe statement', () => {
      const actual = serialize.describe({ type: 'S', name: 'bang' })
      const expected = new BufferList()
        .addChar('S')
        .addCString('bang')
        .join(true, 'D')
      expect(actual).toEqual(expected)
    })

    it('describe unnamed portal', () => {
      const actual = serialize.describe({ type: 'P' })
      const expected = new BufferList()
        .addChar('P')
        .addCString('')
        .join(true, 'D')
      expect(actual).toEqual(expected)
    })
  })

  describe('builds close command', () => {
    it('describe statement', () => {
      const actual = serialize.close({ type: 'S', name: 'bang' })
      const expected = new BufferList()
        .addChar('S')
        .addCString('bang')
        .join(true, 'C')
      expect(actual).toEqual(expected)
    })

    it('describe unnamed portal', () => {
      const actual = serialize.close({ type: 'P' })
      const expected = new BufferList()
        .addChar('P')
        .addCString('')
        .join(true, 'C')
      expect(actual).toEqual(expected)
    })
  })

  describe('copy messages', () => {
    it('builds copyFromChunk', () => {
      const actual = serialize.copyData(new Uint8Array([1, 2, 3]))
      const expected = new BufferList()
        .add(new Uint8Array([1, 2, 3]))
        .join(true, 'd')
      expect(actual).toEqual(expected)
    })

    it('builds copy fail', () => {
      const actual = serialize.copyFail('err!')
      const expected = new BufferList().addCString('err!').join(true, 'f')
      expect(actual).toEqual(expected)
    })

    it('builds copy done', () => {
      const actual = serialize.copyDone()
      const expected = new BufferList().join(true, 'c')
      expect(actual).toEqual(expected)
    })
  })

  it('builds cancel message', () => {
    const actual = serialize.cancel(3, 4)
    const expected = new BufferList()
      .addInt16(1234)
      .addInt16(5678)
      .addInt32(3)
      .addInt32(4)
      .join(true)
    expect(actual).toEqual(expected)
  })
})