import pkg/asynctest
import pkg/chronos
import pkg/codex/proving
import ./helpers/mockproofs
import ./helpers/mockclock
import ./examples

suite "Proving":

  var proving: Proving
  var proofs: MockProofs
  var clock: MockClock

  setup:
    proofs = MockProofs.new()
    clock = MockClock.new()
    proving = Proving.new(proofs, clock)
    await proving.start()

  teardown:
    await proving.stop()

  proc advanceToNextPeriod(proofs: MockProofs) {.async.} =
    let periodicity = await proofs.periodicity()
    clock.advance(periodicity.seconds.truncate(int64))
    await sleepAsync(2.seconds)

  test "maintains a list of contract ids to watch":
    let id1, id2 = SlotId.example
    check proving.slots.len == 0
    proving.add(id1)
    check proving.slots.contains(id1)
    proving.add(id2)
    check proving.slots.contains(id1)
    check proving.slots.contains(id2)

  test "removes duplicate contract ids":
    let id = SlotId.example
    proving.add(id)
    proving.add(id)
    check proving.slots.len == 1

  test "invokes callback when proof is required":
    let id = SlotId.example
    proving.add(id)
    var called: bool
    proc onProofRequired(id: SlotId) =
      called = true
    proving.onProofRequired = onProofRequired
    proofs.setProofRequired(id, true)
    await proofs.advanceToNextPeriod()
    check called

  test "callback receives id of contract for which proof is required":
    let id1, id2 = SlotId.example
    proving.add(id1)
    proving.add(id2)
    var callbackIds: seq[SlotId]
    proc onProofRequired(id: SlotId) =
      callbackIds.add(id)
    proving.onProofRequired = onProofRequired
    proofs.setProofRequired(id1, true)
    await proofs.advanceToNextPeriod()
    check callbackIds == @[id1]
    proofs.setProofRequired(id1, false)
    proofs.setProofRequired(id2, true)
    await proofs.advanceToNextPeriod()
    check callbackIds == @[id1, id2]

  test "invokes callback when proof is about to be required":
    let id = SlotId.example
    proving.add(id)
    var called: bool
    proc onProofRequired(id: SlotId) =
      called = true
    proving.onProofRequired = onProofRequired
    proofs.setProofRequired(id, false)
    proofs.setProofToBeRequired(id, true)
    await proofs.advanceToNextPeriod()
    check called

  test "stops watching when contract has ended":
    let id = SlotId.example
    proving.add(id)
    proofs.setProofEnd(id, clock.now().u256)
    await proofs.advanceToNextPeriod()
    var called: bool
    proc onProofRequired(id: SlotId) =
      called = true
    proving.onProofRequired = onProofRequired
    proofs.setProofRequired(id, true)
    await proofs.advanceToNextPeriod()
    check not proving.slots.contains(id)
    check not called

  test "submits proofs":
    let id = SlotId.example
    let proof = seq[byte].example
    await proving.submitProof(id, proof)

  test "supports proof submission subscriptions":
    let id = SlotId.example
    let proof = seq[byte].example

    var receivedIds: seq[SlotId]
    var receivedProofs: seq[seq[byte]]

    proc onProofSubmission(id: SlotId, proof: seq[byte]) =
      receivedIds.add(id)
      receivedProofs.add(proof)

    let subscription = await proving.subscribeProofSubmission(onProofSubmission)

    await proving.submitProof(id, proof)

    check receivedIds == @[id]
    check receivedProofs == @[proof]

    await subscription.unsubscribe()