mirror of https://github.com/waku-org/nwaku.git
135 lines
4.4 KiB
Nim
135 lines
4.4 KiB
Nim
import
|
|
std/[unittest, options, tables, sets],
|
|
chronos, chronicles, stew/shims/net as stewNet, stew/byteutils,
|
|
libp2p/switch,
|
|
libp2p/protobuf/minprotobuf,
|
|
libp2p/stream/[bufferstream, connection],
|
|
libp2p/crypto/[crypto, secp],
|
|
libp2p/switch,
|
|
eth/keys,
|
|
../../waku/v2/protocol/[waku_message, message_notifier],
|
|
../../waku/v2/protocol/waku_store/waku_store,
|
|
../../waku/v2/protocol/waku_swap/waku_swap,
|
|
../../waku/v2/node/wakunode2,
|
|
../test_helpers, ./utils
|
|
|
|
procSuite "Waku SWAP Accounting":
|
|
test "Handshake Encode/Decode":
|
|
let
|
|
beneficiary = @[byte 0, 1, 2]
|
|
handshake = Handshake(beneficiary: beneficiary)
|
|
pb = handshake.encode()
|
|
|
|
let decodedHandshake = Handshake.init(pb.buffer)
|
|
|
|
check:
|
|
decodedHandshake.isErr == false
|
|
decodedHandshake.get().beneficiary == beneficiary
|
|
|
|
test "Cheque Encode/Decode":
|
|
let
|
|
amount = 1'u32
|
|
date = 9000'u32
|
|
beneficiary = @[byte 0, 1, 2]
|
|
cheque = Cheque(beneficiary: beneficiary, amount: amount, date: date)
|
|
pb = cheque.encode()
|
|
|
|
let decodedCheque = Cheque.init(pb.buffer)
|
|
|
|
check:
|
|
decodedCheque.isErr == false
|
|
decodedCheque.get() == cheque
|
|
|
|
asyncTest "Update accounting state after store operations":
|
|
let
|
|
nodeKey1 = crypto.PrivateKey.random(Secp256k1, rng[])[]
|
|
node1 = WakuNode.init(nodeKey1, ValidIpAddress.init("0.0.0.0"),
|
|
Port(60000))
|
|
nodeKey2 = crypto.PrivateKey.random(Secp256k1, rng[])[]
|
|
node2 = WakuNode.init(nodeKey2, ValidIpAddress.init("0.0.0.0"),
|
|
Port(60001))
|
|
contentTopic = ContentTopic(1)
|
|
message = WakuMessage(payload: "hello world".toBytes(), contentTopic: contentTopic)
|
|
|
|
var completionFut = newFuture[bool]()
|
|
|
|
# Start nodes and mount protocols
|
|
await node1.start()
|
|
node1.mountSwap()
|
|
node1.mountStore()
|
|
await node2.start()
|
|
node2.mountSwap()
|
|
node2.mountStore()
|
|
|
|
await node2.subscriptions.notify("/waku/2/default-waku/proto", message)
|
|
|
|
await sleepAsync(2000.millis)
|
|
|
|
node1.wakuStore.setPeer(node2.peerInfo)
|
|
node1.wakuSwap.setPeer(node2.peerInfo)
|
|
node2.wakuSwap.setPeer(node1.peerInfo)
|
|
|
|
proc storeHandler(response: HistoryResponse) {.gcsafe, closure.} =
|
|
debug "storeHandler hit"
|
|
check:
|
|
response.messages[0] == message
|
|
completionFut.complete(true)
|
|
|
|
await node1.query(HistoryQuery(topics: @[contentTopic]), storeHandler)
|
|
|
|
check:
|
|
(await completionFut.withTimeout(5.seconds)) == true
|
|
# Accounting table updated with credit and debit, respectively
|
|
node1.wakuSwap.accounting[node2.peerInfo.peerId] == 1
|
|
node2.wakuSwap.accounting[node1.peerInfo.peerId] == -1
|
|
await node1.stop()
|
|
await node2.stop()
|
|
|
|
# TODO Add cheque here
|
|
asyncTest "Update accounting state after sending cheque":
|
|
let
|
|
nodeKey1 = crypto.PrivateKey.random(Secp256k1, rng[])[]
|
|
node1 = WakuNode.init(nodeKey1, ValidIpAddress.init("0.0.0.0"),
|
|
Port(60000))
|
|
nodeKey2 = crypto.PrivateKey.random(Secp256k1, rng[])[]
|
|
node2 = WakuNode.init(nodeKey2, ValidIpAddress.init("0.0.0.0"),
|
|
Port(60001))
|
|
contentTopic = ContentTopic(1)
|
|
message = WakuMessage(payload: "hello world".toBytes(), contentTopic: contentTopic)
|
|
|
|
var futures = [newFuture[bool](), newFuture[bool]()]
|
|
|
|
# Start nodes and mount protocols
|
|
await node1.start()
|
|
node1.mountSwap()
|
|
node1.mountStore()
|
|
await node2.start()
|
|
node2.mountSwap()
|
|
node2.mountStore()
|
|
|
|
await node2.subscriptions.notify("/waku/2/default-waku/proto", message)
|
|
|
|
await sleepAsync(2000.millis)
|
|
|
|
node1.wakuStore.setPeer(node2.peerInfo)
|
|
node1.wakuSwap.setPeer(node2.peerInfo)
|
|
node2.wakuSwap.setPeer(node1.peerInfo)
|
|
|
|
proc handler1(response: HistoryResponse) {.gcsafe, closure.} =
|
|
futures[0].complete(true)
|
|
proc handler2(response: HistoryResponse) {.gcsafe, closure.} =
|
|
futures[1].complete(true)
|
|
|
|
# TODO Handshakes - for now we assume implicit, e2e still works for PoC
|
|
await node1.query(HistoryQuery(topics: @[contentTopic]), handler1)
|
|
await node1.query(HistoryQuery(topics: @[contentTopic]), handler2)
|
|
|
|
check:
|
|
(await allFutures(futures).withTimeout(5.seconds)) == true
|
|
# Accounting table updated with credit and debit, respectively
|
|
# After sending a cheque the balance is partially adjusted
|
|
node1.wakuSwap.accounting[node2.peerInfo.peerId] == 1
|
|
node2.wakuSwap.accounting[node1.peerInfo.peerId] == -1
|
|
await node1.stop()
|
|
await node2.stop()
|