mirror of
https://github.com/aljazceru/python-teos.git
synced 2025-12-17 14:14:22 +01:00
Updates e2e to match 635da6447d887d7df9387e1eda63eda2e9a7d577
This commit is contained in:
@@ -3,8 +3,6 @@ import binascii
|
|||||||
from time import sleep
|
from time import sleep
|
||||||
from riemann.tx import Tx
|
from riemann.tx import Tx
|
||||||
|
|
||||||
from teos import HOST, PORT
|
|
||||||
|
|
||||||
from cli import teos_cli, DATA_DIR, DEFAULT_CONF, CONF_FILE_NAME
|
from cli import teos_cli, DATA_DIR, DEFAULT_CONF, CONF_FILE_NAME
|
||||||
|
|
||||||
import common.cryptographer
|
import common.cryptographer
|
||||||
@@ -26,10 +24,14 @@ from test.teos.e2e.conftest import (
|
|||||||
cli_config = get_config(DATA_DIR, CONF_FILE_NAME, DEFAULT_CONF)
|
cli_config = get_config(DATA_DIR, CONF_FILE_NAME, DEFAULT_CONF)
|
||||||
common.cryptographer.logger = Logger(actor="Cryptographer", log_name_prefix="")
|
common.cryptographer.logger = Logger(actor="Cryptographer", log_name_prefix="")
|
||||||
|
|
||||||
# We'll use teos_cli to add appointments. The expected input format is a list of arguments with a json-encoded
|
# # We'll use teos_cli to add appointments. The expected input format is a list of arguments with a json-encoded
|
||||||
# appointment
|
# # appointment
|
||||||
teos_cli.teos_api_server = "http://{}".format(HOST)
|
# teos_cli.teos_api_server = "http://{}".format(HOST)
|
||||||
teos_cli.teos_api_port = PORT
|
# teos_cli.teos_api_port = PORT
|
||||||
|
|
||||||
|
teos_base_endpoint = "http://{}:{}".format(cli_config.get("TEOS_SERVER"), cli_config.get("TEOS_PORT"))
|
||||||
|
teos_add_appointment_endpoint = teos_base_endpoint
|
||||||
|
teos_get_appointment_endpoint = teos_base_endpoint + "/get_appointment"
|
||||||
|
|
||||||
# Run teosd
|
# Run teosd
|
||||||
teosd_process = run_teosd()
|
teosd_process = run_teosd()
|
||||||
@@ -44,7 +46,7 @@ def broadcast_transaction_and_mine_block(bitcoin_cli, commitment_tx, addr):
|
|||||||
def get_appointment_info(locator):
|
def get_appointment_info(locator):
|
||||||
# Check that the justice has been triggered (the appointment has moved from Watcher to Responder)
|
# Check that the justice has been triggered (the appointment has moved from Watcher to Responder)
|
||||||
sleep(1) # Let's add a bit of delay so the state can be updated
|
sleep(1) # Let's add a bit of delay so the state can be updated
|
||||||
return teos_cli.get_appointment(locator, cli_config)
|
return teos_cli.get_appointment(locator, teos_get_appointment_endpoint)
|
||||||
|
|
||||||
|
|
||||||
def test_appointment_life_cycle(bitcoin_cli, create_txs):
|
def test_appointment_life_cycle(bitcoin_cli, create_txs):
|
||||||
@@ -53,7 +55,7 @@ def test_appointment_life_cycle(bitcoin_cli, create_txs):
|
|||||||
appointment_data = build_appointment_data(bitcoin_cli, commitment_tx_id, penalty_tx)
|
appointment_data = build_appointment_data(bitcoin_cli, commitment_tx_id, penalty_tx)
|
||||||
locator = compute_locator(commitment_tx_id)
|
locator = compute_locator(commitment_tx_id)
|
||||||
|
|
||||||
assert teos_cli.add_appointment([json.dumps(appointment_data)], cli_config) is True
|
assert teos_cli.add_appointment([json.dumps(appointment_data)], teos_add_appointment_endpoint, cli_config) is True
|
||||||
|
|
||||||
appointment_info = get_appointment_info(locator)
|
appointment_info = get_appointment_info(locator)
|
||||||
assert appointment_info is not None
|
assert appointment_info is not None
|
||||||
@@ -103,7 +105,7 @@ def test_appointment_malformed_penalty(bitcoin_cli, create_txs):
|
|||||||
appointment_data = build_appointment_data(bitcoin_cli, commitment_tx_id, mod_penalty_tx.hex())
|
appointment_data = build_appointment_data(bitcoin_cli, commitment_tx_id, mod_penalty_tx.hex())
|
||||||
locator = compute_locator(commitment_tx_id)
|
locator = compute_locator(commitment_tx_id)
|
||||||
|
|
||||||
assert teos_cli.add_appointment([json.dumps(appointment_data)], cli_config) is True
|
assert teos_cli.add_appointment([json.dumps(appointment_data)], teos_add_appointment_endpoint, cli_config) is True
|
||||||
|
|
||||||
# Broadcast the commitment transaction and mine a block
|
# Broadcast the commitment transaction and mine a block
|
||||||
new_addr = bitcoin_cli.getnewaddress()
|
new_addr = bitcoin_cli.getnewaddress()
|
||||||
@@ -141,7 +143,7 @@ def test_appointment_wrong_key(bitcoin_cli, create_txs):
|
|||||||
data = {"appointment": appointment.to_dict(), "signature": signature, "public_key": hex_pk_der.decode("utf-8")}
|
data = {"appointment": appointment.to_dict(), "signature": signature, "public_key": hex_pk_der.decode("utf-8")}
|
||||||
|
|
||||||
# Send appointment to the server.
|
# Send appointment to the server.
|
||||||
response = teos_cli.post_appointment(data, cli_config)
|
response = teos_cli.post_appointment(data, teos_add_appointment_endpoint)
|
||||||
response_json = teos_cli.process_post_appointment_response(response)
|
response_json = teos_cli.process_post_appointment_response(response)
|
||||||
|
|
||||||
# Check that the server has accepted the appointment
|
# Check that the server has accepted the appointment
|
||||||
@@ -177,8 +179,8 @@ def test_two_identical_appointments(bitcoin_cli, create_txs):
|
|||||||
locator = compute_locator(commitment_tx_id)
|
locator = compute_locator(commitment_tx_id)
|
||||||
|
|
||||||
# Send the appointment twice
|
# Send the appointment twice
|
||||||
assert teos_cli.add_appointment([json.dumps(appointment_data)], cli_config) is True
|
assert teos_cli.add_appointment([json.dumps(appointment_data)], teos_add_appointment_endpoint, cli_config) is True
|
||||||
assert teos_cli.add_appointment([json.dumps(appointment_data)], cli_config) is True
|
assert teos_cli.add_appointment([json.dumps(appointment_data)], teos_add_appointment_endpoint, cli_config) is True
|
||||||
|
|
||||||
# Broadcast the commitment transaction and mine a block
|
# Broadcast the commitment transaction and mine a block
|
||||||
new_addr = bitcoin_cli.getnewaddress()
|
new_addr = bitcoin_cli.getnewaddress()
|
||||||
@@ -211,8 +213,8 @@ def test_two_appointment_same_locator_different_penalty(bitcoin_cli, create_txs)
|
|||||||
appointment2_data = build_appointment_data(bitcoin_cli, commitment_tx_id, penalty_tx2)
|
appointment2_data = build_appointment_data(bitcoin_cli, commitment_tx_id, penalty_tx2)
|
||||||
locator = compute_locator(commitment_tx_id)
|
locator = compute_locator(commitment_tx_id)
|
||||||
|
|
||||||
assert teos_cli.add_appointment([json.dumps(appointment1_data)], cli_config) is True
|
assert teos_cli.add_appointment([json.dumps(appointment1_data)], teos_add_appointment_endpoint, cli_config) is True
|
||||||
assert teos_cli.add_appointment([json.dumps(appointment2_data)], cli_config) is True
|
assert teos_cli.add_appointment([json.dumps(appointment2_data)], teos_add_appointment_endpoint, cli_config) is True
|
||||||
|
|
||||||
# Broadcast the commitment transaction and mine a block
|
# Broadcast the commitment transaction and mine a block
|
||||||
new_addr = bitcoin_cli.getnewaddress()
|
new_addr = bitcoin_cli.getnewaddress()
|
||||||
@@ -239,7 +241,7 @@ def test_appointment_shutdown_teos_trigger_back_online(create_txs, bitcoin_cli):
|
|||||||
appointment_data = build_appointment_data(bitcoin_cli, commitment_tx_id, penalty_tx)
|
appointment_data = build_appointment_data(bitcoin_cli, commitment_tx_id, penalty_tx)
|
||||||
locator = compute_locator(commitment_tx_id)
|
locator = compute_locator(commitment_tx_id)
|
||||||
|
|
||||||
assert teos_cli.add_appointment([json.dumps(appointment_data)], cli_config) is True
|
assert teos_cli.add_appointment([json.dumps(appointment_data)], teos_add_appointment_endpoint, cli_config) is True
|
||||||
|
|
||||||
# Restart teos
|
# Restart teos
|
||||||
teosd_process.terminate()
|
teosd_process.terminate()
|
||||||
@@ -277,7 +279,7 @@ def test_appointment_shutdown_teos_trigger_while_offline(create_txs, bitcoin_cli
|
|||||||
appointment_data = build_appointment_data(bitcoin_cli, commitment_tx_id, penalty_tx)
|
appointment_data = build_appointment_data(bitcoin_cli, commitment_tx_id, penalty_tx)
|
||||||
locator = compute_locator(commitment_tx_id)
|
locator = compute_locator(commitment_tx_id)
|
||||||
|
|
||||||
assert teos_cli.add_appointment([json.dumps(appointment_data)], cli_config) is True
|
assert teos_cli.add_appointment([json.dumps(appointment_data)], teos_add_appointment_endpoint, cli_config) is True
|
||||||
|
|
||||||
# Check that the appointment is still in the Watcher
|
# Check that the appointment is still in the Watcher
|
||||||
appointment_info = get_appointment_info(locator)
|
appointment_info = get_appointment_info(locator)
|
||||||
|
|||||||
Reference in New Issue
Block a user