From b4fca13fd5ce50550c20d61d01dac054b091bce0 Mon Sep 17 00:00:00 2001
From: Sergi Delgado Segura <sergi.delgado.s@gmail.com>
Date: Thu, 20 Jun 2019 14:59:19 +0100
Subject: [PATCH] Defines tests for the inspector

---
 pisa-btc/tests/appointment_tests.py | 138 ++++++++++++++++++++++++++++
 1 file changed, 138 insertions(+)
 create mode 100644 pisa-btc/tests/appointment_tests.py

diff --git a/pisa-btc/tests/appointment_tests.py b/pisa-btc/tests/appointment_tests.py
new file mode 100644
index 0000000..0bff9f1
--- /dev/null
+++ b/pisa-btc/tests/appointment_tests.py
@@ -0,0 +1,138 @@
+import logging
+from pisa.inspector import Inspector
+from conf import TEST_LOG_FILE
+from pisa import errors
+from utils.authproxy import AuthServiceProxy, JSONRPCException
+from conf import BTC_RPC_USER, BTC_RPC_PASSWD, BTC_RPC_HOST, BTC_RPC_PORT, SUPPORTED_HASH_FUNCTIONS, SUPPORTED_CIPHERS
+
+logging.basicConfig(format='%(asctime)s %(message)s', level=logging.INFO, handlers=[
+    logging.FileHandler(TEST_LOG_FILE)
+])
+
+appointment = {"locator": None, "start_time": None, "end_time": None, "dispute_delta": None,
+               "encrypted_blob": None, "cipher": None, "hash_function": None}
+
+bitcoin_cli = AuthServiceProxy("http://%s:%s@%s:%d" % (BTC_RPC_USER, BTC_RPC_PASSWD, BTC_RPC_HOST,
+                                                       BTC_RPC_PORT))
+
+try:
+    block_height = bitcoin_cli.getblockcount()
+
+except JSONRPCException as e:
+    logging.error("[Inspector] JSONRPCException. Error code {}".format(e))
+
+locators = [None, 0, 'A' * 31, "A" * 31 + "_"]
+start_times = [None, 0, '', 15.0, block_height - 10]
+end_times = [None, 0, '', 26.123, block_height - 11]
+dispute_deltas = [None, 0, '', 1.2, 30, -3]
+encrypted_blobs = [None, 0, '']
+ciphers = [None, 0, '', 'foo']
+hash_functions = [None, 0, '', 'foo']
+
+locators_rets = [errors.APPOINTMENT_EMPTY_FIELD, errors.APPOINTMENT_WRONG_FIELD_TYPE,
+                 errors.APPOINTMENT_WRONG_FIELD_SIZE, errors.APPOINTMENT_WRONG_FIELD_FORMAT]
+
+start_time_rets = [errors.APPOINTMENT_EMPTY_FIELD, errors.APPOINTMENT_FIELD_TOO_SMALL,
+                   errors.APPOINTMENT_WRONG_FIELD_TYPE, errors.APPOINTMENT_WRONG_FIELD_TYPE,
+                   errors.APPOINTMENT_FIELD_TOO_SMALL]
+
+end_time_rets = [errors.APPOINTMENT_EMPTY_FIELD, errors.APPOINTMENT_FIELD_TOO_SMALL,
+                 errors.APPOINTMENT_WRONG_FIELD_TYPE, errors.APPOINTMENT_WRONG_FIELD_TYPE,
+                 errors.APPOINTMENT_FIELD_TOO_SMALL]
+
+dispute_delta_rets = [errors.APPOINTMENT_EMPTY_FIELD, errors.APPOINTMENT_FIELD_TOO_SMALL,
+                      errors.APPOINTMENT_WRONG_FIELD_TYPE, errors.APPOINTMENT_WRONG_FIELD_TYPE,
+                      errors.APPOINTMENT_FIELD_TOO_BIG, errors.APPOINTMENT_FIELD_TOO_SMALL]
+
+encrypted_blob_rets = [errors.APPOINTMENT_EMPTY_FIELD, errors.APPOINTMENT_WRONG_FIELD_TYPE,
+                       errors.APPOINTMENT_WRONG_FIELD]
+
+cipher_rets = [errors.APPOINTMENT_EMPTY_FIELD, errors.APPOINTMENT_WRONG_FIELD_TYPE,
+               errors.APPOINTMENT_CIPHER_NOT_SUPPORTED, errors.APPOINTMENT_CIPHER_NOT_SUPPORTED]
+
+hash_function_rets = [errors.APPOINTMENT_EMPTY_FIELD, errors.APPOINTMENT_WRONG_FIELD_TYPE,
+                      errors.APPOINTMENT_HASH_FUNCTION_NOT_SUPPORTED, errors.APPOINTMENT_HASH_FUNCTION_NOT_SUPPORTED]
+
+inspector = Inspector(debug=True, logging=logging)
+
+print("Locator tests\n")
+for locator, ret in zip(locators, locators_rets):
+    appointment["locator"] = locator
+    r = inspector.inspect(appointment)
+
+    assert r[0] == ret
+    print(r)
+
+# Set locator to a 'valid' one
+appointment['locator'] = 'A' * 32
+
+print("\nStart time tests\n")
+for start_time, ret in zip(start_times, start_time_rets):
+    appointment["start_time"] = start_time
+    r = inspector.inspect(appointment)
+
+    assert r[0] == ret
+    print(r)
+# Setting the start time to some time in the future
+appointment['start_time'] = block_height + 10
+
+print("\nEnd time tests\n")
+for end_time, ret in zip(end_times, end_time_rets):
+    appointment["end_time"] = end_time
+    r = inspector.inspect(appointment)
+
+    assert r[0] == ret
+    print(r)
+
+# Setting the end time to something consistent with start time
+appointment['end_time'] = block_height + 30
+
+print("\nDelta tests\n")
+for dispute_delta, ret in zip(dispute_deltas, dispute_delta_rets):
+    appointment["dispute_delta"] = dispute_delta
+    r = inspector.inspect(appointment)
+
+    assert r[0] == ret
+    print(r)
+
+# Setting the a proper dispute delta
+appointment['dispute_delta'] = appointment['end_time'] - appointment['start_time']
+
+print("\nEncrypted blob tests\n")
+for encrypted_blob, ret in zip(encrypted_blobs, encrypted_blob_rets):
+    appointment["encrypted_blob"] = encrypted_blob
+    r = inspector.inspect(appointment)
+
+    assert r[0] == ret
+    print(r)
+
+# Setting the encrypted blob to something that may pass
+appointment['encrypted_blob'] = 'A' * 32
+
+print("\nCipher tests\n")
+for cipher, ret in zip(ciphers, cipher_rets):
+    appointment["cipher"] = cipher
+    r = inspector.inspect(appointment)
+
+    assert r[0] == ret
+    print(r)
+
+# Setting the cipher to the only supported one for now
+appointment['cipher'] = SUPPORTED_CIPHERS[0]
+
+print("\nHash function tests\n")
+for hash_function, ret in zip(hash_functions, hash_function_rets):
+    appointment["hash_function"] = hash_function
+    r = inspector.inspect(appointment)
+
+    assert r[0] == ret
+    print(r)
+
+# Setting the cipher to the only supported one for now
+appointment['hash_function'] = SUPPORTED_HASH_FUNCTIONS[0]
+
+r = inspector.inspect(appointment)
+assert r[0] == 0
+
+print("\nAll tests passed!")
+