From 44d819a16ddfd96712adcd159494c7f3457d8be8 Mon Sep 17 00:00:00 2001 From: Maddie Ford Date: Sun, 6 Aug 2023 09:56:38 -0700 Subject: [PATCH] Import datetime --- tests_e2e/tests/agent_status/agent_status.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests_e2e/tests/agent_status/agent_status.py b/tests_e2e/tests/agent_status/agent_status.py index bd3408d837..5402528791 100644 --- a/tests_e2e/tests/agent_status/agent_status.py +++ b/tests_e2e/tests/agent_status/agent_status.py @@ -144,14 +144,14 @@ def run(self): vm = VirtualMachineClient(self._context.vm) - timeout = datetime.datetime.now() + datetime.timedelta(minutes=5) + timeout = datetime.now() + datetime.timedelta(minutes=5) instance_view_exception = None status_updated = 0 prev_status_timestamp = None prev_gs_processed_log = None # Retry validating agent status updates 3 times with timeout of 5 minutes - while datetime.datetime.now() <= timeout and status_updated < 3: + while datetime.now() <= timeout and status_updated < 3: instance_view = vm.get_instance_view() log.info("") log.info(