nimbus-eth1/tests/test_beacon/test_skeleton.nim

# 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,
  test_1_initsync,
  test_2_extend,
  test_3_sethead_genesis,
  test_4_fill_canonical,
  test_5_canonical_past_genesis,
  test_6_abort_filling,
  test_7_abort_and_backstep,
  test_8_pos_too_early,
  ../../nimbus/sync/beacon/skeleton_main,
  ./setup_env

proc noTtdAndNoTtdPassed(cc: NetworkParams) =
  cc.config.terminalTotalDifficultyPassed = Opt.none(bool)
  cc.config.terminalTotalDifficulty = Opt.none(UInt256)
  cc.genesis.difficulty = 1.u256

proc ttdPassedNoTtd(cc: NetworkParams) =
  cc.config.terminalTotalDifficultyPassed = Opt.some(true)
  cc.config.terminalTotalDifficulty = Opt.none(UInt256)
  cc.genesis.difficulty = 1.u256

proc ttdNoTtdPassed(cc: NetworkParams) =
  cc.config.terminalTotalDifficultyPassed = Opt.none(bool)
  cc.config.terminalTotalDifficulty = Opt.some(0.u256)
  cc.genesis.difficulty = 1.u256

proc ttdAndTtdPassed(cc: NetworkParams) =
  cc.config.terminalTotalDifficultyPassed = Opt.some(true)
  cc.config.terminalTotalDifficulty = Opt.some(0.u256)
  cc.genesis.difficulty = 1.u256

proc ttdPassedFalseNoTtd(cc: NetworkParams) =
  cc.config.terminalTotalDifficultyPassed = Opt.some(false)
  cc.config.terminalTotalDifficulty = Opt.none(UInt256)
  cc.genesis.difficulty = 1.u256

proc skeletonMain*() =
  test1()
  test2()
  test3()
  test4()
  test5()
  test6()
  test7()
  test8()

  suite "test skeleton open":
    test "skeleton open should error if both ttd and ttdPassed not set":
      let env = setupEnv(extraValidation = true, noTtdAndNoTtdPassed)
      let skel = SkeletonRef.new(env.chain)
      let res = skel.open()
      check res.isErr

    test "skeleton open should ok if ttdPassed is set":
      let env = setupEnv(extraValidation = true, ttdPassedNoTtd)
      let skel = SkeletonRef.new(env.chain)
      let res = skel.open()
      check res.isOk

    test "skeleton open should ok if ttd is set":
      let env = setupEnv(extraValidation = true, ttdNoTtdPassed)
      let skel = SkeletonRef.new(env.chain)
      let res = skel.open()
      check res.isOk

    test "skeleton open should ok if both ttd and ttdPassed are set":
      let env = setupEnv(extraValidation = true, ttdAndTtdPassed)
      let skel = SkeletonRef.new(env.chain)
      let res = skel.open()
      check res.isOk

    test "skeleton open should error if ttd not set and ttdPassed are false":
      let env = setupEnv(extraValidation = true, ttdPassedFalseNoTtd)
      let skel = SkeletonRef.new(env.chain)
      let res = skel.open()
      check res.isErr

when isMainModule:
  skeletonMain()