phil-core 0.4.0

MongoDB cluster initializer
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70

use mongodb::bson::{self, doc};
use serde::Deserialize;
use tempdir::TempDir;
use uuid::Uuid;

use super::*;

#[derive(Debug)]
struct AutoShutdownCluster {
    cluster: Cluster,
}

impl AutoShutdownCluster {
    fn new(options: ClusterOptions) -> Self {
        Self {
            cluster: Cluster::new(options).unwrap(),
        }
    }
}

impl Drop for AutoShutdownCluster {
    fn drop(&mut self) {
        let _ = self
            .client
            .database("admin")
            .run_command(doc! { "shutdown": 1 }, None);
    }
}

impl std::ops::Deref for AutoShutdownCluster {
    type Target = Cluster;

    fn deref(&self) -> &Self::Target {
        &self.cluster
    }
}

#[derive(Debug, Deserialize)]
struct ReplSetStatus {
    set: String,
}

fn create_temp_dir() -> TempDir {
    TempDir::new(&Uuid::new_v4().to_string()).unwrap()
}

#[test]
fn create_and_initiate_repl_set() {
    let db_dirs: Vec<_> = (0..3).map(|_| create_temp_dir()).collect();

    let cluster_options = ClusterOptions::builder()
        .topology(Topology::ReplicaSet {
            set_name: "test-repl-set".into(),
            db_paths: db_dirs.iter().map(|t| t.path().to_path_buf()).collect(),
        })
        .version_id("4.2".into())
        .build();

    let cluster = AutoShutdownCluster::new(cluster_options);

    let response = cluster
        .client
        .database("admin")
        .run_command(doc! { "replSetGetStatus" : 1 }, None)
        .unwrap();

    let ReplSetStatus { set } = bson::from_document(response).unwrap();

    assert_eq!(set, "test-repl-set");
}