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

use schema_core::{
    commands::mark_migration_rolled_back, json_rpc::types::*, schema_connector::SchemaConnector, CoreError, CoreResult,
};

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

impl<'a> MarkMigrationRolledBack<'a> {
    pub fn new(api: &'a mut dyn SchemaConnector, migration_name: String) -> Self {
        MarkMigrationRolledBack { api, migration_name }
    }

    fn send_impl(self) -> CoreResult<MarkMigrationRolledBackAssertion> {
        let output = test_setup::runtime::run_with_thread_local_runtime(mark_migration_rolled_back(
            MarkMigrationRolledBackInput {
                migration_name: self.migration_name,
            },
            self.api,
        ))?;

        Ok(MarkMigrationRolledBackAssertion { _output: output })
    }

    pub fn send(self) -> MarkMigrationRolledBackAssertion {
        self.send_impl().unwrap()
    }

    pub fn send_unwrap_err(self) -> CoreError {
        self.send_impl().unwrap_err()
    }
}

pub struct MarkMigrationRolledBackAssertion {
    _output: MarkMigrationRolledBackOutput,
}

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