nim-libp2p/tests/pubsub/testtimedcache.nim

{.used.}

import std/unittest
import chronos/timer
import ../../libp2p/protocols/pubsub/timedcache

suite "TimedCache":
  test "put/get":
    var cache = TimedCache[int].init(5.seconds)

    let now = Moment.now()
    check:
      not cache.put(1, now)
      not cache.put(2, now + 3.seconds)

    check:
      1 in cache
      2 in cache

    check: not cache.put(3, now + 6.seconds) # expires 1

    check:
      1 notin cache
      2 in cache
      3 in cache

    check:
      cache.put(2, now + 7.seconds) # refreshes 2
      not cache.put(4, now + 12.seconds) # expires 3

    check:
      2 in cache
      3 notin cache
      4 in cache

    check:
      not cache.put(100, now + 100.seconds) # expires everything
      100 in cache