import
  std/[os],
  unittest2, eth/common, nimcrypto/hash,
  ../nimbus/[genesis, config, chain_config]

const dataFolder = "tests" / "customgenesis"

proc genesisTest() =
  suite "Genesis":
    test "Correct mainnet hash":
      let g = defaultGenesisBlockForNetwork(MainNet)
      let b = g.toBlock
      check(b.blockHash == "D4E56740F876AEF8C010B86A40D5F56745A118D0906A34E69AEC8C0DB1CB8FA3".toDigest)

    test "Correct ropstennet hash":
      let g = defaultGenesisBlockForNetwork(RopstenNet)
      let b = g.toBlock
      check(b.blockHash == "41941023680923e0fe4d74a34bdac8141f2540e3ae90623718e47d66d1ca4a2d".toDigest)

    test "Correct rinkebynet hash":
      let g = defaultGenesisBlockForNetwork(RinkebyNet)
      let b = g.toBlock
      check(b.blockHash == "6341fd3daf94b748c72ced5a5b26028f2474f5f00d824504e4fa37a75767e177".toDigest)

    test "Correct goerlinet hash":
      let g = defaultGenesisBlockForNetwork(GoerliNet)
      let b = g.toBlock
      check(b.blockHash == "bf7e331f7f7c1dd2e05159666b3bf8bc7a8a3a9eb1d518969eab529dd9b88c1a".toDigest)

proc customGenesisTest() =
  suite "Custom Genesis":
    test "loadCustomGenesis":
      var cga, cgb, cgc: CustomGenesis
      check loadCustomGenesis(dataFolder / "berlin2000.json", cga)
      check loadCustomGenesis(dataFolder / "chainid7.json", cgb)
      check loadCustomGenesis(dataFolder / "noconfig.json", cgc)

proc genesisMain*() =
  genesisTest()
  customGenesisTest()

when isMainModule:
  genesisMain()