mockforge_grpc/
protocol_server.rs

1//! Unified protocol server implementation for the gRPC mock server.
2
3use async_trait::async_trait;
4use mockforge_core::protocol_abstraction::Protocol;
5use mockforge_core::protocol_server::MockProtocolServer;
6
7use crate::dynamic::DynamicGrpcConfig;
8use mockforge_core::LatencyProfile;
9
10/// A `MockProtocolServer` wrapper around the gRPC server startup.
11///
12/// Wraps [`crate::start_with_config`] with shutdown-signal integration.
13pub struct GrpcMockServer {
14    port: u16,
15    latency_profile: Option<LatencyProfile>,
16    config: DynamicGrpcConfig,
17}
18
19impl GrpcMockServer {
20    /// Create a new `GrpcMockServer` with the given configuration.
21    pub fn new(
22        port: u16,
23        latency_profile: Option<LatencyProfile>,
24        config: DynamicGrpcConfig,
25    ) -> Self {
26        Self {
27            port,
28            latency_profile,
29            config,
30        }
31    }
32}
33
34#[async_trait]
35impl MockProtocolServer for GrpcMockServer {
36    fn protocol(&self) -> Protocol {
37        Protocol::Grpc
38    }
39
40    async fn start(
41        &self,
42        mut shutdown: tokio::sync::watch::Receiver<()>,
43    ) -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
44        let port = self.port;
45        let latency = self.latency_profile.clone();
46        let config = self.config.clone();
47
48        tokio::select! {
49            result = crate::start_with_config(port, latency, config) => {
50                result
51            }
52            _ = shutdown.changed() => {
53                tracing::info!("Shutting down gRPC server on port {}", port);
54                Ok(())
55            }
56        }
57    }
58
59    fn port(&self) -> u16 {
60        self.port
61    }
62
63    fn description(&self) -> String {
64        format!("gRPC server on port {}", self.port)
65    }
66}
67
68#[cfg(test)]
69mod tests {
70    use super::*;
71
72    #[test]
73    fn test_grpc_mock_server_protocol() {
74        let server = GrpcMockServer::new(50051, None, DynamicGrpcConfig::default());
75        assert_eq!(server.protocol(), Protocol::Grpc);
76    }
77
78    #[test]
79    fn test_grpc_mock_server_port() {
80        let server = GrpcMockServer::new(50051, None, DynamicGrpcConfig::default());
81        assert_eq!(server.port(), 50051);
82    }
83
84    #[test]
85    fn test_grpc_mock_server_description() {
86        let server = GrpcMockServer::new(50051, None, DynamicGrpcConfig::default());
87        assert_eq!(server.description(), "gRPC server on port 50051");
88    }
89}
mockforge_grpc/protocol_server.rs

mockforge_grpc/
protocol_server.rs
