* tests: invert message logic on expect from #688
* fix: export pubsub_errors for backward compatibility
This commit is contained in:
Eric Mastro 2022-02-25 03:32:20 +11:00 committed by GitHub
parent c09d032133
commit 44a7260f07
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 19 additions and 18 deletions

View File

@ -30,6 +30,7 @@ export results
export PubSubPeer export PubSubPeer
export PubSubObserver export PubSubObserver
export protocol export protocol
export pubsub_errors
logScope: logScope:
topics = "libp2p pubsub" topics = "libp2p pubsub"

View File

@ -39,7 +39,7 @@ proc randomPeerId(): PeerId =
except CatchableError as exc: except CatchableError as exc:
raise newException(Defect, exc.msg) raise newException(Defect, exc.msg)
const MsgIdFail = "msg id gen failure" const MsgIdSuccess = "msg id gen success"
suite "GossipSub internal": suite "GossipSub internal":
teardown: teardown:
@ -310,7 +310,7 @@ suite "GossipSub internal":
conn.peerId = peerId conn.peerId = peerId
inc seqno inc seqno
let msg = Message.init(peerId, ("HELLO" & $i).toBytes(), topic, some(seqno)) let msg = Message.init(peerId, ("HELLO" & $i).toBytes(), topic, some(seqno))
gossipSub.mcache.put(gossipSub.msgIdProvider(msg).expect(MsgIdFail), msg) gossipSub.mcache.put(gossipSub.msgIdProvider(msg).expect(MsgIdSuccess), msg)
check gossipSub.fanout[topic].len == 15 check gossipSub.fanout[topic].len == 15
check gossipSub.mesh[topic].len == 15 check gossipSub.mesh[topic].len == 15
@ -357,7 +357,7 @@ suite "GossipSub internal":
conn.peerId = peerId conn.peerId = peerId
inc seqno inc seqno
let msg = Message.init(peerId, ("HELLO" & $i).toBytes(), topic, some(seqno)) let msg = Message.init(peerId, ("HELLO" & $i).toBytes(), topic, some(seqno))
gossipSub.mcache.put(gossipSub.msgIdProvider(msg).expect(MsgIdFail), msg) gossipSub.mcache.put(gossipSub.msgIdProvider(msg).expect(MsgIdSuccess), msg)
let peers = gossipSub.getGossipPeers() let peers = gossipSub.getGossipPeers()
check peers.len == gossipSub.parameters.d check peers.len == gossipSub.parameters.d
@ -398,7 +398,7 @@ suite "GossipSub internal":
conn.peerId = peerId conn.peerId = peerId
inc seqno inc seqno
let msg = Message.init(peerId, ("HELLO" & $i).toBytes(), topic, some(seqno)) let msg = Message.init(peerId, ("HELLO" & $i).toBytes(), topic, some(seqno))
gossipSub.mcache.put(gossipSub.msgIdProvider(msg).expect(MsgIdFail), msg) gossipSub.mcache.put(gossipSub.msgIdProvider(msg).expect(MsgIdSuccess), msg)
let peers = gossipSub.getGossipPeers() let peers = gossipSub.getGossipPeers()
check peers.len == gossipSub.parameters.d check peers.len == gossipSub.parameters.d
@ -439,7 +439,7 @@ suite "GossipSub internal":
conn.peerId = peerId conn.peerId = peerId
inc seqno inc seqno
let msg = Message.init(peerId, ("bar" & $i).toBytes(), topic, some(seqno)) let msg = Message.init(peerId, ("bar" & $i).toBytes(), topic, some(seqno))
gossipSub.mcache.put(gossipSub.msgIdProvider(msg).expect(MsgIdFail), msg) gossipSub.mcache.put(gossipSub.msgIdProvider(msg).expect(MsgIdSuccess), msg)
let peers = gossipSub.getGossipPeers() let peers = gossipSub.getGossipPeers()
check peers.len == 0 check peers.len == 0

View File

