From d0d55258c3c48178611919685861c8bc247e4bea Mon Sep 17 00:00:00 2001
From: Alexey Stratulat <alexey.stratulat@demlabs.net>
Date: Fri, 26 Jul 2019 18:00:04 +0700
Subject: [PATCH] [+] Added return value depending on whether the test passed.

---
 test/test_b58.py         |  5 +++--
 test/test_b64.py         |  2 ++
 test/test_iaes256_cbc.py | 13 +++----------
 test/test_oaes.py        | 10 +---------
 4 files changed, 9 insertions(+), 21 deletions(-)

diff --git a/test/test_b58.py b/test/test_b58.py
index aae03f30..fbbbf316 100644
--- a/test/test_b58.py
+++ b/test/test_b58.py
@@ -15,8 +15,9 @@ print ("Decoded data: \t")
 print(decrypt)
 out_data = pickle.loads(decrypt)
 if s == out_data:
-    print ("TEST 1. Encode/Decode base64 done")
+    print ("TEST 1. Encode/Decode base58 done")
 else:
-    print ("TEST 1. Encode/Decode base64 faild")
+    print ("TEST 1. Encode/Decode base58 faild")
+    sys.exit(1)
 
 sys.exit(0)
diff --git a/test/test_b64.py b/test/test_b64.py
index 574d9940..781b6d44 100644
--- a/test/test_b64.py
+++ b/test/test_b64.py
@@ -14,6 +14,7 @@ if bytes(s, "utf-8") == decrypt:
     print ("TEST 1. Encode/Decode base64 done")
 else:
     print ("TEST 1. Encode/Decode base64 faild")
+    sys.exit(1)
 
 print ("Test Base64 URLSAFE")
 u = "http://kelvin.foundation/"
@@ -23,6 +24,7 @@ if bytes(u, "utf-8") == decrypt_u:
      print ("TEST 2. Encode/Decode base64 urlsafe done")
 else:
      print ("TEST 2. Encode/Decode base64 urlsafe faild")
+     sys.exit(2)
 
 sys.exit(0)
 
diff --git a/test/test_iaes256_cbc.py b/test/test_iaes256_cbc.py
index 33216c3f..58fa69ee 100644
--- a/test/test_iaes256_cbc.py
+++ b/test/test_iaes256_cbc.py
@@ -17,19 +17,12 @@ decrypt = crypto.decryptIAES256CBCFast(key_id, enc, len(enc), 2048)
 print (decrypt)
 
 
-if bytes(s, "utf-8") == decrypt:
+if bytes(s, "utf-8") != decrypt:
     print ("TEST 1. Encode/Decode IAES256 CBC FAST done")
 else:
     print ("TEST 1. Encode/Decode IAES256 CBC FAST faild")
+    sys.exit(1)
+
 
-#print ("Test Base64 URLSAFE")
-#u = "http://kelvin.foundation/"
-#crypt_u = crypto.encodeBase64(bytes(u, "utf-8"), 2)
-#decrypt_u = crypto.decodeBase64(crypt_u)
-#if bytes(u, "utf-8") == decrypt_u:
-#     print ("TEST 2. Encode/Decode base64 urlsafe done")
-#else:
-#     print ("TEST 2. Encode/Decode base64 urlsafe faild")
-#
 sys.exit(0)
 
diff --git a/test/test_oaes.py b/test/test_oaes.py
index d0dead21..7a4d02f6 100644
--- a/test/test_oaes.py
+++ b/test/test_oaes.py
@@ -20,15 +20,7 @@ if bytes(s, "utf-8") == decrypt:
     print ("TEST 1. Encode/Decode OAES FAST done")
 else:
     print ("TEST 1. Encode/Decode OAES CBC FAST faild")
+    sys.exit(1)
 
-#print ("Test Base64 URLSAFE")
-#u = "http://kelvin.foundation/"
-#crypt_u = crypto.encodeBase64(bytes(u, "utf-8"), 2)
-#decrypt_u = crypto.decodeBase64(crypt_u)
-#if bytes(u, "utf-8") == decrypt_u:
-#     print ("TEST 2. Encode/Decode base64 urlsafe done")
-#else:
-#     print ("TEST 2. Encode/Decode base64 urlsafe faild")
-#
 sys.exit(0)
 
-- 
GitLab