mirror of https://github.com/sunface/rust-course
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
3.1 KiB
93 lines
3.1 KiB
use mini_redis::{client, server};
|
|
use std::net::SocketAddr;
|
|
use tokio::net::TcpListener;
|
|
use tokio::task::JoinHandle;
|
|
|
|
/// A basic "hello world" style test. A server instance is started in a
|
|
/// background task. A client instance is then established and set and get
|
|
/// commands are sent to the server. The response is then evaluated
|
|
#[tokio::test]
|
|
async fn key_value_get_set() {
|
|
let (addr, _) = start_server().await;
|
|
|
|
let mut client = client::connect(addr).await.unwrap();
|
|
client.set("hello", "world".into()).await.unwrap();
|
|
|
|
let value = client.get("hello").await.unwrap().unwrap();
|
|
assert_eq!(b"world", &value[..])
|
|
}
|
|
|
|
/// similar to the "hello world" style test, But this time
|
|
/// a single channel subscription will be tested instead
|
|
#[tokio::test]
|
|
async fn receive_message_subscribed_channel() {
|
|
let (addr, _) = start_server().await;
|
|
|
|
let client = client::connect(addr.clone()).await.unwrap();
|
|
let mut subscriber = client.subscribe(vec!["hello".into()]).await.unwrap();
|
|
|
|
tokio::spawn(async move {
|
|
let mut client = client::connect(addr).await.unwrap();
|
|
client.publish("hello", "world".into()).await.unwrap()
|
|
});
|
|
|
|
let message = subscriber.next_message().await.unwrap().unwrap();
|
|
assert_eq!("hello", &message.channel);
|
|
assert_eq!(b"world", &message.content[..])
|
|
}
|
|
|
|
/// test that a client gets messages from multiple subscribed channels
|
|
#[tokio::test]
|
|
async fn receive_message_multiple_subscribed_channels() {
|
|
let (addr, _) = start_server().await;
|
|
|
|
let client = client::connect(addr.clone()).await.unwrap();
|
|
let mut subscriber = client
|
|
.subscribe(vec!["hello".into(), "world".into()])
|
|
.await
|
|
.unwrap();
|
|
|
|
tokio::spawn(async move {
|
|
let mut client = client::connect(addr).await.unwrap();
|
|
client.publish("hello", "world".into()).await.unwrap()
|
|
});
|
|
|
|
let message1 = subscriber.next_message().await.unwrap().unwrap();
|
|
assert_eq!("hello", &message1.channel);
|
|
assert_eq!(b"world", &message1.content[..]);
|
|
|
|
tokio::spawn(async move {
|
|
let mut client = client::connect(addr).await.unwrap();
|
|
client.publish("world", "howdy?".into()).await.unwrap()
|
|
});
|
|
|
|
let message2 = subscriber.next_message().await.unwrap().unwrap();
|
|
assert_eq!("world", &message2.channel);
|
|
assert_eq!(b"howdy?", &message2.content[..])
|
|
}
|
|
|
|
/// test that a client accurately removes its own subscribed chanel list
|
|
/// when unbscribing to all subscribed channels by submitting an empty vec
|
|
#[tokio::test]
|
|
async fn unsubscribes_from_channels() {
|
|
let (addr, _) = start_server().await;
|
|
|
|
let client = client::connect(addr.clone()).await.unwrap();
|
|
let mut subscriber = client
|
|
.subscribe(vec!["hello".into(), "world".into()])
|
|
.await
|
|
.unwrap();
|
|
|
|
subscriber.unsubscribe(&[]).await.unwrap();
|
|
assert_eq!(subscriber.get_subscribed().len(), 0);
|
|
}
|
|
|
|
async fn start_server() -> (SocketAddr, JoinHandle<()>) {
|
|
let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
|
|
let addr = listener.local_addr().unwrap();
|
|
|
|
let handle = tokio::spawn(async move { server::run(listener, tokio::signal::ctrl_c()).await });
|
|
|
|
(addr, handle)
|
|
}
|