Skip to content

Commit 5fce15c

Browse files
committed
chore(test): Refactor default stubs test
1 parent e58aec1 commit 5fce15c

File tree

3 files changed

+10
-16
lines changed

3 files changed

+10
-16
lines changed

tests/default_stubs/Cargo.toml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,10 @@ name = "default_stubs"
77
[dependencies]
88
tokio = {version = "1.0", features = ["macros", "rt-multi-thread", "net"]}
99
tokio-stream = {version = "0.1", features = ["net"]}
10-
rand = "0.9"
1110
tonic = {path = "../../tonic"}
1211

12+
[dev-dependencies]
13+
rand = "0.9"
14+
1315
[build-dependencies]
1416
tonic-build = {path = "../../tonic-build" }

tests/default_stubs/src/lib.rs

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,12 @@
1-
#![allow(unused_imports)]
2-
3-
mod test_defaults;
4-
51
use std::pin::Pin;
6-
use tokio_stream::{Stream, StreamExt};
2+
use tokio_stream::Stream;
73
use tonic::{Request, Response, Status, Streaming};
84

95
tonic::include_proto!("test");
106
tonic::include_proto!("test_default");
117

128
#[derive(Debug, Default)]
13-
struct Svc;
9+
pub struct Svc;
1410

1511
#[tonic::async_trait]
1612
impl test_server::Test for Svc {

tests/default_stubs/src/test_defaults.rs renamed to tests/default_stubs/tests/default.rs

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,17 @@
1-
#![allow(unused_imports)]
2-
3-
use crate::test_client::TestClient;
4-
use crate::*;
5-
use rand::Rng as _;
1+
use default_stubs::test_client::TestClient;
2+
use default_stubs::*;
63
use std::env;
74
use std::fs;
85
use std::net::SocketAddr;
96
use tokio::net::TcpListener;
7+
use tokio_stream::{Stream, StreamExt};
108
use tonic::transport::Channel;
119
use tonic::transport::Server;
1210

13-
#[cfg(test)]
1411
fn echo_requests_iter() -> impl Stream<Item = ()> {
1512
tokio_stream::iter(1..usize::MAX).map(|_| ())
1613
}
1714

18-
#[cfg(test)]
1915
async fn test_default_stubs(
2016
mut client: TestClient<Channel>,
2117
mut client_default_stubs: TestClient<Channel>,
@@ -100,7 +96,6 @@ async fn test_default_stubs_uds() {
10096
test_default_stubs(client, client_default_stubs).await;
10197
}
10298

103-
#[cfg(test)]
10499
async fn run_services_in_background() -> (SocketAddr, SocketAddr) {
105100
let svc = test_server::TestServer::new(Svc {});
106101
let svc_default_stubs = test_default_server::TestDefaultServer::new(Svc {});
@@ -132,8 +127,9 @@ async fn run_services_in_background() -> (SocketAddr, SocketAddr) {
132127
(addr, addr_default_stubs)
133128
}
134129

135-
#[cfg(all(test, not(target_os = "windows")))]
130+
#[cfg(not(target_os = "windows"))]
136131
async fn run_services_in_background_uds() -> (String, String) {
132+
use rand::Rng as _;
137133
use tokio::net::UnixListener;
138134

139135
let svc = test_server::TestServer::new(Svc {});

0 commit comments

Comments
 (0)