import {test, expect} from 'bun:test'

import {signature} from '../index'
import {derive_keypair, gen_keypair, sign, verify} from "../src/signature";

test('base case', async () => {
    const keypair = await gen_keypair()
    const data = new TextEncoder().encode("Message 123 !")

    const sig = await sign(data, keypair.privateKey)
    const verification = await verify(data, keypair.publicKey, sig)

    expect(verification).toBe(true)
})

test('inverted keys', async () => {
    const keypair = await signature.gen_keypair()
    const data = new TextEncoder().encode("Message 123 !")

    expect(async () => await sign(data, keypair.publicKey)).toThrow()
})

test('tampered message', async () => {
    const keypair = await signature.gen_keypair()
    const data1 = new TextEncoder().encode("Message 123 !")
    const data2 = new TextEncoder().encode("Message 321 !")
    expect(data1).not.toEqual(data2)

    const sig = await sign(data1, keypair.privateKey)
    const verification = await verify(data2, keypair.publicKey, sig)

    expect(verification).toBe(false)
})

test('different keypair', async () => {
    const keypair = await signature.gen_keypair()
    const keypair2 = await signature.gen_keypair()
    const data = new TextEncoder().encode("Message 123 !")

    const sig = await sign(data, keypair.privateKey)
    const verification = await verify(data, keypair2.publicKey, sig)

    expect(verification).toBe(false)
})

test('tampered signature', async () => {
    const keypair = await signature.gen_keypair()
    const data = new TextEncoder().encode("Message 123 !")

    const sig = await sign(data, keypair.privateKey)
    sig[0] ^= 1
    const verification = await verify(data, keypair.publicKey, sig)

    expect(verification).toBe(false)
})