From 9ca3218a6dad49e22c236e4fa309c3c281b9668a Mon Sep 17 00:00:00 2001
From: Timothee Cour <timothee.cour2@gmail.com>
Date: Fri, 7 May 2021 17:40:16 -0700
Subject: [PATCH] improve tbase64.nim (#17964)

---
 tests/stdlib/tbase64.nim | 18 ++++++------------
 1 file changed, 6 insertions(+), 12 deletions(-)

diff --git a/tests/stdlib/tbase64.nim b/tests/stdlib/tbase64.nim
index 19b1264373..491ec5db0b 100644
--- a/tests/stdlib/tbase64.nim
+++ b/tests/stdlib/tbase64.nim
@@ -1,14 +1,11 @@
 discard """
-  output: '''YQ=='''
-  nimout: '''YQ=='''
+  targets: "c js"
 """
-import base64
 
-import base64
-static: echo encode("a")
-echo encode("a")
+import std/base64
 
-proc main() =
+template main() =
+  doAssert encode("a") == "YQ=="
   doAssert encode("Hello World") == "SGVsbG8gV29ybGQ="
   doAssert encode("leasure.") == "bGVhc3VyZS4="
   doAssert encode("easure.") == "ZWFzdXJlLg=="
@@ -38,11 +35,7 @@ proc main() =
     doAssert decode(encodeMIME(t, lineLen=40)) == t
     doAssert decode(encodeMIME(t, lineLen=76)) == t
 
-  const invalid = "SGVsbG\x008gV29ybGQ="
-  try:
-    doAssert decode(invalid) == "will throw error"
-  except ValueError:
-    discard
+  doAssertRaises(ValueError): discard decode("SGVsbG\x008gV29ybGQ=")
 
   block base64urlSafe:
     doAssert encode("c\xf7>", safe = true) == "Y_c-"
@@ -59,4 +52,5 @@ proc main() =
     doAssert encode("", safe = true) == ""
     doAssert encode("the quick brown dog jumps over the lazy fox", safe = true) == "dGhlIHF1aWNrIGJyb3duIGRvZyBqdW1wcyBvdmVyIHRoZSBsYXp5IGZveA=="
 
+static: main()
 main()