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

use crate::util::new_random_socket_addr;
use ::anyhow::Context;
use ::anyhow::Result;
use ::std::net::SocketAddr;

/// The basic setup for the `TestServer`.
#[derive(Debug, Clone)]
pub struct TestServerConfig {
    /// Set the default content type for all requests created by the `TestServer`.
    ///
    /// This overrides the default 'best efforts' approach of requests.
    pub default_content_type: Option<String>,

    /// Set the socket to use for the server.
    ///
    /// **Defaults** to a _random_ socket.
    pub socket_address: Option<SocketAddr>,

    /// Set for the server to save cookies that are returned,
    /// for use in future requests.
    ///
    /// This is useful for automatically saving session cookies (and similar)
    /// like a browser would do.
    ///
    /// **Defaults** to false (being turned off).
    pub save_cookies: bool,
}

impl TestServerConfig {
    pub(crate) fn build_socket_address(&self) -> Result<SocketAddr> {
        let socket_address = match self.socket_address {
            Some(socket_address) => socket_address,
            None => new_random_socket_addr().context("Cannot create socket address for use")?,
        };

        Ok(socket_address)
    }
}

impl Default for TestServerConfig {
    fn default() -> Self {
        Self {
            default_content_type: None,
            socket_address: None,
            save_cookies: false,
        }
    }
}