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 crate::connection::Connection;
use crate::datasource_error::DatasourceError;
use crate::isolation_level::IsolationLevel;
use crate::result::Result;
use crate::stubs::StubConnection;
use crate::transaction::Transaction;
use async_trait::async_trait;
use std::sync::{Arc, Mutex};

pub struct MockTransaction {
    connection: Arc<dyn Connection>,
    isolation: Mutex<IsolationLevel>,
    commit_calls: Mutex<usize>,
    abort_calls: Mutex<usize>,
    commit_error: Mutex<Option<String>>,
    abort_error: Mutex<Option<String>>,
    committed: Mutex<bool>,
    aborted: Mutex<bool>,
}

impl MockTransaction {
    pub fn new_unbound() -> Self {
        Self {
            connection: Arc::new(StubConnection::new("mock")),
            isolation: Mutex::new(IsolationLevel::ReadCommitted),
            commit_calls: Mutex::new(0),
            abort_calls: Mutex::new(0),
            commit_error: Mutex::new(None),
            abort_error: Mutex::new(None),
            committed: Mutex::new(false),
            aborted: Mutex::new(false),
        }
    }

    pub fn set_isolation(&self, isolation: IsolationLevel) {
        *self.isolation.lock().expect("mock") = isolation;
    }

    pub fn set_commit_error(&self, message: impl Into<String>) {
        *self.commit_error.lock().expect("mock") = Some(message.into());
    }

    pub fn set_abort_error(&self, message: impl Into<String>) {
        *self.abort_error.lock().expect("mock") = Some(message.into());
    }

    pub fn commit_calls(&self) -> usize {
        *self.commit_calls.lock().expect("mock")
    }

    pub fn abort_calls(&self) -> usize {
        *self.abort_calls.lock().expect("mock")
    }

    pub fn committed(&self) -> bool {
        *self.committed.lock().expect("mock")
    }

    pub fn aborted(&self) -> bool {
        *self.aborted.lock().expect("mock")
    }
}

#[async_trait]
impl Transaction for MockTransaction {
    fn connection(&self) -> Arc<dyn Connection> {
        Arc::clone(&self.connection)
    }

    fn isolation(&self) -> IsolationLevel {
        *self.isolation.lock().expect("mock")
    }

    async fn commit(&self) -> Result<()> {
        *self.commit_calls.lock().expect("mock") += 1;
        if let Some(error) = &*self.commit_error.lock().expect("mock") {
            return Err(DatasourceError::message(error.clone()));
        }
        *self.committed.lock().expect("mock") = true;
        Ok(())
    }

    async fn abort(&self) -> Result<()> {
        *self.abort_calls.lock().expect("mock") += 1;
        if let Some(error) = &*self.abort_error.lock().expect("mock") {
            return Err(DatasourceError::message(error.clone()));
        }
        *self.aborted.lock().expect("mock") = true;
        Ok(())
    }
}