def test_on_assignment_start_failed(self, monitor: PrometheusMonitor,
metrics: FaustMetrics) -> None:
assignor = Mock(name="assignor")
state = monitor.on_assignment_start(assignor)
monitor.on_assignment_error(assignor, state, KeyError("foo"))
self.assert_has_sample_value(
metrics.assignment_operations,
"assignment_operations_total",
{"operation": monitor.ERROR},
1,
)
self.assert_has_sample_value(
metrics.assign_latency,
"assign_latency_sum",
{},
monitor.ms_since(state["time_start"]),
)
def test_on_assignment_start_completed(self, monitor: PrometheusMonitor,
metrics: FaustMetrics) -> None:
assignor = Mock(name="assignor")
state = monitor.on_assignment_start(assignor)
monitor.on_assignment_completed(assignor, state)
self.assert_has_sample_value(
metrics.assignment_operations,
"assignment_operations_total",
{"operation": monitor.COMPLETED},
1,
)
self.assert_has_sample_value(
metrics.assign_latency,
"assign_latency_sum",
{},
monitor.ms_since(state["time_start"]),
)