nwaku/tests/v2/test_message_cache.nim

154 lines
3.6 KiB
Nim

{.used.}
import
stew/[results, byteutils],
testutils/unittests,
chronicles
import
../../waku/v2/protocol/waku_message,
../../waku/v2/node/message_cache,
./testlib/common,
./testlib/waku2
type TestMessageCache = MessageCache[(PubsubTopic, ContentTopic)]
suite "MessageCache":
test "subscribe to topic":
## Given
let testTopic = (PubsubTopic("test-pubsub-topic"), ContentTopic("test-content-topic"))
let cache = TestMessageCache.init()
## When
cache.subscribe(testTopic)
## Then
check:
cache.isSubscribed(testTopic)
test "unsubscribe from topic":
## Given
let testTopic = (PubsubTopic("test-pubsub-topic"), ContentTopic("test-content-topic"))
let cache = TestMessageCache.init()
# Init cache content
cache.subscribe(testTopic)
## When
cache.unsubscribe(testTopic)
## Then
check:
not cache.isSubscribed(testTopic)
test "get messages of a subscribed topic":
## Given
let testTopic = (PubsubTopic("test-pubsub-topic"), ContentTopic("test-content-topic"))
let testMessage = fakeWakuMessage()
let cache = TestMessageCache.init()
# Init cache content
cache.subscribe(testTopic)
cache.addMessage(testTopic, testMessage)
## When
let res = cache.getMessages(testTopic)
## Then
check:
res.isOk()
res.get() == @[testMessage]
test "get messages with clean flag shoud clear the messages cache":
## Given
let testTopic = (PubsubTopic("test-pubsub-topic"), ContentTopic("test-content-topic"))
let testMessage = fakeWakuMessage()
let cache = TestMessageCache.init()
# Init cache content
cache.subscribe(testTopic)
cache.addMessage(testTopic, testMessage)
## When
var res = cache.getMessages(testTopic, clear=true)
require(res.isOk())
res = cache.getMessages(testTopic)
## Then
check:
res.isOk()
res.get().len == 0
test "get messages of a non-subscribed topic":
## Given
let testTopic = (PubsubTopic("test-pubsub-topic"), ContentTopic("test-content-topic"))
let cache = TestMessageCache.init()
## When
let res = cache.getMessages(testTopic)
## Then
check:
res.isErr()
res.error() == "Not subscribed to topic"
test "add messages to subscribed topic":
## Given
let testTopic = (PubsubTopic("test-pubsub-topic"), ContentTopic("test-content-topic"))
let testMessage = fakeWakuMessage()
let cache = TestMessageCache.init()
cache.subscribe(testTopic)
## When
cache.addMessage(testTopic, testMessage)
## Then
let messages = cache.getMessages(testTopic).tryGet()
check:
messages == @[testMessage]
test "add messages to non-subscribed topic":
## Given
let testTopic = (PubsubTopic("test-pubsub-topic"), ContentTopic("test-content-topic"))
let testMessage = fakeWakuMessage()
let cache = TestMessageCache.init()
## When
cache.addMessage(testTopic, testMessage)
## Then
let res = cache.getMessages(testTopic)
check:
res.isErr()
res.error() == "Not subscribed to topic"
test "add messages beyond the capacity":
## Given
let testTopic = (PubsubTopic("test-pubsub-topic"), ContentTopic("test-content-topic"))
let testMessages = @[
fakeWakuMessage(toBytes("MSG-1")),
fakeWakuMessage(toBytes("MSG-2")),
fakeWakuMessage(toBytes("MSG-3"))
]
let cache = TestMessageCache.init(capacity = 2)
cache.subscribe(testTopic)
## When
for msg in testMessages:
cache.addMessage(testTopic, msg)
## Then
let messages = cache.getMessages(testTopic).tryGet()
check:
messages == testMessages[1..2]