Skip to content

Commit 7e5785d

Browse files
committed
Fix test_cloudwatch_task_handler
1 parent a974979 commit 7e5785d

File tree

1 file changed

+18
-16
lines changed

1 file changed

+18
-16
lines changed

providers/amazon/tests/unit/amazon/aws/log/test_cloudwatch_task_handler.py

Lines changed: 18 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -153,23 +153,9 @@ def test_log_message(self):
153153
assert metadata == [
154154
f"Reading remote log from Cloudwatch log_group: log_group_name log_stream: {stream_name}"
155155
]
156-
assert logs == ['[2025-03-27T21:58:01Z] {"foo": "bar", "event": "Hi", "level": "info"}']
157-
158-
def test_event_to_str(self):
159-
handler = self.subject
160-
current_time = int(time.time()) * 1000
161-
events = [
162-
{"timestamp": current_time - 2000, "message": "First"},
163-
{"timestamp": current_time - 1000, "message": "Second"},
164-
{"timestamp": current_time, "message": "Third"},
165-
]
166-
assert [handler._event_to_str(event) for event in events] == (
167-
[
168-
f"[{get_time_str(current_time - 2000)}] First",
169-
f"[{get_time_str(current_time - 1000)}] Second",
170-
f"[{get_time_str(current_time)}] Third",
156+
assert logs == [
157+
'{"foo": "bar", "event": "Hi", "level": "info", "timestamp": "2025-03-27T21:58:01.002000+00:00"}'
171158
]
172-
)
173159

174160

175161
@pytest.mark.db_test
@@ -412,6 +398,22 @@ def test_filename_template_for_backward_compatibility(self):
412398
filename_template=None,
413399
)
414400

401+
def test_event_to_str(self):
402+
handler = self.cloudwatch_task_handler
403+
current_time = int(time.time()) * 1000
404+
events = [
405+
{"timestamp": current_time - 2000, "message": "First"},
406+
{"timestamp": current_time - 1000, "message": "Second"},
407+
{"timestamp": current_time, "message": "Third"},
408+
]
409+
assert [handler._event_to_str(event) for event in events] == (
410+
[
411+
f"[{get_time_str(current_time - 2000)}] First",
412+
f"[{get_time_str(current_time - 1000)}] Second",
413+
f"[{get_time_str(current_time)}] Third",
414+
]
415+
)
416+
415417

416418
def generate_log_events(conn, log_group_name, log_stream_name, log_events):
417419
conn.create_log_group(logGroupName=log_group_name)

0 commit comments

Comments
 (0)