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
50
51
52
53
54
55
56
57
58

use schema_core::{
    commands::mark_migration_applied, json_rpc::types::*, schema_connector::SchemaConnector, CoreError, CoreResult,
};
use tempfile::TempDir;

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

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

    pub fn send_impl(self) -> CoreResult<MarkMigrationAppliedAssertion<'a>> {
        let output = test_setup::runtime::run_with_thread_local_runtime(mark_migration_applied(
            MarkMigrationAppliedInput {
                migrations_directory_path: self.migrations_directory.path().to_str().unwrap().to_owned(),
                migration_name: self.migration_name,
            },
            self.api,
        ))?;
        Ok(MarkMigrationAppliedAssertion {
            _output: output,
            _migrations_directory: self.migrations_directory,
        })
    }

    pub fn send(self) -> MarkMigrationAppliedAssertion<'a> {
        self.send_impl().unwrap()
    }

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

pub struct MarkMigrationAppliedAssertion<'a> {
    _output: MarkMigrationAppliedOutput,
    _migrations_directory: &'a TempDir,
}

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