jyapayne/nim-faststreams
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add OutputStream.getWritableBytes(n)

Browse source
This commit is contained in:
Zahary Karadjov 2022-01-22 19:02:41 +02:00
commit 5a7d1d196f
No known key found for this signature in database
GPG key ID: C8936F8A3073D609
3 changed files with 70 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/test_outputs.nim
View file

@ -213,6 +213,11 @@ suite "output stream":
checkOutputsMatch()
proc writeBlock(data: openArray[byte], output: openArray[byte]): int =
doAssert data.len <= output.len
copyMem(unsafeAddr output[0], unsafeAddr data[0], data.len)
data.len
suite "randomized tests":
type
WriteTypes = enum
@ -357,7 +362,14 @@ suite "randomized tests":
for i in 0 .. 3000:
let bytes = randomBytes(rand(9999) + 1)
referenceBytes.add bytes
output.write bytes
var openArraySize = rand(12000)
if openArraySize >= bytes.len:
# Make sure that sometimes `writeBlock` populates the entire span
if i < 100: openArraySize = bytes.len
output.advance writeBlock(bytes, output.getWritableBytes(openArraySize))
else:
output.write bytes
close output