From 14afee9ff32c16aa5614d3f4c86b44076c0bed0f Mon Sep 17 00:00:00 2001
From: Fredrik Ekre <ekrefredrik@gmail.com>
Date: Sun, 5 Apr 2020 01:10:24 +0200
Subject: [PATCH] Some more tests for exotic mime types.

---
 Project.toml     |  3 ++-
 test/runtests.jl | 42 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 44 insertions(+), 1 deletion(-)

diff --git a/Project.toml b/Project.toml
index 9f552d1..b0d0fc1 100644
--- a/Project.toml
+++ b/Project.toml
@@ -12,7 +12,8 @@ JSON = "0.18, 0.19, 0.20, 0.21"
 julia = "1"
 
 [extras]
+DisplayAs = "0b91fe84-8a4c-11e9-3e1d-67c38462b6d6"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["Test"]
+test = ["DisplayAs", "Test"]
diff --git a/test/runtests.jl b/test/runtests.jl
index f5ca7b9..924cdf8 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -958,6 +958,48 @@ end end
             Literate.notebook(inputfile, outdir)
             notebook = read(joinpath(outdir, "inputfile.ipynb"), String)
             @test occursin("\"application/vnd.vegalite.v2+json\":", notebook)
+
+
+            # Capturing output of more exotic types when executing a notebook
+            script = """
+                using DisplayAs
+                struct X end
+                Base.show(io::IO, ::MIME"text/plain", ::X) = print(io, "X as text/plain")
+                Base.show(io::IO, ::MIME"image/svg+xml", ::X) = print(io, "X as image/svg+xml")
+                Base.show(io::IO, ::MIME"image/png", ::X) = print(io, "X as image/png")
+                Base.show(io::IO, ::MIME"image/jpeg", ::X) = print(io, "X as image/jpeg")
+                Base.show(io::IO, ::MIME"text/markdown", ::X) = print(io, "X as text/markdown")
+                Base.show(io::IO, ::MIME"text/html", ::X) = print(io, "X as text/html")
+                Base.show(io::IO, ::MIME"text/latex", ::X) = print(io, "X as text/latex")
+                Base.show(io::IO, ::MIME"application/x-latex", ::X) = print(io, "X as application/x-latex")
+                Base.show(io::IO, ::MIME"application/vnd.vegalite.v2+json", ::X) = print(io, "{\\"X\\": \\"as application/vnd.vegalite.v2+json\\"}")
+
+                # DisplayAs does not define the following
+                Base.show(io::IO, ::MIME"application/x-latex", s::DisplayAs.Showable{>:MIME"application/x-latex"}) =
+                    show(io, MIME"application/x-latex"(), s.content)
+                Base.show(io::IO, ::MIME"application/vnd.vegalite.v2+json", s::DisplayAs.Showable{>:MIME"application/vnd.vegalite.v2+json"}) =
+                    show(io, MIME"application/vnd.vegalite.v2+json"(), s.content)
+                xs = []
+                for mime in [
+                             MIME"text/plain",
+                             MIME"image/svg+xml",
+                             MIME"image/png",
+                             MIME"image/jpeg",
+                             MIME"text/markdown",
+                             MIME"text/html",
+                             MIME"text/latex",
+                             MIME"application/x-latex",
+                             MIME"application/vnd.vegalite.v2+json"
+                            ]
+                    x = X() |> DisplayAs.Showable{mime} |> DisplayAs.Text
+                    push!(xs, x)
+                end
+                """
+            for i in 1:9
+                script *= "\n#-\nxs[$i]"
+            end
+            write(inputfile, script)
+            Literate.notebook(inputfile, outdir)
         end
     end
 end end