From 742d124d6234befe8051f98e30c7a50698d34f33 Mon Sep 17 00:00:00 2001 From: Tom Ritter Date: Thu, 30 Jul 2020 14:20:21 -0500 Subject: [PATCH] Improve the MockTreeherderServer to allow unlimited responses for a url --- tests/mock_treeherder_server.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/mock_treeherder_server.py b/tests/mock_treeherder_server.py index 8026e053..0af6eddb 100644 --- a/tests/mock_treeherder_server.py +++ b/tests/mock_treeherder_server.py @@ -28,9 +28,10 @@ # x: push_id. from above in TRY_REVISIONS # y: the page. Unless a result is multi_page it will just be 1 # x: request number. The first time a document is requested, this will be 1, the second time, 2, etc + # specify 'A' to indicate this response should be used for all requests # rev_broken/good - '1_1_1': "treeherder_api_response_1.txt", - '1_2_1': "treeherder_api_response_2.txt", + '1_1_A': "treeherder_api_response_1.txt", + '1_2_A': "treeherder_api_response_2.txt", # testExistingJobSucceeded '2_1_1': "treeherder_api_response_jobs_still_running.txt", '2_1_2': "treeherder_api_response_all_succeeded.txt", @@ -68,7 +69,10 @@ def get_appropriate_filename(path): print("Checking for push_id", push_id, "page", page, "seen", seen_counter) if key not in PUSH_IDS: - assert False, "Could not find the key " + key + " in PUSH_IDS" + key = push_id + "_" + page + "_" + "A" + print("Response-specific key missing, checking for key ", key) + if key not in PUSH_IDS: + assert False, "Could not find either key in PUSH_IDS" return PUSH_IDS[key]