Skip to content

Commit

Permalink
Expand unit testing to complete code coverage
Browse files Browse the repository at this point in the history
  • Loading branch information
jshcodes committed Jan 31, 2024
1 parent 1cef597 commit 5d08836
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions tests/test_real_time_response.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit 5d08836

Please sign in to comment.