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"]), )