import {beforeAll, expect, test} from 'bun:test'
import {SecretBox, misc} from '..'

let k1!: CryptoKey
let k2!: CryptoKey
let message!: Uint8Array

beforeAll(async () => {
  k1 = await SecretBox.gen(false)
  k2 = await SecretBox.gen(true)
  message = misc.payload_fromString('Salut ! ça va ?')
})

test('base case', async () => {
  const box = await SecretBox.encrypt(message, k1)
  expect(box).not.toBeNull()
  const unboxed = await box!.decrypt(k1)
  expect(unboxed).toEqual(message)
})
test("Different key can't decrypt", async () => {
  const box = await SecretBox.encrypt(message, k1)
  expect(box).not.toBeNull()
  const unboxed = await box!.decrypt(k2)
  expect(unboxed).toBeNull()
})

test('Key generation', async () => {
  expect(k1.extractable).toBeFalse()
  expect(k2.extractable).toBeTrue()
})

test('String serialization', async () => {
  const box = await SecretBox.encrypt(message, k1)
  expect(box).not.toBeNull()

  const ser = box!.toString()
  const de = SecretBox.fromString(ser)

  expect(de).not.toBeNull()
  expect(de).toEqual(box)

  const unboxed = await de!.decrypt(k1)
  expect(unboxed).not.toBeNull()
  expect(unboxed).toEqual(message)
})

test('Byte serialization', async () => {
  const box = await SecretBox.encrypt(message, k1)
  expect(box).not.toBeNull()

  const ser = box!.toBytes()
  const de = SecretBox.fromBytes(ser)

  expect(de).not.toBeNull()
  expect(de).toEqual(box)

  const unboxed = await de!.decrypt(k1)
  expect(unboxed).not.toBeNull()
  expect(unboxed).toEqual(message)
})