datasources 0.1.5

Async datasource interfaces, middleware, registry, and test doubles
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
90
91

use super::mock_command::MockCommand;
use super::mock_connection::MockConnection;
use super::mock_repository::MockRepository;
use crate::command::Command;
use crate::datasource_error::DatasourceError;
use crate::connection::Connection;
use crate::converters::TypeConverter;
use crate::crud_repository_options::CRUDRepositoryOptions;
use crate::datasource::Datasource;
use crate::repository_factory::RepositoryFactory;
use crate::readonly_repository_mapping_options::ReadonlyRepositoryMappingOptions;
use crate::repository::{as_readonly, ICRUDRepository, IReadonlyRepository};
use crate::result::Result;
use crate::value::Value;
use async_trait::async_trait;
use std::collections::HashMap;
use std::sync::Arc;

pub struct MockDatasource {
    provider: String,
    connection: Arc<dyn Connection>,
    command: Arc<dyn Command>,
    repository: Arc<dyn ICRUDRepository>,
    repositories: RepositoryFactory,
}

impl MockDatasource {
    pub fn new(provider: impl Into<String>) -> Self {
        Self {
            provider: provider.into(),
            connection: MockConnection::new(),
            command: Arc::new(MockCommand::new()),
            repository: Arc::new(MockRepository::new()),
            repositories: RepositoryFactory::new(),
        }
    }
}

#[async_trait]
impl Datasource for MockDatasource {
    fn provider(&self) -> &str {
        &self.provider
    }

    async fn create_connection(&self) -> Result<Arc<dyn Connection>> {
        Ok(self.connection.clone())
    }

    async fn create_command(
        &self,
        connection: Option<Arc<dyn Connection>>,
    ) -> Result<Arc<dyn Command>> {
        if connection.is_none() {
            if self.connection.is_open() {
                return Err(DatasourceError::message("Connection is already open"));
            }
            self.connection.open().await?;
        }
        Ok(self.command.clone())
    }

    fn bind_connection(&self, connection: Option<Arc<dyn Connection>>) -> Arc<dyn Connection> {
        connection.unwrap_or_else(|| self.connection.clone())
    }

    fn row_type_converters(
        &self,
    ) -> HashMap<String, Box<dyn TypeConverter<Value, Value> + Send + Sync>> {
        HashMap::new()
    }

    fn repository_from(
        &self,
        _connection: Arc<dyn Connection>,
        _options: CRUDRepositoryOptions,
    ) -> Arc<dyn ICRUDRepository> {
        self.repository.clone()
    }

    fn readonly_repository_from(
        &self,
        _connection: Arc<dyn Connection>,
        _options: ReadonlyRepositoryMappingOptions,
    ) -> Arc<dyn IReadonlyRepository> {
        as_readonly(self.repository.clone())
    }

    fn repositories(&self) -> &RepositoryFactory {
        &self.repositories
    }
}