Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license
This commit is contained in:
@@ -0,0 +1,32 @@
|
||||
"""Memory-trim coverage for the long-lived messaging gateway housekeeper."""
|
||||
|
||||
import gateway.run as gateway_run
|
||||
|
||||
|
||||
class _OneTickStopEvent:
|
||||
"""Run one housekeeping tick without a sleep or background thread."""
|
||||
|
||||
def __init__(self):
|
||||
self.waited = False
|
||||
|
||||
def is_set(self):
|
||||
return self.waited
|
||||
|
||||
def wait(self, timeout=None):
|
||||
self.waited = True
|
||||
return True
|
||||
|
||||
|
||||
def test_gateway_housekeeping_calls_periodic_memory_trim(monkeypatch):
|
||||
import hermes_cli.mem_trim as mem_trim
|
||||
|
||||
calls = []
|
||||
monkeypatch.setattr(
|
||||
mem_trim,
|
||||
"trim_memory",
|
||||
lambda **kwargs: calls.append(kwargs) or True,
|
||||
)
|
||||
|
||||
gateway_run._start_gateway_housekeeping(_OneTickStopEvent(), interval=0)
|
||||
|
||||
assert calls == [{"reason": "messaging gateway housekeeping"}]
|
||||
Reference in New Issue
Block a user