From 0996ab2857c69a436e9d7090bf266568f4f0730b Mon Sep 17 00:00:00 2001 From: Clark Boylan Date: Thu, 19 Oct 2017 11:03:10 -0700 Subject: [PATCH] Make tests' encrypt_secret.py work with python3 This was not compatible with python3 due to encoding issues of the input and the out. Ensure we pass the input plaintext as bytes to the encryption routine and use the base64 module to convert the output to base64. Change-Id: Ie8b3a8e5c93544e448016829c1071240b68e8957 --- tests/encrypt_secret.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/encrypt_secret.py b/tests/encrypt_secret.py index 0b0cf19a1d..4bc514ac28 100644 --- a/tests/encrypt_secret.py +++ b/tests/encrypt_secret.py @@ -12,6 +12,7 @@ # License for the specific language governing permissions and limitations # under the License. +import base64 import sys import os @@ -27,8 +28,10 @@ def main(): private_key, public_key = \ encryption.deserialize_rsa_keypair(f.read()) - ciphertext = encryption.encrypt_pkcs1_oaep(sys.argv[1], public_key) - print(ciphertext.encode('base64')) + plaintext = sys.argv[1].encode('utf-8') + + ciphertext = encryption.encrypt_pkcs1_oaep(plaintext, public_key) + print(base64.b64encode(ciphertext).decode('utf-8')) if __name__ == '__main__':