Merge pull request #340 from buffer/atob

atob() function minor improvement
buffer · Nov 16, 2022 · 167fa76 · 167fa76
2 parents c2c5fd2 + ac5e3cd
commit 167fa76
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/tests/functional/test_misc_ie90.py b/tests/functional/test_misc_ie90.py
@@ -1266,7 +1266,7 @@ def test_testPlayStateChange(self, caplog):
     def test_testAtob(self, caplog):
         sample   = os.path.join(self.misc_path, "testAtob.html")
         expected = ['Encoded String: SGVsbG8gV29ybGQ=',
-                    'Decoded String: Hello World']
+                    'Decoded String: b\'Hello World\'']
 
         self.do_perform_test(caplog, sample, expected)
 

diff --git a/thug/DOM/Window.py b/thug/DOM/Window.py
@@ -1056,10 +1056,7 @@ def atob(self, s):
         """
         The atob method decodes a base-64 encoded string
         """
-        if isinstance(s, str):
-            s = s.encode()
-
-        return base64.b64decode(s)
+        return str(base64.b64decode(s))
 
     def btoa(self, s):
         """