Skip to content

Commit fa56374

Browse files
committed
add check_interval to tests
1 parent 87c340e commit fa56374

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

tests/test_flusher.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -17,11 +17,12 @@ class TestFlushWorker(unittest.TestCase):
1717
source_token = 'dummy_source_token'
1818
buffer_capacity = 5
1919
flush_interval = 2
20+
check_interval = 0.01
2021

2122
def _setup_worker(self, uploader=None):
2223
pipe = queue.Queue(maxsize=self.buffer_capacity)
2324
uploader = uploader or Uploader(self.source_token, self.host)
24-
fw = FlushWorker(uploader, pipe, self.buffer_capacity, self.flush_interval)
25+
fw = FlushWorker(uploader, pipe, self.buffer_capacity, self.flush_interval, self.check_interval)
2526
return pipe, uploader, fw
2627

2728
def test_is_thread(self):

tests/test_handler.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,8 @@ def test_handler_creates_pipe_from_args(self, MockWorker):
3434
def test_handler_creates_and_starts_worker_from_args_after_first_log(self, MockWorker):
3535
buffer_capacity = 9
3636
flush_interval = 9
37-
handler = LogtailHandler(source_token=self.source_token, buffer_capacity=buffer_capacity, flush_interval=flush_interval)
37+
check_interval = 4
38+
handler = LogtailHandler(source_token=self.source_token, buffer_capacity=buffer_capacity, flush_interval=flush_interval, check_interval=check_interval)
3839

3940
self.assertFalse(MockWorker.called)
4041

@@ -47,7 +48,8 @@ def test_handler_creates_and_starts_worker_from_args_after_first_log(self, MockW
4748
handler.uploader,
4849
handler.pipe,
4950
buffer_capacity,
50-
flush_interval
51+
flush_interval,
52+
check_interval,
5153
)
5254
self.assertEqual(handler.flush_thread.start.call_count, 1)
5355

0 commit comments

Comments
 (0)