From 5d088360588b4ec4f5fcafa01f77f1f5a9969286 Mon Sep 17 00:00:00 2001 From: Joshua Hiller Date: Mon, 29 Jan 2024 18:45:19 -0500 Subject: [PATCH] Expand unit testing to complete code coverage --- tests/test_real_time_response.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/test_real_time_response.py b/tests/test_real_time_response.py index 6d2eb805b..fc2684a81 100644 --- a/tests/test_real_time_response.py +++ b/tests/test_real_time_response.py @@ -49,10 +49,12 @@ def rtr_session_tester(self): # pytest.skip("Race condition met, skipping.") if aid_to_check: - result = falcon.RTR_InitSession(body={"device_id": aid_to_check}) + result = falcon.batch_init_sessions(host_ids=aid_to_check) if "resources" in result["body"]: if result["body"]["resources"]: - session_id = result["body"]["resources"][0]["session_id"] + session_id = result["body"]["resources"][aid_to_check]["session_id"] + batch_id = result["body"].get("combined") + falcon.batch_get_command(batch_id=batch_id, file_path="/tmp/testfile.txt") if falcon.RTR_DeleteSession(session_id=session_id)["status_code"] in AllowedResponses: returned = True else: