Minimal version of piping autoscaler events to driver logs (#13434)

This commit is contained in:
Eric Liang
2021-01-16 10:06:20 -08:00
committed by GitHub
parent 7e54911093
commit 8c8af2616e
13 changed files with 274 additions and 30 deletions
+7
View File
@@ -75,9 +75,13 @@ def setup_monitor(address):
def verify_load_metrics(monitor, expected_resource_usage=None, timeout=30):
request_resources(num_cpus=42)
# Disable event clearing for test.
monitor.event_summarizer.clear = lambda *a: None
while True:
monitor.update_load_metrics()
monitor.update_resource_requests()
monitor.update_event_summary()
resource_usage = monitor.load_metrics._get_resource_usage()
# Check resource request propagation.
@@ -113,6 +117,9 @@ def verify_load_metrics(monitor, expected_resource_usage=None, timeout=30):
raise ValueError("Timeout. {} != {}".format(
resource_usage, expected_resource_usage))
# Sanity check we emitted a resize event.
assert any("Resized to" in x for x in monitor.event_summarizer.summary())
return resource_usage