@ -13,13 +13,13 @@ var rng = newRng()
proc randomPeerId(): PeerId = proc randomPeerId(): PeerId =
PeerId.init(PrivateKey.random(ECDSA, rng[]).get()).get() PeerId.init(PrivateKey.random(ECDSA, rng[]).get()).get()
const MsgIdGenFail = "msg id gen failure" const MsgIdGenSuccess = "msg id generation success"
suite "MCache": suite "MCache":
test "put/get": test "put/get":
var mCache = MCache.init(3, 5) var mCache = MCache.init(3, 5)
var msg = Message(fromPeer: randomPeerId(), seqno: "12345".toBytes()) var msg = Message(fromPeer: randomPeerId(), seqno: "12345".toBytes())
let msgId = defaultMsgIdProvider(msg).expect(MsgIdGenFail) let msgId = defaultMsgIdProvider(msg).expect(MsgIdGenSuccess)
mCache.put(msgId, msg) mCache.put(msgId, msg)
check mCache.get(msgId).isSome and mCache.get(msgId).get() == msg check mCache.get(msgId).isSome and mCache.get(msgId).get() == msg
@ -30,13 +30,13 @@ suite "MCache":
var msg = Message(fromPeer: randomPeerId(), var msg = Message(fromPeer: randomPeerId(),
seqno: "12345".toBytes(), seqno: "12345".toBytes(),
topicIDs: @["foo"]) topicIDs: @["foo"])
mCache.put(defaultMsgIdProvider(msg).expect(MsgIdGenFail), msg) mCache.put(defaultMsgIdProvider(msg).expect(MsgIdGenSuccess), msg)
for i in 0..<5: for i in 0..<5:
var msg = Message(fromPeer: randomPeerId(), var msg = Message(fromPeer: randomPeerId(),
seqno: "12345".toBytes(), seqno: "12345".toBytes(),
topicIDs: @["bar"]) topicIDs: @["bar"])
mCache.put(defaultMsgIdProvider(msg).expect(MsgIdGenFail), msg) mCache.put(defaultMsgIdProvider(msg).expect(MsgIdGenSuccess), msg)
var mids = mCache.window("foo") var mids = mCache.window("foo")
check mids.len == 3 check mids.len == 3
@ -51,7 +51,7 @@ suite "MCache":
var msg = Message(fromPeer: randomPeerId(), var msg = Message(fromPeer: randomPeerId(),
seqno: "12345".toBytes(), seqno: "12345".toBytes(),
topicIDs: @["foo"]) topicIDs: @["foo"])
mCache.put(defaultMsgIdProvider(msg).expect(MsgIdGenFail), msg) mCache.put(defaultMsgIdProvider(msg).expect(MsgIdGenSuccess), msg)
mCache.shift() mCache.shift()
check mCache.window("foo").len == 0 check mCache.window("foo").len == 0
@ -60,7 +60,7 @@ suite "MCache":
var msg = Message(fromPeer: randomPeerId(), var msg = Message(fromPeer: randomPeerId(),
seqno: "12345".toBytes(), seqno: "12345".toBytes(),
topicIDs: @["bar"]) topicIDs: @["bar"])
mCache.put(defaultMsgIdProvider(msg).expect(MsgIdGenFail), msg) mCache.put(defaultMsgIdProvider(msg).expect(MsgIdGenSuccess), msg)
mCache.shift() mCache.shift()
check mCache.window("bar").len == 0 check mCache.window("bar").len == 0
@ -69,7 +69,7 @@ suite "MCache":
var msg = Message(fromPeer: randomPeerId(), var msg = Message(fromPeer: randomPeerId(),
seqno: "12345".toBytes(), seqno: "12345".toBytes(),
topicIDs: @["baz"]) topicIDs: @["baz"])
mCache.put(defaultMsgIdProvider(msg).expect(MsgIdGenFail), msg) mCache.put(defaultMsgIdProvider(msg).expect(MsgIdGenSuccess), msg)
mCache.shift() mCache.shift()
check mCache.window("baz").len == 0 check mCache.window("baz").len == 0
@ -81,19 +81,19 @@ suite "MCache":
var msg = Message(fromPeer: randomPeerId(), var msg = Message(fromPeer: randomPeerId(),
seqno: "12345".toBytes(), seqno: "12345".toBytes(),
topicIDs: @["foo"]) topicIDs: @["foo"])
mCache.put(defaultMsgIdProvider(msg).expect(MsgIdGenFail), msg) mCache.put(defaultMsgIdProvider(msg).expect(MsgIdGenSuccess), msg)
for i in 0..<3: for i in 0..<3:
var msg = Message(fromPeer: randomPeerId(), var msg = Message(fromPeer: randomPeerId(),
seqno: "12345".toBytes(), seqno: "12345".toBytes(),
topicIDs: @["bar"]) topicIDs: @["bar"])
mCache.put(defaultMsgIdProvider(msg).expect(MsgIdGenFail), msg) mCache.put(defaultMsgIdProvider(msg).expect(MsgIdGenSuccess), msg)
for i in 0..<3: for i in 0..<3:
var msg = Message(fromPeer: randomPeerId(), var msg = Message(fromPeer: randomPeerId(),
seqno: "12345".toBytes(), seqno: "12345".toBytes(),
topicIDs: @["baz"]) topicIDs: @["baz"])
mCache.put(defaultMsgIdProvider(msg).expect(MsgIdGenFail), msg) mCache.put(defaultMsgIdProvider(msg).expect(MsgIdGenSuccess), msg)
mCache.shift() mCache.shift()
check mCache.window("foo").len == 0 check mCache.window("foo").len == 0

