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

use schema_core::{
    schema_connector::{Namespaces, SchemaConnector},
    CoreResult,
};

#[must_use = "This struct does nothing on its own. See Reset::send()"]
pub struct Reset<'a> {
    api: &'a mut dyn SchemaConnector,
    soft: bool,
}

impl<'a> Reset<'a> {
    pub fn new(api: &'a mut dyn SchemaConnector) -> Self {
        Reset { api, soft: false }
    }

    pub fn soft(mut self, value: bool) -> Self {
        self.soft = value;
        self
    }

    pub async fn send(self, namespaces: Option<Namespaces>) -> CoreResult<ResetAssertion> {
        self.api.reset(self.soft, namespaces).await?;

        Ok(ResetAssertion {})
    }

    /// Execute the command and expect it to succeed.
    #[track_caller]
    pub fn send_sync(self, namespaces: Option<Namespaces>) -> ResetAssertion {
        test_setup::runtime::run_with_thread_local_runtime(self.send(namespaces)).unwrap()
    }
}

pub struct ResetAssertion {}

impl std::fmt::Debug for ResetAssertion {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "ResetAssertion {{ .. }}")
    }
}