mirror of
https://github.com/codex-storage/nim-libp2p.git
synced 2025-01-10 02:55:45 +00:00
e623e70e7b
This adds gossipsub and floodsub, as well as basic interop testing with the go libp2p daemon. * add close event * wip: gossipsub * splitting rpc message * making message handling more consistent * initial gossipsub implementation * feat: nim 1.0 cleanup * wip: gossipsub protobuf * adding encoding/decoding of gossipsub messages * add disconnect handler * add proper gossipsub msg handling * misc: cleanup for nim 1.0 * splitting floodsub and gossipsub tests * feat: add mesh rebalansing * test pubsub * add mesh rebalansing tests * testing mesh maintenance * finishing mcache implementatin * wip: commenting out broken tests * wip: don't run heartbeat for now * switchout debug for trace logging * testing gossip peer selection algorithm * test stream piping * more work around message amplification * get the peerid from message * use timed cache as backing store * allow setting timeout in constructor * several changes to improve performance * more through testing of msg amplification * prevent gc issues * allow piping to self and prevent deadlocks * improove floodsub * allow running hook on cache eviction * prevent race conditions * prevent race conditions and improove tests * use hashes as cache keys * removing useless file * don't create a new seq * re-enable pubsub tests * fix imports * reduce number of runs to speed up tests * break out control message processing * normalize sleeps between steps * implement proper transport filtering * initial interop testing * clean up floodsub publish logic * allow dialing without a protocol * adding multiple reads/writes * use protobuf varint in mplex * don't loose conn's peerInfo * initial interop pubsub tests * don't duplicate connections/peers * bring back interop tests * wip: interop * re-enable interop and daemon tests * add multiple read write tests from handlers * don't cleanup channel prematurely * use correct channel to send/receive msgs * adjust tests with latest changes * include interop tests * remove temp logging output * fix ci * use correct public key serialization * additional tests for pubsub interop
94 lines
2.9 KiB
Nim
94 lines
2.9 KiB
Nim
import options, sets, sequtils
|
|
import unittest
|
|
import ../../libp2p/[peer,
|
|
crypto/crypto,
|
|
protocols/pubsub/mcache,
|
|
protocols/pubsub/rpc/message,
|
|
protocols/pubsub/rpc/messages]
|
|
|
|
suite "MCache":
|
|
test "put/get":
|
|
var mCache = newMCache(3, 5)
|
|
var msg = Message(fromPeer: PeerID.init(PrivateKey.random(RSA)).data,
|
|
seqno: cast[seq[byte]]("12345"))
|
|
mCache.put(msg)
|
|
check mCache.get(msg.msgId).isSome and mCache.get(msg.msgId).get() == msg
|
|
|
|
test "window":
|
|
var mCache = newMCache(3, 5)
|
|
|
|
for i in 0..<3:
|
|
var msg = Message(fromPeer: PeerID.init(PrivateKey.random(RSA)).data,
|
|
seqno: cast[seq[byte]]("12345"),
|
|
topicIDs: @["foo"])
|
|
mCache.put(msg)
|
|
|
|
for i in 0..<5:
|
|
var msg = Message(fromPeer: PeerID.init(PrivateKey.random(RSA)).data,
|
|
seqno: cast[seq[byte]]("12345"),
|
|
topicIDs: @["bar"])
|
|
mCache.put(msg)
|
|
|
|
var mids = mCache.window("foo")
|
|
check mids.len == 3
|
|
|
|
var id = toSeq(mids)[0]
|
|
check mCache.get(id).get().topicIDs[0] == "foo"
|
|
|
|
test "shift - shift 1 window at a time":
|
|
var mCache = newMCache(1, 5)
|
|
|
|
for i in 0..<3:
|
|
var msg = Message(fromPeer: PeerID.init(PrivateKey.random(RSA)).data,
|
|
seqno: cast[seq[byte]]("12345"),
|
|
topicIDs: @["foo"])
|
|
mCache.put(msg)
|
|
|
|
mCache.shift()
|
|
check mCache.window("foo").len == 0
|
|
|
|
for i in 0..<3:
|
|
var msg = Message(fromPeer: PeerID.init(PrivateKey.random(RSA)).data,
|
|
seqno: cast[seq[byte]]("12345"),
|
|
topicIDs: @["bar"])
|
|
mCache.put(msg)
|
|
|
|
mCache.shift()
|
|
check mCache.window("bar").len == 0
|
|
|
|
for i in 0..<3:
|
|
var msg = Message(fromPeer: PeerID.init(PrivateKey.random(RSA)).data,
|
|
seqno: cast[seq[byte]]("12345"),
|
|
topicIDs: @["baz"])
|
|
mCache.put(msg)
|
|
|
|
mCache.shift()
|
|
check mCache.window("baz").len == 0
|
|
|
|
test "shift - 2 windows at a time":
|
|
var mCache = newMCache(1, 5)
|
|
|
|
for i in 0..<3:
|
|
var msg = Message(fromPeer: PeerID.init(PrivateKey.random(RSA)).data,
|
|
seqno: cast[seq[byte]]("12345"),
|
|
topicIDs: @["foo"])
|
|
mCache.put(msg)
|
|
|
|
for i in 0..<3:
|
|
var msg = Message(fromPeer: PeerID.init(PrivateKey.random(RSA)).data,
|
|
seqno: cast[seq[byte]]("12345"),
|
|
topicIDs: @["bar"])
|
|
mCache.put(msg)
|
|
|
|
for i in 0..<3:
|
|
var msg = Message(fromPeer: PeerID.init(PrivateKey.random(RSA)).data,
|
|
seqno: cast[seq[byte]]("12345"),
|
|
topicIDs: @["baz"])
|
|
mCache.put(msg)
|
|
|
|
mCache.shift()
|
|
check mCache.window("foo").len == 0
|
|
|
|
mCache.shift()
|
|
check mCache.window("bar").len == 0
|