Skip to content

Commit 14a494d

Browse files
authored
add more tests, randomize taskChainId (#17)
1 parent a52c0e8 commit 14a494d

File tree

2 files changed

+41
-2
lines changed

2 files changed

+41
-2
lines changed

enochecker_test/tests.py

Lines changed: 40 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
global_round_id = 0
1717
FLAG_REGEX = r"ENO[A-Za-z0-9+\/=]{48}"
1818
REQUEST_TIMEOUT = 10
19+
CHAIN_ID_PREFIX = secrets.token_hex(20)
1920

2021

2122
@pytest.fixture
@@ -111,7 +112,9 @@ def _create_request_message(
111112
prefix = "noise"
112113
elif method == "exploit":
113114
prefix = "exploit"
114-
task_chain_id = f"{prefix}_s0_r{round_id}_t0_i{unique_variant_index}"
115+
task_chain_id = (
116+
f"{CHAIN_ID_PREFIX}_{prefix}_s0_r{round_id}_t0_i{unique_variant_index}"
117+
)
115118

116119
return CheckerTaskMessage(
117120
task_id=round_id,
@@ -375,6 +378,32 @@ def test_getflag(round_id, flag_id, service_address, checker_url):
375378
_test_getflag(flag, round_id, flag_id, service_address, checker_url)
376379

377380

381+
def test_getflag_wrong_flag(round_id, flag_id, service_address, checker_url):
382+
flag = generate_dummyflag()
383+
_test_putflag(flag, round_id, flag_id, service_address, checker_url)
384+
wrong_flag = generate_dummyflag()
385+
_test_getflag(
386+
wrong_flag,
387+
round_id,
388+
flag_id,
389+
service_address,
390+
checker_url,
391+
expected_result=CheckerTaskResult.MUMBLE,
392+
)
393+
394+
395+
def test_getflag_without_putflag(round_id, flag_id, service_address, checker_url):
396+
flag = generate_dummyflag()
397+
_test_getflag(
398+
flag,
399+
round_id,
400+
flag_id,
401+
service_address,
402+
checker_url,
403+
expected_result=CheckerTaskResult.MUMBLE,
404+
)
405+
406+
378407
def test_getflag_multiplied(
379408
round_id, flag_id_multiplied, flag_variants, service_address, checker_url
380409
):
@@ -442,6 +471,16 @@ def test_getnoise(round_id, noise_id, service_address, checker_url):
442471
_test_getnoise(round_id, noise_id, service_address, checker_url)
443472

444473

474+
def test_getnoise_without_putnoise(round_id, noise_id, service_address, checker_url):
475+
_test_getnoise(
476+
round_id,
477+
noise_id,
478+
service_address,
479+
checker_url,
480+
expected_result=CheckerTaskResult.MUMBLE,
481+
)
482+
483+
445484
def test_getnoise_multiplied(
446485
round_id, noise_id_multiplied, noise_variants, service_address, checker_url
447486
):

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
setuptools.setup(
1111
name="enochecker_test",
12-
version="0.5.1",
12+
version="0.6.0",
1313
author="ldruschk",
1414
author_email="[email protected]",
1515
description="Library to help testing checker scripts based on enochecker",

0 commit comments

Comments
 (0)