# Nimbus
# Copyright (c) 2023-2024 Status Research & Development GmbH
# Licensed and distributed under either of
#   * MIT license (license terms in the root directory or at
#     https://opensource.org/licenses/MIT).
#   * Apache v2 license (license terms in the root directory or at
#     https://www.apache.org/licenses/LICENSE-2.0).
# at your option. This file may not be copied, modified, or distributed
# except according to those terms.

import
  unittest2,
  stew/byteutils,
  ./setup_env,
  ../../nimbus/sync/beacon/skeleton_main,
  ../../nimbus/sync/beacon/skeleton_utils,
  ../../nimbus/sync/beacon/skeleton_db

proc ccm(cc: NetworkParams) =
  cc.config.terminalTotalDifficulty = Opt.some(262000.u256)
  cc.genesis.extraData = hexToSeqByte("0x000000000000000000")
  cc.genesis.difficulty = 1.u256

proc test8*() =
  suite "should abort filling the canonical chain if a PoS block comes too early without hitting ttd":
    let env = setupEnv(extraValidation = true, ccm)
    let skel = SkeletonRef.new(env.chain)
    skel.fillCanonicalBackStep = 0

    test "skel open ok":
      let res = skel.open()
      check res.isOk
      if res.isErr:
        debugEcho res.error
        check false
        break

    let
      genesis = env.chain.com.genesisHeader
      block1 = env.chain.com.header(1, genesis, genesis,
        "6BD9564DD3F4028E3E56F62F1BE52EC8F893CC4FD7DB75DB6A1DC3EB2858998C")
      block2 = env.chain.com.header(2, block1, block1,
        "32DAA84E151F4C8C6BD4D9ADA4392488FFAFD42ACDE1E9C662B3268C911A5CCC")
      block2PoS = header(2, block1, block1, 0)
      block3 = header(3, block2, block2, 0)
      emptyBody = emptyBody()

    test "put body":
      for header in [block1, block2, block2PoS, block3]:
        let res = skel.putBody(header, emptyBody)
        check res.isOk

    test "canonical height should stop at block 1":
      # (valid PoW block), since block 2 is invalid (invalid PoS, not past ttd)
      skel.initSyncT(block2PoS, true)
      skel.putBlocksT([block1], 1, {FillCanonical})
      check skel.blockHeight == 1

    test "canonical height should now be at head with correct chain":
      # Put correct chain
      skel.initSyncT(block3, true)
      skel.putBlocksT([block2], 1, {FillCanonical})
      check skel.blockHeight == 3
      let latestHash = env.chain.currentBlock().blockHash
      check latestHash == block3.blockHash