js-libp2p-crypto/test/keys/secp256k1.spec.js

/* eslint-env mocha */
'use strict'

const chai = require('chai')
const dirtyChai = require('dirty-chai')
const expect = chai.expect
chai.use(dirtyChai)
const fixtures = require('../fixtures/secp256k1')
const crypto = require('../../src')

const mockPublicKey = {
  bytes: fixtures.pbmPublicKey
}

const mockPrivateKey = {
  bytes: fixtures.pbmPrivateKey,
  public: mockPublicKey
}

const mockSecp256k1Module = {
  generateKeyPair (bits) {
    return mockPrivateKey
  },

  unmarshalSecp256k1PrivateKey (buf) {
    return mockPrivateKey
  },

  unmarshalSecp256k1PublicKey (buf) {
    return mockPublicKey
  }
}

describe('without libp2p-crypto-secp256k1 module present', () => {
  crypto.keys.supportedKeys.secp256k1 = undefined

  it('fails to generate a secp256k1 key', async () => {
    try {
      await crypto.keys.generateKeyPair('secp256k1', 256)
    } catch (err) {
      return // expected
    }
    throw new Error('Expected error to be thrown')
  })

  it('fails to unmarshal a secp256k1 private key', async () => {
    try {
      await crypto.keys.unmarshalPrivateKey(fixtures.pbmPrivateKey)
    } catch (err) {
      return // expected
    }
    throw new Error('Expected error to be thrown')
  })

  it('fails to unmarshal a secp256k1 public key', () => {
    expect(() => {
      crypto.keys.unmarshalPublicKey(fixtures.pbmPublicKey)
    }).to.throw(Error)
  })
})

describe('with libp2p-crypto-secp256k1 module present', () => {
  let key

  before(async () => {
    crypto.keys.supportedKeys.secp256k1 = mockSecp256k1Module
    key = await crypto.keys.generateKeyPair('secp256k1', 256)
  })

  after(() => {
    delete crypto.keys.secp256k1
  })

  it('generates a valid key', () => {
    expect(key).to.exist()
  })

  it('protobuf encoding', async () => {
    const keyMarshal = crypto.keys.marshalPrivateKey(key)
    const key2 = await crypto.keys.unmarshalPrivateKey(keyMarshal)
    const keyMarshal2 = crypto.keys.marshalPrivateKey(key2)

    expect(keyMarshal).to.eql(keyMarshal2)

    const pk = key.public
    const pkMarshal = crypto.keys.marshalPublicKey(pk)
    const pk2 = crypto.keys.unmarshalPublicKey(pkMarshal)
    const pkMarshal2 = crypto.keys.marshalPublicKey(pk2)

    expect(pkMarshal).to.eql(pkMarshal2)
  })
})