diff --git a/influxdb_iox/src/influxdb_ioxd/server_type/test.rs b/influxdb_iox/src/influxdb_ioxd/server_type/test.rs index 5129f64e79..4daeb1b187 100644 --- a/influxdb_iox/src/influxdb_ioxd/server_type/test.rs +++ b/influxdb_iox/src/influxdb_ioxd/server_type/test.rs @@ -97,17 +97,16 @@ impl ServerType for TestServerType { } async fn server_grpc(self: Arc, builder_input: RpcBuilderInput) -> Result<(), RpcError> { - if self.test_action == TestAction::PanicInGrpcWorker { - panic!("Test panic in gRPC worker"); - } - if self.test_action == TestAction::EarlyReturnFromGrpcWorker { - return Ok(()); - } + match self.test_action { + TestAction::PanicInGrpcWorker => panic!("Test panic in gRPC worker"), + TestAction::EarlyReturnFromGrpcWorker => Ok(()), + _ => { + let builder = setup_builder!(builder_input, self); + serve_builder!(builder); - let builder = setup_builder!(builder_input, self); - serve_builder!(builder); - - Ok(()) + Ok(()) + } + } } async fn join(self: Arc) {