Support read from IO (#20)

rfourquet · Sep 29, 2020 · 5913c82 · 5913c82 · rfourquet · Sep 29, 2020
1 parent 3458b92
commit 5913c82
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "BitIntegers"
 uuid = "c3b6d118-76ef-56ca-8cc7-ebb389d030a1"
 authors = ["Rafael Fourquet <[email protected]>"]
-version = "0.2.3"
+version = "0.2.4"
 
 [compat]
 julia = "1"

diff --git a/src/BitIntegers.jl b/src/BitIntegers.jl
@@ -453,6 +453,10 @@ function read(from::GenericIOBuffer, ::Type{T}) where {T<:XBI}
     return x
 end
 
+function read(io::IO, ::Type{T}) where {T<:XBI}
+    return read!(io, Ref{T}(0))[]::T
+end
+
 function peek(from::GenericIOBuffer, ::Type{T}) where {T<:XBI}
     from.readable || Base._throw_not_readable()
     avail = bytesavailable(from)

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -272,8 +272,7 @@ end
 end
 
 @testset "read/write" begin
-    for X in XInts
-        io = IOBuffer()
+    function read_write_tests(io, X)
         x, y = rand(X, 2)
         @test write(io, x) == sizeof(x)
         @test write(io, y) == sizeof(y)
@@ -296,6 +295,20 @@ end
         q = deserialize(io)
         @test q === true
     end
+
+    @testset "IOBuffer" begin
+        for X in XInts
+            read_write_tests(IOBuffer(), X)
+        end
+    end
+
+    @testset "IOStream" begin
+        for X in XInts
+            open(tempname(), "w+") do iostream
+                read_write_tests(iostream, X)
+            end
+        end
+    end
 end
 
 @testset "rand" begin