Skip to content

Commit a190037

Browse files
committed
add back test_chunked_write
1 parent a64b01b commit a190037

File tree

2 files changed

+21
-0
lines changed

2 files changed

+21
-0
lines changed

ext/TestExt.jl

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -97,4 +97,24 @@ function TranscodingStreams.test_chunked_read(Encoder, Decoder)
9797
finalize(encoder)
9898
end
9999

100+
function TranscodingStreams.test_chunked_write(Encoder, Decoder)
101+
seed!(TEST_RANDOM_SEED)
102+
alpha = b"空即是色"
103+
encoder = Encoder()
104+
initialize(encoder)
105+
for _ in 1:500
106+
chunks = [rand(alpha, rand(0:100)) for _ in 1:2]
107+
data = map(x->transcode(encoder, x), chunks)
108+
buffer = IOBuffer()
109+
stream = TranscodingStream(Decoder(), buffer, stop_on_end=true)
110+
write(stream, vcat(data...))
111+
close(stream)
112+
ok = true
113+
ok &= hash(take!(buffer)) == hash(vcat(chunks...))
114+
ok &= buffersize(stream.state.buffer1) == 0
115+
Test.@test ok
116+
end
117+
finalize(encoder)
118+
end
119+
100120
end # module

test/codecdoubleframe.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -248,4 +248,5 @@ DoubleFrameDecoderStream(stream::IO; kwargs...) = TranscodingStream(DoubleFrameD
248248
test_roundtrip_transcode(DoubleFrameEncoder, DoubleFrameDecoder)
249249
test_roundtrip_fileio(DoubleFrameEncoder, DoubleFrameDecoder)
250250
test_chunked_read(DoubleFrameEncoder, DoubleFrameDecoder)
251+
test_chunked_write(DoubleFrameEncoder, DoubleFrameDecoder)
251252
end

0 commit comments

Comments
 (0)