import std/unittest
import pkg/dagger/bitswap/engine/payments
import ../../examples

suite "engine payments":

  let amountOfBytes = 42

  var wallet: Wallet
  var peer: BitswapPeerCtx

  setup:
    wallet = Wallet.example
    peer = BitswapPeerCtx.example
    peer.pricing = Pricing.example.some

  test "pays for received bytes":
    let payment = wallet.pay(peer, amountOfBytes).get
    let pricing = peer.pricing.get
    let balances = payment.state.outcome.balances(pricing.asset)
    let destination = pricing.address.toDestination
    check balances[destination].get == amountOfBytes.u256 * pricing.price

  test "no payment when no price is set":
    peer.pricing = Pricing.none
    check wallet.pay(peer, amountOfBytes).isErr

  test "uses same channel for consecutive payments":
    let payment1, payment2 = wallet.pay(peer, amountOfBytes)
    let channel1 = payment1.?state.?channel.?getChannelId
    let channel2 = payment2.?state.?channel.?getChannelId
    check channel1 == channel2