wacker 0.10.0

wacker core
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

use std::fs::remove_dir_all;
use std::thread::sleep;
use std::time::Duration;
use tokio::sync::broadcast::{channel, Receiver, Sender};
use tonic::transport::Channel;
use wacker::{new_client_with_path, utils::generate_random_string, Client, Server};

pub struct TestServer {
    sender: Sender<()>,
    receiver: Receiver<()>,
    dir: String,
}

impl TestServer {
    pub fn new() -> Self {
        let (sender, receiver) = channel(1);
        Self {
            sender,
            receiver,
            dir: format!("wacker-test-{}", generate_random_string(5)),
        }
    }

    pub async fn start(&mut self) {
        let dir = self.dir.clone();
        let mut receiver = self.receiver.resubscribe();

        Server::new()
            .with_dir(dir.clone())
            .is_test(true)
            .start(async move {
                receiver.recv().await.expect("");
                remove_dir_all(dir).expect("remove dir failed");
            })
            .await
            .unwrap();
    }

    pub async fn client(&self) -> Client<Channel> {
        new_client_with_path(format!("{}/wacker.sock", self.dir)).await.unwrap()
    }
}

impl Drop for TestServer {
    fn drop(&mut self) {
        let _ = self.sender.send(());
        sleep(Duration::from_secs(1));
    }
}