Skip to content

Commit

Permalink
revert change to repeated TOKEN_END
Browse files Browse the repository at this point in the history
  • Loading branch information
nhz2 committed Mar 10, 2024
1 parent a190037 commit 4aa49d3
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 5 deletions.
3 changes: 0 additions & 3 deletions src/stream.jl
Original file line number Diff line number Diff line change
Expand Up @@ -514,9 +514,6 @@ const TOKEN_END = EndToken()

function Base.write(stream::TranscodingStream, ::EndToken)
changemode!(stream, :write)
if stream.state.code == :end
callstartproc(stream, :write)
end
flushbufferall(stream)
flushuntilend(stream)
return 0
Expand Down
5 changes: 3 additions & 2 deletions test/codecdoubleframe.jl
Original file line number Diff line number Diff line change
Expand Up @@ -228,7 +228,7 @@ DoubleFrameDecoderStream(stream::IO; kwargs...) = TranscodingStream(DoubleFrameD
end == "[ HHeelllloo,, wwoorrlldd..\n\n ]"
end

@testset "TOKEN_END repeated creates empty frames" begin
@testset "TOKEN_END repeated doesn't create more empty frames" begin
sink = IOBuffer()
stream = TranscodingStream(DoubleFrameEncoder(), sink, stop_on_end=true)
write(stream, TranscodingStreams.TOKEN_END)
Expand All @@ -237,9 +237,10 @@ DoubleFrameDecoderStream(stream::IO; kwargs...) = TranscodingStream(DoubleFrameD
write(stream, TranscodingStreams.TOKEN_END)
write(stream, "de")
write(stream, TranscodingStreams.TOKEN_END)
write(stream, "") # This doesn't create an empty frame
write(stream, TranscodingStreams.TOKEN_END)
close(stream)
@test String(take!(sink)) == "[ ][ ][ aabbcc ][ ddee ][ ]"
@test String(take!(sink)) == "[ ][ aabbcc ][ ddee ]"
end

test_roundtrip_read(DoubleFrameEncoderStream, DoubleFrameDecoderStream)
Expand Down

0 comments on commit 4aa49d3

Please sign in to comment.