Skip to content

Commit 0e772f7

Browse files
committed
test(integration): update collector tests for offline mode
1 parent 2efc9e3 commit 0e772f7

1 file changed

Lines changed: 15 additions & 6 deletions

File tree

tests/test_integration.py

Lines changed: 15 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -186,10 +186,19 @@ def test_get_tap(self) -> None:
186186
class TestBackgroundCollector:
187187
"""Test background collector functionality."""
188188

189+
def setup_method(self) -> None:
190+
"""Reset config and stop collector before each test."""
191+
import trueentropy
192+
# Use offline mode for faster tests (no network harvesters blocking)
193+
trueentropy.configure(offline_mode=True)
194+
trueentropy.stop_collector()
195+
time.sleep(0.1)
196+
189197
def teardown_method(self) -> None:
190198
"""Ensure collector is stopped after each test."""
191199
import trueentropy
192200
trueentropy.stop_collector()
201+
trueentropy.reset_config() # Reset config for next test
193202
time.sleep(0.1) # Give time for thread to stop
194203

195204
def test_start_and_stop_collector(self) -> None:
@@ -201,9 +210,9 @@ def test_start_and_stop_collector(self) -> None:
201210
trueentropy.stop_collector()
202211
time.sleep(0.1)
203212

204-
# Start collector
213+
# Start collector (uses offline mode from setup_method)
205214
trueentropy.start_collector(interval=0.5)
206-
time.sleep(0.1) # Give time for thread to start
215+
time.sleep(0.2) # Give time for thread to start
207216

208217
# Should be running
209218
assert is_collector_running()
@@ -213,7 +222,7 @@ def test_start_and_stop_collector(self) -> None:
213222

214223
# Should be stopped
215224
# Give it a moment to stop
216-
time.sleep(0.2)
225+
time.sleep(0.3)
217226
assert not is_collector_running()
218227

219228
def test_collector_feeds_pool(self) -> None:
@@ -227,15 +236,15 @@ def test_collector_feeds_pool(self) -> None:
227236
pool = trueentropy.get_pool()
228237
initial_fed = pool.total_fed
229238

230-
# Start collector with fast interval
239+
# Start collector with fast interval (uses offline mode)
231240
trueentropy.start_collector(interval=0.2)
232241

233242
# Wait for some collections
234-
time.sleep(0.6)
243+
time.sleep(0.8)
235244

236245
# Stop collector
237246
trueentropy.stop_collector()
238-
time.sleep(0.1)
247+
time.sleep(0.2)
239248

240249
# Should have fed some data
241250
assert pool.total_fed > initial_fed

0 commit comments

Comments
 (0)