View File

@ -29,7 +29,7 @@ suite "Message":
E6B946186A4EB44E0D714B2A2D48263D75CF52D30BEF9D9AE2A9FEB7DAF1775F E6B946186A4EB44E0D714B2A2D48263D75CF52D30BEF9D9AE2A9FEB7DAF1775F
E731065A""" E731065A"""
seckey = PrivateKey.init(fromHex(stripSpaces(pkHex))) seckey = PrivateKey.init(fromHex(stripSpaces(pkHex)))
.expect("invalid private key bytes") .expect("valid private key bytes")
peer = PeerInfo.new(seckey) peer = PeerInfo.new(seckey)
msg = Message.init(some(peer), @[], "topic", some(seqno), sign = true) msg = Message.init(some(peer), @[], "topic", some(seqno), sign = true)
msgIdResult = msg.defaultMsgIdProvider() msgIdResult = msg.defaultMsgIdProvider()
@ -47,7 +47,7 @@ suite "Message":
E6B946186A4EB44E0D714B2A2D48263D75CF52D30BEF9D9AE2A9FEB7DAF1775F E6B946186A4EB44E0D714B2A2D48263D75CF52D30BEF9D9AE2A9FEB7DAF1775F
E731065A""" E731065A"""
seckey = PrivateKey.init(fromHex(stripSpaces(pkHex))) seckey = PrivateKey.init(fromHex(stripSpaces(pkHex)))
.expect("invalid private key bytes") .expect("valid private key bytes")
peer = PeerInfo.new(seckey) peer = PeerInfo.new(seckey)
var msg = Message.init(peer.some, @[], "topic", some(seqno), sign = true) var msg = Message.init(peer.some, @[], "topic", some(seqno), sign = true)
@ -65,7 +65,7 @@ suite "Message":
E6B946186A4EB44E0D714B2A2D48263D75CF52D30BEF9D9AE2A9FEB7DAF1775F E6B946186A4EB44E0D714B2A2D48263D75CF52D30BEF9D9AE2A9FEB7DAF1775F
E731065A""" E731065A"""
seckey = PrivateKey.init(fromHex(stripSpaces(pkHex))) seckey = PrivateKey.init(fromHex(stripSpaces(pkHex)))
.expect("invalid private key bytes") .expect("valid private key bytes")
peer = PeerInfo.new(seckey) peer = PeerInfo.new(seckey)
msg = Message.init(some(peer), @[], "topic", uint64.none, sign = true) msg = Message.init(some(peer), @[], "topic", uint64.none, sign = true)
msgIdResult = msg.defaultMsgIdProvider() msgIdResult = msg.defaultMsgIdProvider()