nim-libp2p-experimental/tests/teststreamseq.nim

{.used.}

import unittest2
import stew/byteutils
import ../libp2p/stream/streamseq

suite "StreamSeq":
  test "basics":
    var s: StreamSeq

    check:
      s.data().len == 0

    s.add([byte 0, 1, 2, 3])

    check:
      @(s.data()) == [byte 0, 1, 2, 3]

    s.prepare(10)[0..<3] = [byte 4, 5, 6]

    check:
      @(s.data()) == [byte 0, 1, 2, 3]

    s.commit(3)

    check:
      @(s.data()) == [byte 0, 1, 2, 3, 4, 5, 6]

    s.consume(1)

    check:
      @(s.data()) == [byte 1, 2, 3, 4, 5, 6]

    s.consume(6)

    check: @(s.data()) == []

    s.add([])
    check: @(s.data()) == []

    var o: seq[byte]

    check: 0 == s.consumeTo(o)

    s.add([byte 1, 2, 3])

    o.setLen(2)
    o.setLen(s.consumeTo(o))
    check:
      o == [byte 1, 2]

    o.setLen(s.consumeTo(o))

    check:
      o == [byte 3]