def test_grpc_server_handled_latency_seconds_with_normal(
target_count, grpc_server, grpc_stub
): # pylint: disable=unused-argument
for i in range(target_count):
grpc_stub.SayHello(hello_world_pb2.HelloRequest(name=str(i)))
target_metric = get_server_metric("grpc_server_handled_latency_seconds")
assert target_metric.samples == []
def test_grpc_server_handling_seconds_with_stream_unary(
number_of_names, grpc_server, grpc_stub, stream_request_generator
): # pylint: disable=unused-argument
grpc_stub.SayHelloStreamUnary(stream_request_generator(number_of_names))
target_metric = get_server_metric("grpc_server_handling_seconds")
assert reduce(
lambda acc, x: acc if acc > x.value else x.value,
list(
filter(
lambda x: x.name == "grpc_server_handling_seconds_bucket",
target_metric.samples
)
),
0
) == 1
assert reduce(
lambda acc, x: acc if acc > x.value else x.value,
list(
filter(
lambda x: x.name == "grpc_server_handling_seconds_count",
target_metric.samples
)
),
0
) == 1
assert reduce(
lambda acc, x: acc if acc > x.value else x.value,
list(
filter(
lambda x: x.name == "grpc_server_handling_seconds_sum",
target_metric.samples
)
),
0
) > 0
def test_grpc_server_msg_sent_with_normal(target_count, grpc_server,
grpc_stub): # pylint: disable=unused-argument
for i in range(target_count):
grpc_stub.SayHello(hello_world_pb2.HelloRequest(name=str(i)))
target_metric = get_server_metric("grpc_server_msg_sent")
# None streaming request has no this metrics
assert target_metric.samples == []
def test_grpc_server_handled_latency_seconds_with_stream_unary(
number_of_names, grpc_server, grpc_stub, stream_request_generator
): # pylint: disable=unused-argument
grpc_stub.SayHelloStreamUnary(
stream_request_generator(number_of_names)
)
target_metric = get_server_metric("grpc_server_handled_latency_seconds")
assert target_metric.samples == []
def test_grpc_server_started_with_bidi_stream(number_of_names, number_of_res,
grpc_server, grpc_stub,
bidi_request_generator): # pylint: disable=unused-argument
list(
grpc_stub.SayHelloBidiStream(
bidi_request_generator(number_of_names, number_of_res)))
target_metric = get_server_metric("grpc_server_started")
assert target_metric.samples == []
def test_grpc_server_msg_sent_with_unary_stream(number_of_res, grpc_server,
grpc_stub): # pylint: disable=unused-argument
list(
grpc_stub.SayHelloUnaryStream(
hello_world_pb2.MultipleHelloResRequest(name="unary stream",
res=number_of_res)))
target_metric = get_server_metric("grpc_server_msg_sent")
assert target_metric.samples[0].value == number_of_res
def test_grpc_server_handled_with_server_error(
target_count, grpc_server, grpc_stub
): # pylint: disable=unused-argument
for _ in range(target_count):
with pytest.raises(grpc.RpcError):
grpc_stub.SayHello(hello_world_pb2.HelloRequest(name="unknownError"))
target_metric = get_server_metric("grpc_server_handled")
assert target_metric.samples == []
def test_legacy_grpc_server_handled_latency_seconds_with_unary_stream(
number_of_res, grpc_legacy_server, grpc_stub
): # pylint: disable=unused-argument
list(
grpc_stub.SayHelloUnaryStream(
hello_world_pb2.MultipleHelloResRequest(
name="unary stream", res=number_of_res
)
)
)
target_metric = get_server_metric("grpc_server_handled_latency_seconds")
# No grpc_server_handled_latency_seconds for streaming response
assert target_metric.samples == []
def test_grpc_server_handling_seconds_with_unary_stream(
number_of_res, grpc_server, grpc_stub
): # pylint: disable=unused-argument
list(
grpc_stub.SayHelloUnaryStream(
hello_world_pb2.MultipleHelloResRequest(
name="unary stream", res=number_of_res
)
)
)
target_metric = get_server_metric("grpc_server_handling_seconds")
# Only one request sent
assert target_metric.samples == []
def test_grpc_server_handled_with_interceptor_error_and_skip_exceptions(
target_count, grpc_server_with_exception_handling, grpc_stub
): # pylint: disable=unused-argument
for _ in range(target_count):
with patch(
'py_grpc_prometheus.prometheus_server_interceptor.'\
'PromServerInterceptor._compute_status_code',
side_effect=Exception('mocked error')
):
assert grpc_stub.SayHello(
hello_world_pb2.HelloRequest(name="unary")
).message == "Hello, unary!"
target_metric = get_server_metric("grpc_server_handled")
assert target_metric.samples == []
def test_legacy_grpc_server_handled_latency_seconds_with_normal(
target_count, grpc_legacy_server, grpc_stub
): # pylint: disable=unused-argument
for i in range(target_count):
grpc_stub.SayHello(hello_world_pb2.HelloRequest(name=str(i)))
target_metric = get_server_metric("grpc_server_handled_latency_seconds")
assert reduce(
lambda acc, x: acc if acc > x.value else x.value,
list(
filter(
lambda x: x.name == "grpc_server_handled_latency_seconds_bucket",
target_metric.samples
)
),
0
) == target_count
assert reduce(
lambda acc, x: acc if acc > x.value else x.value,
list(
filter(
lambda x: x.name == "grpc_server_handled_latency_seconds_count",
target_metric.samples
)
),
0
) == target_count
assert reduce(
lambda acc, x: acc if acc > x.value else x.value,
list(
filter(
lambda x: x.name == "grpc_server_handled_latency_seconds_sum",
target_metric.samples
)
),
0
) > 0
def test_grpc_server_msg_received_with_stream_unary(
number_of_names, grpc_server, grpc_stub, stream_request_generator
): # pylint: disable=unused-argument
grpc_stub.SayHelloStreamUnary(stream_request_generator(number_of_names))
target_metric = get_server_metric("grpc_server_msg_received")
assert target_metric.samples[0].value == number_of_names
def test_grpc_server_started_with_normal(target_count, grpc_server, grpc_stub): # pylint: disable=unused-argument
for i in range(target_count):
grpc_stub.SayHello(hello_world_pb2.HelloRequest(name=str(i)))
target_metric = get_server_metric("grpc_server_started")
assert target_metric.samples[0].value == target_count
