nim-eth-keys/tests/test_key_and_signature_datastructures.nim

# Nim Eth-keys
# Copyright (c) 2018 Status Research & Development GmbH
# Licensed under either of
#
#  * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE) or http://www.apache.org/licenses/LICENSE-2.0)
#  * MIT license ([LICENSE-MIT](LICENSE-MIT) or http://opensource.org/licenses/MIT)
#
# at your option. This file may not be copied, modified, or distributed except according to those terms.

import  ../src/eth_keys,
        ./config

import  unittest

suite "Test key and signature data structure":
  test "Signing from private key object":
    for person in [alice, bob, eve]:
      let
        pk = initPrivateKey(person.privkey)
        signature = pk.sign_msg(MSG)

      check:
        signature.v == person.raw_sig.v
        signature.r == person.raw_sig.r.u256
        signature.s == person.raw_sig.s.u256

  test "Signing from private key object (ported from official eth-keys)":
    for person in [alice, bob, eve]:
      let
        pk = initPrivateKey(person.privkey)
        signature = pk.sign_msg(MSG)

      check: verify_msg(pk.public_key, MSG, signature)

  test "Hash signing from private key object":
    for person in [alice, bob, eve]:
      let
        pk = initPrivateKey(person.privkey)
        signature = pk.sign_msg(MSG)

      check:
        signature.v == person.raw_sig.v
        signature.r == person.raw_sig.r.u256
        signature.s == person.raw_sig.s.u256

  test "Hash signing from private key object (ported from official eth-keys)":
    for person in [alice, bob, eve]:
      let
        pk = initPrivateKey(person.privkey)
        signature = pk.sign_msg(MSGHASH)

      check: verify_msg(pk.public_key, MSGHASH, signature)

  test "Recover public key from message":
    for person in [alice, bob, eve]:
      let
        pk = initPrivateKey(person.privkey)
        signature = pk.sign_msg(MSG)

        recovered_pubkey = recover_pubkey_from_msg(MSG, signature)

      check: pk.public_key == recovered_pubkey

  test "Recover public key from message hash":
    for person in [alice, bob, eve]:
      let
        pk = initPrivateKey(person.privkey)
        signature = pk.sign_msg(MSGHASH)

        recovered_pubkey = recover_pubkey_from_msg(MSGHASH, signature)

      check: pk.public_key == recovered_pubkey

  test "Signature serialization and deserialization":
    for person in [alice, bob, eve]:
      let
        pk = initPrivateKey(person.privkey)
        signature = pk.sign_msg(MSG)
        deserializedSignature = parseSignature(hexToSeqByteBE(person.serialized_sig))

      var serialized_sig: array[65, byte]
      signature.serialize(serialized_sig)

      check:
        signature == deserializedSignature
        serialized_sig.toHex() == person.serialized_sig
        $signature == person.serialized_sig