## Nim-Libp2p
## Copyright (c) 2018 Status Research & Development GmbH
## Licensed under either of
##  * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE))
##  * MIT license ([LICENSE-MIT](LICENSE-MIT))
## at your option.
## This file may not be copied, modified, or distributed except according to
## those terms.
import unittest
import ../libp2p/crypto/secp
import nimcrypto/utils

suite "Secp256k1 testing suite":
  const TestsCount = 20
  test "Private key serialize/deserialize test":
    for i in 0..<TestsCount:
      var rkey1, rkey2: SkPrivateKey
      var skey2 = newSeq[byte](256)
      var key = SkPrivateKey.random()
      var skey1 = key.getBytes()
      check:
        key.toBytes(skey2) > 0
      check:
        rkey1.init(skey1) == true
        rkey2.init(skey2) == true
      var rkey3 = SkPrivateKey.init(skey1)
      var rkey4 = SkPrivateKey.init(skey2)
      check:
        rkey1 == key
        rkey2 == key
        rkey3 == key
        rkey4 == key
      rkey1.clear()
      rkey2.clear()
      check:
        isFullZero(rkey1.data) == true
        isFullZero(rkey2.data) == true
  test "Public key serialize/deserialize test":
    for i in 0..<TestsCount:
      var rkey1, rkey2: SkPublicKey
      var skey2 = newSeq[byte](256)
      var pair = SkKeyPair.random()
      var skey1 = pair.pubkey.getBytes()
      check:
        pair.pubkey.toBytes(skey2) > 0
        rkey1.init(skey1) == true
        rkey2.init(skey2) == true
      var rkey3 = SkPublicKey.init(skey1)
      var rkey4 = SkPublicKey.init(skey2)
      check:
        rkey1 == pair.pubkey
        rkey2 == pair.pubkey
        rkey3 == pair.pubkey
        rkey4 == pair.pubkey
      rkey1.clear()
      rkey2.clear()
  test "Generate/Sign/Serialize/Deserialize/Verify test":
    var message = "message to sign"
    for i in 0..<TestsCount:
      var kp = SkKeyPair.random()
      var sig = kp.seckey.sign(message)
      var sersk = kp.seckey.getBytes()
      var serpk = kp.pubkey.getBytes()
      var sersig = sig.getBytes()
      var seckey = SkPrivateKey.init(sersk)
      var pubkey = SkPublicKey.init(serpk)
      var csig = SkSignature.init(sersig)
      check csig.verify(message, pubkey) == true
      let error = len(csig.data) - 1
      csig.data[error] = not(csig.data[error])
      check csig.verify(message, pubkey) == false