nim-libp2p/tests/testpeerinfo.nim

{.used.}

import options, bearssl
import chronos
import ../libp2p/crypto/crypto,
       ../libp2p/peerinfo,
       ../libp2p/peerid

import ./helpers

suite "PeerInfo":
  test "Should init with private key":
    let seckey = PrivateKey.random(ECDSA, rng[]).get()
    var peerInfo = PeerInfo.init(seckey)
    var peerId = PeerID.init(seckey).get()

    check peerId == peerInfo.peerId
    check seckey == peerInfo.privateKey
    check seckey.getPublicKey().get() == peerInfo.publicKey.get()

  test "Should init with public key":
    let seckey = PrivateKey.random(ECDSA, rng[]).get()
    var peerInfo = PeerInfo.init(seckey.getPublicKey().get())
    var peerId = PeerID.init(seckey.getPublicKey().get()).get()

    check peerId == peerInfo.peerId
    check seckey.getPublicKey.get() == peerInfo.publicKey.get()

  test "Should init from PeerId with public key":
    let seckey = PrivateKey.random(Ed25519, rng[]).get()
    var peerInfo = PeerInfo.init(PeerID.init(seckey.getPublicKey.get()).get())
    var peerId = PeerID.init(seckey.getPublicKey.get()).get()

    check peerId == peerInfo.peerId
    check seckey.getPublicKey.get() == peerInfo.publicKey.get()

  test "Should init from CIDv0 string":
    var peerInfo: PeerInfo
    try:
      peerInfo = PeerInfo.init("QmYyQSo1c1Ym7orWxLYvCrM2EmxFTANf8wXmmE7DWjhx5N")
    except CatchableError:
      check false

    check:
      PeerID.init("QmYyQSo1c1Ym7orWxLYvCrM2EmxFTANf8wXmmE7DWjhx5N").get() == peerInfo.peerId

  # TODO: CIDv1 handling is missing from PeerID
  # https://github.com/status-im/nim-libp2p/issues/53
  # test "Should init from CIDv1 string":
  #   var peerInfo = PeerInfo.init("bafzbeie5745rpv2m6tjyuugywy4d5ewrqgqqhfnf445he3omzpjbx5xqxe")

  #   check:
  #     PeerID.init("bafzbeie5745rpv2m6tjyuugywy4d5ewrqgqqhfnf445he3omzpjbx5xqxe") == peerInfo.peerId

  test "Should return none if pubkey is missing from id":
    let peerInfo = PeerInfo.init(PeerID.init(PrivateKey.random(ECDSA, rng[]).get()).get())
    check peerInfo.publicKey.isNone

  test "Should return some if pubkey is present in id":
    let peerInfo = PeerInfo.init(PeerID.init(PrivateKey.random(Ed25519, rng[]).get()).get())
    check peerInfo.publicKey.isSome