test seekstart deletes data

JuliaIO · Oct 16, 2023 · 7961e40 · 7961e40
1 parent 1cb583d
commit 7961e40
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/test/codecnoop.jl b/test/codecnoop.jl
@@ -333,4 +333,20 @@
             end
         end
     end
+
+    @testset "seekstart doesn't delete data" begin
+        sink = IOBuffer()
+        stream = NoopStream(sink, bufsize=16)
+        write(stream, "x")
+        # seekstart must not delete user data even if it errors.
+        try
+            seekstart(stream)
+        catch e
+            e isa ArgumentError || rethrow()
+        end
+        write(stream, TranscodingStreams.TOKEN_END)
+        flush(stream)
+        @test_broken take!(sink) == b"x"
+        close(stream)
+    end
 end
diff --git a/test/codecquadruple.jl b/test/codecquadruple.jl
@@ -107,4 +107,20 @@ end
             end
         end
     end
+
+    @testset "seekstart doesn't delete data" begin
+        sink = IOBuffer()
+        stream = TranscodingStream(QuadrupleCodec(), sink, bufsize=16)
+        write(stream, "x")
+        # seekstart must not delete user data even if it errors.
+        try
+            seekstart(stream)
+        catch e
+            e isa ArgumentError || rethrow()
+        end
+        write(stream, TranscodingStreams.TOKEN_END)
+        flush(stream)
+        @test_broken take!(sink) == b"xxxx"
+        close(stream)
+    end
 end