atlas-local 0.7.1

A library for managing MongoDB Atlas local environments.
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89

use crate::{
    client::Client,
    docker::{DockerError, DockerPullImage},
};

#[derive(Debug, Clone, PartialEq, thiserror::Error)]
#[error("Failed to pull image: {0}")]
pub struct PullImageError(#[from] DockerError);

impl<D: DockerPullImage> Client<D> {
    /// Pulls the Atlas Local image.
    ///
    /// # Arguments
    ///
    /// * `image` - The image to pull.
    /// * `tag` - The tag to pull.
    pub async fn pull_image(&self, image: &str, tag: &str) -> Result<(), PullImageError> {
        self.docker.pull_image(image, tag).await?;
        Ok(())
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::docker::DockerError;
    use mockall::mock;

    mock! {
        Docker {}

        impl DockerPullImage for Docker {
            async fn pull_image(&self, image: &str, tag: &str) -> Result<(), DockerError>;
        }
    }

    #[tokio::test]
    async fn test_pull_image() {
        // Arrange
        let mut mock_docker = MockDocker::new();

        // Set up expectations
        mock_docker
            .expect_pull_image()
            .with(
                mockall::predicate::eq("mongodb/mongodb-atlas-local"),
                mockall::predicate::eq("8.0.0"),
            )
            .times(1)
            .returning(|_, _| Ok(()));

        let client = Client::new(mock_docker);

        // Act
        let result = client
            .pull_image("mongodb/mongodb-atlas-local", "8.0.0")
            .await;

        // Assert
        assert!(result.is_ok());
    }

    #[tokio::test]
    async fn test_pull_image_docker_error() {
        // Arrange
        let mut mock_docker = MockDocker::new();

        // Set up expectations
        mock_docker
            .expect_pull_image()
            .with(
                mockall::predicate::eq("mongodb/mongodb-atlas-local"),
                mockall::predicate::eq("invalid-tag"),
            )
            .times(1)
            .returning(|_, _| Err(DockerError::NotFound));

        let client = Client::new(mock_docker);

        // Act
        let result = client
            .pull_image("mongodb/mongodb-atlas-local", "invalid-tag")
            .await;

        // Assert
        assert!(result.is_err());
        assert!(matches!(result.unwrap_err(), PullImageError(_)));
    }
}