# Nimbus
# Copyright (c) 2018-2019 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
  std/[algorithm, sequtils, strformat, strutils],
  ../nimbus/p2p/[clique, clique/snapshot],
  ../nimbus/utils,
  ./test_clique/pool,
  eth/keys,
  stint,
  unittest2

# clique/snapshot_test.go(99): func TestClique(t *testing.T) {
proc cliqueMain*(noisy = defined(debug)) =
  ## Clique PoA Snapshot
  ## ::
  ##    Tests that Clique signer voting is evaluated correctly for various
  ##    simple and complex scenarios, as well as that a few special corner
  ##    cases fail correctly.
  ##
  suite "Clique PoA Snapshot":
    var
      pool = newTesterPool()
      skipSet = {999}
      testSet = {0 .. 999}

    pool.setDebug(noisy)

    # clique/snapshot_test.go(379): for i, tt := range tests {
    for tt in voterSamples.filterIt(it.id in testSet):

      test &"Snapshots {tt.id:2}: {tt.info.substr(0,50)}...":
        pool.say "\n"

        if tt.id in skipSet:
          skip()

        else:
          # Assemble a chain of headers from the cast votes
          # see clique/snapshot_test.go(407): config := *params.TestChainConfig
          pool.resetVoterChain(tt.signers, tt.epoch)

          # see clique/snapshot_test.go(425): for j, block := range blocks {
          for voter in tt.votes:
            pool.appendVoter(voter)
          pool.commitVoterChain

          # see clique/snapshot_test.go(476): snap, err := engine.snapshot( [..]
          let topHeader = pool.topVoterHeader
          var snap = pool.snapshot(topHeader.blockNumber, topHeader.hash, @[])

          # see clique/snapshot_test.go(477): if err != nil {
          if snap.isErr:
            # Note that clique/snapshot_test.go does not verify _here_ against
            # the scheduled test error -- rather this voting error is supposed
            # to happen earlier (processed at clique/snapshot_test.go(467)) when
            # assembling the block chain (sounds counter intuitive to the author
            # of this source file as the scheduled errors are _clique_ related).
            check snap.error[0] == tt.failure
          else:
            let
              expected = tt.results.mapIt("@" & it).sorted
              snapResult = pool.pp(snap.value.signers).sorted
            pool.say "*** snap state=", snap.pp(16)
            pool.say "        result=[", snapResult.join(",") & "]"
            pool.say "      expected=[", expected.join(",") & "]"

            # Verify the final list of signers against the expected ones
            check snapResult == expected


when isMainModule:
  cliqueMain()

# End