aljaz/python-teos
1
0
Fork 0
mirror of https://github.com/aljazceru/python-teos.git synced 2025-12-18 14:44:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

Minor refactoring of test_watcher; revert change DEFAULT_PISA_API_SERVER to correct value

Browse Source
This commit is contained in:
Salvatore Ingala
2019-10-25 15:03:57 +08:00
parent 21b54f048e
commit df9a1173a5
2 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
apps/cli/__init__.py
View File

@@ -2,7 +2,7 @@ import logging
from .logger import Logger from .logger import Logger
# PISA-SERVER # PISA-SERVER
DEFAULT_PISA_API_SERVER = 'localhost' DEFAULT_PISA_API_SERVER = 'btc.pisa.watch'
DEFAULT_PISA_API_PORT = 9814 DEFAULT_PISA_API_PORT = 9814
# PISA-CLI # PISA-CLI

11
test/unit/test_watcher.py
View File

@@ -82,13 +82,14 @@ def create_appointments(n):
return appointments, locator_uuid_map, dispute_txs return appointments, locator_uuid_map, dispute_txs
def verify_signature(appointment, signature, pk): def is_signature_valid(appointment, signature, pk):
# verify the signature # verify the signature
try: try:
data = appointment.to_json().encode('utf-8') data = appointment.to_json().encode('utf-8')
pk.verify(signature, data, ec.ECDSA(hashes.SHA256())) pk.verify(signature, data, ec.ECDSA(hashes.SHA256()))
except InvalidSignature: except InvalidSignature:
assert False, "The appointment's signature is not correct" return False
return True
def test_init(watcher): def test_init(watcher):
@@ -104,7 +105,7 @@ def test_init(watcher):
def test_sign_appointment(watcher): def test_sign_appointment(watcher):
appointment, _ = generate_dummy_appointment() appointment, _ = generate_dummy_appointment()
signature = watcher.sign_appointment(appointment) signature = watcher.sign_appointment(appointment)
verify_signature(appointment, signature, public_key) assert is_signature_valid(appointment, signature, public_key)
def test_add_appointment(run_bitcoind, watcher): def test_add_appointment(run_bitcoind, watcher):
@@ -118,7 +119,7 @@ def test_add_appointment(run_bitcoind, watcher):
added_appointment, sig = watcher.add_appointment(appointment) added_appointment, sig = watcher.add_appointment(appointment)
assert added_appointment is True assert added_appointment is True
verify_signature(appointment, sig, public_key) assert is_signature_valid(appointment, sig, public_key)
def test_add_too_many_appointments(watcher): def test_add_too_many_appointments(watcher):
@@ -130,7 +131,7 @@ def test_add_too_many_appointments(watcher):
added_appointment, sig = watcher.add_appointment(appointment) added_appointment, sig = watcher.add_appointment(appointment)
assert added_appointment is True assert added_appointment is True
verify_signature(appointment, sig, public_key) assert is_signature_valid(appointment, sig, public_key)
appointment, dispute_tx = generate_dummy_appointment() appointment, dispute_tx = generate_dummy_appointment()
added_appointment, sig = watcher.add_appointment(appointment) added_appointment, sig = watcher.add_appointment(appointment)