Save stdout/err of tests with colors (#70)

* Save stdout/err of tests with colors

* Add test for colorful output

* Bump patch version number

* Update src/ParallelTestRunner.jl

Co-authored-by: Tim Besard <[email protected]>

---------

Co-authored-by: Tim Besard <[email protected]>

Author: giordano

Project.toml

@@ -1,7 +1,7 @@
 name = "ParallelTestRunner"
 uuid = "d3525ed8-44d0-4b2c-a655-542cee43accc"
 authors = ["Valentin Churavy <[email protected]>"]
-version = "2.0.1"
+version = "2.0.2"
 
 [deps]
 Dates = "ade2ca70-3891-5945-98fb-dc099432e06a"

src/ParallelTestRunner.jl

@@ -258,7 +258,8 @@ function runtest(f, name, init_code, color)
                 take!(pipe_initialized)
                 read(pipe, String)
             end
-            stats = redirect_stdio(stdout=pipe, stderr=pipe) do
+            io = IOContext(pipe, :color=>$(color))
+            stats = redirect_stdio(; stdout=io, stderr=io) do
                 put!(pipe_initialized, nothing)
 
                 # @testset CustomTestRecord switches the all lower-level testset to our custom testset,

test/runtests.jl

@@ -230,4 +230,32 @@ end
     @test contains(str, "SUCCESS")
 end
 
+# Issue <https://github.com/JuliaTesting/ParallelTestRunner.jl/issues/69>.
+@testset "colorful output" begin
+    testsuite = Dict(
+        "color" => quote
+            printstyled("Roses Are Red"; color=:red)
+        end
+    )
+    io = IOBuffer()
+    ioc = IOContext(io, :color => true)
+    runtests(ParallelTestRunner, String[]; testsuite, stdout=ioc, stderr=ioc)
+    str = String(take!(io))
+    @test contains(str, "\e[31mRoses Are Red\e[39m\n")
+    @test contains(str, "SUCCESS")
+
+    testsuite = Dict(
+        "no color" => quote
+            print("Violets are ")
+            printstyled("blue"; color=:blue)
+        end
+    )
+    io = IOBuffer()
+    ioc = IOContext(io, :color => false)
+    runtests(ParallelTestRunner, String[]; testsuite, stdout=ioc, stderr=ioc)
+    str = String(take!(io))
+    @test contains(str, "Violets are blue\n")
+    @test contains(str, "SUCCESS")
+end
+
 end