mockito 0.32.3

HTTP mocking for Rust.
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

use crate::Server;
use crate::{Error, ErrorKind};
use async_trait::async_trait;
use deadpool::managed::{self, Pool, RecycleError};
use lazy_static::lazy_static;

const DEFAULT_POOL_SIZE: usize = 100;

lazy_static! {
    pub(crate) static ref SERVER_POOL: Pool<ServerPool> = ServerPool::new(DEFAULT_POOL_SIZE);
}

pub(crate) struct ServerPool {}

impl ServerPool {
    fn new(max_size: usize) -> Pool<ServerPool> {
        let server_pool = ServerPool {};
        Pool::builder(server_pool)
            .max_size(max_size)
            .build()
            .expect("Could not create server pool")
    }
}

#[async_trait]
impl managed::Manager for ServerPool {
    type Type = Server;
    type Error = Error;

    async fn create(&self) -> Result<Server, Error> {
        Server::try_new_with_port_async(0).await
    }

    async fn recycle(&self, server: &mut Server) -> managed::RecycleResult<Error> {
        if server.busy() {
            return Err(RecycleError::Backend(Error::new(ErrorKind::ServerBusy)));
        }

        server.reset_async().await;
        Ok(())
    }
}