actix-extras/actix-mqtt/tests/test_server.rs

use actix_service::Service;
use actix_testing::TestServer;
use bytes::Bytes;
use bytestring::ByteString;
use futures::future::ok;

use actix_mqtt::{client, Connect, ConnectAck, MqttServer, Publish};

#[derive(Clone)]
struct Session;

async fn connect<Io>(packet: Connect<Io>) -> Result<ConnectAck<Io, Session>, ()> {
    println!("CONNECT: {:?}", packet);
    Ok(packet.ack(Session, false))
}

#[actix_rt::test]
async fn test_simple() -> std::io::Result<()> {
    std::env::set_var(
        "RUST_LOG",
        "actix_codec=info,actix_server=trace,actix_connector=trace",
    );
    env_logger::init();

    let srv = TestServer::with(|| MqttServer::new(connect).finish(|_t| ok(())));

    #[derive(Clone)]
    struct ClientSession;

    let mut client = client::Client::new(ByteString::from_static("user"))
        .state(|ack: client::ConnectAck<_>| async move {
            ack.sink()
                .publish_qos0(ByteString::from_static("#"), Bytes::new(), false);
            ack.sink().close();
            Ok(ack.state(ClientSession))
        })
        .finish(|_t: Publish<_>| {
            async {
                // t.sink().close();
                Ok(())
            }
        });

    let conn = actix_connect::default_connector()
        .call(actix_connect::Connect::with(String::new(), srv.addr()))
        .await
        .unwrap();

    client.call(conn.into_parts().0).await.unwrap();

    Ok(())
}