1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90
|
# This file is a part of Julia. License is MIT: https://julialang.org/license
using Test, Random
import Base64:
Base64EncodePipe,
base64encode,
Base64DecodePipe,
base64decode,
stringmime
const inputText = "Man is distinguished, not only by his reason, but by this singular passion from other animals, which is a lust of the mind, that by a perseverance of delight in the continued and indefatigable generation of knowledge, exceeds the short vehemence of any carnal pleasure."
const encodedMaxLine76 = """
TWFuIGlzIGRpc3Rpbmd1aXNoZWQsIG5vdCBvbmx5IGJ5IGhpcyByZWFzb24sIGJ1dCBieSB0aGlz
IHNpbmd1bGFyIHBhc3Npb24gZnJvbSBvdGhlciBhbmltYWxzLCB3aGljaCBpcyBhIGx1c3Qgb2Yg
dGhlIG1pbmQsIHRoYXQgYnkgYSBwZXJzZXZlcmFuY2Ugb2YgZGVsaWdodCBpbiB0aGUgY29udGlu
dWVkIGFuZCBpbmRlZmF0aWdhYmxlIGdlbmVyYXRpb24gb2Yga25vd2xlZGdlLCBleGNlZWRzIHRo
ZSBzaG9ydCB2ZWhlbWVuY2Ugb2YgYW55IGNhcm5hbCBwbGVhc3VyZS4="""
@testset "Examples" begin
# Encode and decode
fname = tempname()
open(fname, "w") do f
opipe = Base64EncodePipe(f)
write(opipe,inputText)
@test close(opipe) === nothing
end
open(fname, "r") do f
ipipe = Base64DecodePipe(f)
@test read(ipipe, String) == inputText
@test close(ipipe) === nothing
end
rm(fname)
# Byte-by-byte encode and decode.
buf = IOBuffer()
pipe = Base64EncodePipe(buf)
for char in inputText
write(pipe, UInt8(char))
end
close(pipe)
pipe = Base64DecodePipe(IOBuffer(take!(buf)))
decoded = UInt8[]
while !eof(pipe)
push!(decoded, read(pipe, UInt8))
end
@test String(decoded) == inputText
# Encode to string and decode
@test String(base64decode(base64encode(inputText))) == inputText
# Decode with max line chars = 76 and padding
ipipe = Base64DecodePipe(IOBuffer(encodedMaxLine76))
@test read(ipipe, String) == inputText
# Decode with max line chars = 76 and no padding
#ipipe = Base64DecodePipe(IOBuffer(encodedMaxLine76[1:end-1]))
#@test read(ipipe, String) == inputText
# Decode with two padding characters ("==")
ipipe = Base64DecodePipe(IOBuffer(string(encodedMaxLine76[1:end-2],"==")))
@test read(ipipe, String) == inputText[1:end-1]
# Test incorrect format
ipipe = Base64DecodePipe(IOBuffer(encodedMaxLine76[1:end-3]))
@test_throws ArgumentError read(ipipe, String)
# issue #21314
@test base64decode(chomp("test")) == base64decode("test")
end
@testset "Random data" begin
mt = MersenneTwister(1234)
for _ in 1:1000
data = rand(mt, UInt8, rand(0:300))
@test hash(base64decode(base64encode(data))) == hash(data)
end
end
struct PNG end
Base.show(io::IO, ::MIME"image/png", ::PNG) = print(io, "PNG")
@testset "stringmime" begin
@test stringmime("text/plain", [1 2;3 4]) == repr("text/plain", [1 2;3 4])
@test stringmime("text/html", "raw html data") == "raw html data"
@test stringmime("text/plain", "string") == "\"string\""
@test stringmime("image/png", UInt8[2,3,4,7]) == "AgMEBw=="
@test stringmime("text/plain", 3.141592653589793, context=:compact=>true) == "3.14159"
@test stringmime("image/png", PNG()) == stringmime(MIME("image/png"), PNG()) == "UE5H"
end
|