Make Simulator CLI Tests Less Flaky (#15163)

* make less flaky by adding finally

* add different path for each test

in case path delete fails
This commit is contained in:
Jack Nelson 2023-05-09 16:15:21 -04:00 committed by GitHub
parent 434cbc42dd
commit 432e881439
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -19,26 +19,25 @@ burn_address = "txch1qqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqm6ksh7qddh"
def test_every_simulator_command() -> None:
runner: CliRunner = CliRunner()
address = std_farming_address
start_result: Result = runner.invoke(cli, ["dev", "sim", "-n", "ci_test", "create", "-bm", mnemonic])
simulator_name = "ci_test"
start_result: Result = runner.invoke(cli, ["dev", "sim", "-n", simulator_name, "create", "-bm", mnemonic])
assert start_result.exit_code == 0
assert f"Farming & Prefarm reward address: {address}" in start_result.output
assert "chia_full_node_simulator: started" in start_result.output
assert "Genesis block generated, exiting." in start_result.output
try:
# run all tests
run_all_tests(runner, address)
except AssertionError:
stop_simulator(runner)
raise
stop_simulator(runner)
run_all_tests(runner, address, simulator_name)
finally:
stop_simulator(runner, simulator_name)
def test_custom_farming_address() -> None:
runner: CliRunner = CliRunner()
address = burn_address
simulator_name = "ci_address_test"
start_result: Result = runner.invoke(
cli, ["dev", "sim", "-n", "ci_test", "create", "-bm", mnemonic, "--reward-address", address]
cli, ["dev", "sim", "-n", simulator_name, "create", "-bm", mnemonic, "--reward-address", address]
)
assert start_result.exit_code == 0
assert f"Farming & Prefarm reward address: {address}" in start_result.output
@ -47,31 +46,28 @@ def test_custom_farming_address() -> None:
try:
# just run status test
_test_sim_status(runner, address)
except AssertionError:
stop_simulator(runner)
raise
stop_simulator(runner)
_test_sim_status(runner, address, simulator_name)
finally:
stop_simulator(runner, simulator_name)
def stop_simulator(runner: CliRunner) -> None:
def stop_simulator(runner: CliRunner, simulator_name: str) -> None:
"""Stop simulator."""
result: Result = runner.invoke(cli, ["dev", "sim", "-n", "ci_test", "-n", "ci_test", "stop", "-d"])
result: Result = runner.invoke(cli, ["dev", "sim", "-n", simulator_name, "stop", "-d"])
assert result.exit_code == 0
assert "chia_full_node_simulator: Stopped\nDaemon stopped\n" == result.output
rmtree(SIMULATOR_ROOT_PATH / "ci_test")
rmtree(SIMULATOR_ROOT_PATH / simulator_name)
def run_all_tests(runner: CliRunner, address: str) -> None:
def run_all_tests(runner: CliRunner, address: str, simulator_name: str) -> None:
"""Run all tests."""
_test_sim_status(runner, address)
_test_farm_and_revert_block(runner, address)
_test_sim_status(runner, address, simulator_name)
_test_farm_and_revert_block(runner, address, simulator_name)
def _test_sim_status(runner: CliRunner, address: str) -> None:
def _test_sim_status(runner: CliRunner, address: str, simulator_name: str) -> None:
# show everything
result: Result = runner.invoke(cli, ["dev", "sim", "-n", "ci_test", "status", "--show-key", "-cia"])
result: Result = runner.invoke(cli, ["dev", "sim", "-n", simulator_name, "status", "--show-key", "-cia"])
assert result.exit_code == 0
# asserts are grouped by arg
assert f"Fingerprint: {fingerprint}" and f"Mnemonic seed (24 secret words):\n{mnemonic}" in result.output # -k
@ -87,43 +83,45 @@ def _test_sim_status(runner: CliRunner, address: str) -> None:
assert "Coin Amount: 2625000000000000000 mojo" in result.output # -ic
def _test_farm_and_revert_block(runner: CliRunner, address: str) -> None:
def _test_farm_and_revert_block(runner: CliRunner, address: str, simulator_name: str) -> None:
# make 5 blocks
five_blocks_result: Result = runner.invoke(cli, ["dev", "sim", "-n", "ci_test", "farm", "-b", "5", "-a", address])
five_blocks_result: Result = runner.invoke(
cli, ["dev", "sim", "-n", simulator_name, "farm", "-b", "5", "-a", address]
)
assert five_blocks_result.exit_code == 0
assert "Farmed 5 Transaction blocks" in five_blocks_result.output
# check that height increased
five_blocks_check: Result = runner.invoke(cli, ["dev", "sim", "-n", "ci_test", "status"])
five_blocks_check: Result = runner.invoke(cli, ["dev", "sim", "-n", simulator_name, "status"])
assert five_blocks_check.exit_code == 0
assert "Height: 6" in five_blocks_check.output
# do a reorg, 3 blocks back, 2 blocks forward, height now 8
reorg_result: Result = runner.invoke(cli, ["dev", "sim", "-n", "ci_test", "revert", "-b", "3", "-n", "2"])
reorg_result: Result = runner.invoke(cli, ["dev", "sim", "-n", simulator_name, "revert", "-b", "3", "-n", "2"])
assert reorg_result.exit_code == 0
assert "Block: 3 and above " and "Block Height is now: 8" in reorg_result.output
# check that height changed by 2
reorg_check: Result = runner.invoke(cli, ["dev", "sim", "-n", "ci_test", "status"])
reorg_check: Result = runner.invoke(cli, ["dev", "sim", "-n", simulator_name, "status"])
assert reorg_check.exit_code == 0
assert "Height: 8" in reorg_check.output
# do a forceful reorg 4 blocks back
forced_reorg_result: Result = runner.invoke(cli, ["dev", "sim", "-n", "ci_test", "revert", "-b", "4", "-fd"])
forced_reorg_result: Result = runner.invoke(cli, ["dev", "sim", "-n", simulator_name, "revert", "-b", "4", "-fd"])
assert forced_reorg_result.exit_code == 0
assert "Block: 8 and above were successfully deleted" and "Block Height is now: 4" in forced_reorg_result.output
# check that height changed by 4
forced_reorg_check: Result = runner.invoke(cli, ["dev", "sim", "-n", "ci_test", "status"])
forced_reorg_check: Result = runner.invoke(cli, ["dev", "sim", "-n", simulator_name, "status"])
assert forced_reorg_check.exit_code == 0
assert "Height: 4" in forced_reorg_check.output
# test chain reset to genesis
genesis_reset_result: Result = runner.invoke(cli, ["dev", "sim", "-n", "ci_test", "revert", "-fd", "--reset"])
genesis_reset_result: Result = runner.invoke(cli, ["dev", "sim", "-n", simulator_name, "revert", "-fd", "--reset"])
assert genesis_reset_result.exit_code == 0
assert "Block: 2 and above were successfully deleted" and "Block Height is now: 1" in genesis_reset_result.output
# check that height changed to 1
genesis_reset_check: Result = runner.invoke(cli, ["dev", "sim", "-n", "ci_test", "status"])
genesis_reset_check: Result = runner.invoke(cli, ["dev", "sim", "-n", simulator_name, "status"])
assert genesis_reset_check.exit_code == 0
assert "Height: 1" in genesis_reset_check.output