Fix test_roundtrip_write (#233)

JuliaIO · Jul 20, 2024 · 466b6d0 · 466b6d0 · nhz2 · Jul 20, 2024
1 parent 76543ed
commit 466b6d0
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 5 deletions.
diff --git a/Project.toml b/Project.toml
@@ -2,7 +2,7 @@ name = "TranscodingStreams"
 uuid = "3bb67fe8-82b1-5028-8e26-92a6c54297fa"
 license = "MIT"
 authors = ["Kenta Sato <bicycle1885@gmail.com>"]
-version = "0.11.0"
+version = "0.11.1"
 
 [deps]
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"

diff --git a/ext/TestExt.jl b/ext/TestExt.jl
@@ -19,14 +19,25 @@ function TranscodingStreams.test_roundtrip_read(encoder, decoder)
     end
 end
 
+# flush all nested streams and return final data
+function take_all(stream)
+    if stream isa Base.GenericIOBuffer
+        seekstart(stream)
+        read(stream)
+    else
+        write(stream, TranscodingStreams.TOKEN_END)
+        flush(stream)
+        take_all(stream.stream)
+    end
+end
+
 function TranscodingStreams.test_roundtrip_write(encoder, decoder)
     seed!(TEST_RANDOM_SEED)
     for n in vcat(0:30, sort!(rand(500:100_000, 30))), alpha in (0x00:0xff, 0x00:0x0f)
         data = rand(alpha, n)
-        file = IOBuffer()
-        stream = encoder(decoder(file))
-        write(stream, data, TOKEN_END); flush(stream)
-        Test.@test hash(take!(file)) == hash(data)
+        stream = encoder(decoder(IOBuffer()))
+        write(stream, data)
+        Test.@test take_all(stream) == data
         close(stream)
     end
 end