Skip to content

Commit 8d643ae

Browse files
committed
Update test_detector.py
1 parent 4fd50f6 commit 8d643ae

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/detector/test_detector.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -274,6 +274,7 @@ def test_save_warning(self, mock_clickhouse, mock_kafka_consume_handler):
274274
"sha256": "ba1f718179191348fe2abd51644d76191d42a5d967c6844feb3371b6f798bf06",
275275
},
276276
]
277+
sut.messages = [{"logline_id": "test_id"}]
277278
open_mock = mock_open()
278279
with patch("src.detector.detector.open", open_mock, create=True):
279280
sut.send_warning()
@@ -299,6 +300,7 @@ def test_save_empty_warning(self, mock_clickhouse, mock_kafka_consume_handler):
299300

300301
sut = Detector()
301302
sut.warnings = []
303+
sut.messages = [{"logline_id": "test_id"}]
302304
open_mock = mock_open()
303305
with patch("src.detector.detector.open", open_mock, create=True):
304306
sut.send_warning()

0 commit comments

Comments
 (0)