clean up

Author: sophia-chen-ttd

tests/test_refresh_keys_util.py

@@ -17,7 +17,7 @@ def _make_post_response(self, request_data, response_payload):
         payload += nonce
         payload += response_payload
         encrypted_payload = _encrypt_gcm(payload, None, client_secret_bytes)
-        encrypted_string = base64.b64encode(encrypted_payload).decode('utf-8')
+        encrypted_string = base64.b64encode(encrypted_payload)
         return Uid2Response.from_string(encrypted_string)
 
     def _get_post_refresh_keys_response(self, base_url, path, headers, envelope):

uid2_client/request_response_util.py

@@ -51,8 +51,7 @@ def parse_response(secret_key, uid2_response: Uid2Response, nonce):
         return _decrypt_payload(secret_key, uid2_response.as_bytes, nonce)
     else:
         encrypted_string = base64.b64decode(uid2_response.as_string)
-        decrypted_string = _decrypt_payload(secret_key, encrypted_string, nonce).decode('utf-8')
-        return decrypted_string
+        return _decrypt_payload(secret_key, encrypted_string, nonce)
 
 def _decrypt_payload(secret_key, encrypted, nonce):
     payload = _decrypt_gcm(encrypted, secret_key)