datasources_sqlite/
lib.rs

1mod alter_table_statement_renderer;
2mod command;
3mod connection;
4mod connection_string;
5mod converters;
6mod create_table_statement_renderer;
7mod create_trigger_statement_renderer;
8mod driver;
9mod reader;
10mod select_inserted_id_statement_text_renderer;
11mod sqlite_path;
12mod default_value_converter;
13mod default_value_options;
14mod sql_generate;
15mod sql_seed_statement_renderer;
16mod type_map;
17mod quoted_renderers;
18mod statement_renderer;
19mod transaction;
20
21pub use driver::SqliteDriver;
22
23#[cfg(test)]
24pub(crate) fn env_lock() -> std::sync::MutexGuard<'static, ()> {
25    static LOCK: std::sync::Mutex<()> = std::sync::Mutex::new(());
26    let guard = LOCK.lock().unwrap_or_else(|error| error.into_inner());
27    std::env::remove_var("INSTALLED_PROVIDERS");
28    std::env::remove_var("DATASOURCE_URL");
29    std::env::set_var("PROVIDER", "sqlite");
30    datasources::reset();
31    datasources::register("sqlite", std::sync::Arc::new(crate::SqliteDriver));
32    guard
33}
34
35#[cfg(test)]
36#[path = "../tests/command.rs"]
37mod command_tests;
38#[cfg(test)]
39#[path = "../tests/contacts.rs"]
40mod contacts;
41#[cfg(test)]
42#[path = "../tests/create_from_driver.rs"]
43mod create_from_driver;
44#[cfg(test)]
45#[path = "../tests/datasource_plural.rs"]
46mod datasource_plural;
47#[cfg(test)]
48#[path = "../tests/integration.rs"]
49mod integration;
50#[cfg(test)]
51#[path = "../tests/null_write.rs"]
52mod null_write;
53#[cfg(test)]
54#[path = "../tests/read_pipeline.rs"]
55mod read_pipeline;
56#[cfg(test)]
57#[path = "../tests/readonly_repository.rs"]
58mod readonly_repository;
59#[cfg(test)]
60#[path = "../tests/repository_spec.rs"]
61mod repository_spec;
62#[cfg(test)]
63#[path = "../tests/generate_sql.rs"]
64mod generate_sql;
65
66#[cfg(test)]
67mod tests {
68    use crate::command::SqliteCommand;
69    use crate::connection::SqliteConnection;
70    use crate::transaction::SqliteTransaction;
71    use datasources::{
72        Row, create, downcast_connection, register, reset, BuiltDatasource, Command,
73        Connection, Datasource, Driver, IsolationLevel, Transaction,
74    };
75    use std::collections::HashMap;
76    use std::sync::Arc;
77
78    struct With(Option<String>);
79
80    impl Driver for With {
81        fn connection(&self, _: Option<&HashMap<String, String>>) -> Arc<dyn Connection> {
82            Arc::new(SqliteConnection::new(self.0.clone()))
83        }
84
85        fn command(&self, connection: Arc<dyn Connection>) -> Arc<dyn Command> {
86            connection.create_command()
87        }
88
89        fn transaction(
90            &self,
91            connection: Arc<dyn Connection>,
92            isolation: IsolationLevel,
93        ) -> Arc<dyn Transaction> {
94            let conn = downcast_connection::<SqliteConnection>(connection).expect("sqlite");
95            Arc::new(SqliteTransaction::new(conn.share(), isolation))
96        }
97
98        fn row_type_converters(
99            &self,
100        ) -> HashMap<
101            String,
102            Box<
103                dyn datasources::TypeConverter<datasources::Value, datasources::Value>
104                    + Send
105                    + Sync,
106            >,
107        > {
108            crate::converters::row_type_converters()
109        }
110    }
111
112    #[tokio::test]
113    async fn sqlite_command_and_repository() {
114        let conn = Arc::new(SqliteConnection::new(Some(":memory:".into())));
115        conn.open().await.unwrap();
116        let command = SqliteCommand::new(conn.clone());
117        command
118            .execute_native(&datasources::Native::new(""), None)
119            .await
120            .unwrap();
121        command
122            .execute_native(
123                &datasources::Native::new("CREATE TABLE t (id INTEGER)"),
124                None,
125            )
126            .await
127            .unwrap();
128        command
129            .execute_native(
130                &datasources::Native::new("INSERT INTO t (id) VALUES (1) RETURNING id"),
131                None,
132            )
133            .await
134            .unwrap();
135        command
136            .execute_reader_native(&datasources::Native::new("SELECT 1"), None)
137            .await
138            .unwrap();
139        let sets = command
140            .execute_reader_native(
141                &datasources::Native::new("SELECT 1 AS id; SELECT 2 AS id"),
142                None,
143            )
144            .await
145            .unwrap();
146        let mut first = Vec::new();
147        while sets.next().await.unwrap() {
148            first.push(sets.current());
149        }
150        let moved = sets.next_result().await.unwrap();
151        let mut second = Vec::new();
152        while sets.next().await.unwrap() {
153            second.push(sets.current());
154        }
155        assert!(moved);
156        assert_eq!(
157            first[0].as_ref().unwrap().get("id"),
158            Some(&datasources::Value::Int(1))
159        );
160        assert_eq!(
161            second[0].as_ref().unwrap().get("id"),
162            Some(&datasources::Value::Int(2))
163        );
164        assert!(!sets.next_result().await.unwrap());
165        drop(sets);
166        command
167            .execute_reader_native(&datasources::Native::new(""), None)
168            .await
169            .unwrap();
170        let txn = conn.start_transaction(None).await.unwrap();
171        assert_eq!(
172            datasources::Transaction::isolation(&*txn),
173            IsolationLevel::ReadCommitted
174        );
175        txn.commit().await.unwrap();
176        let txn = conn
177            .start_transaction(Some(IsolationLevel::ReadUncommitted))
178            .await
179            .unwrap();
180        assert_eq!(
181            datasources::Transaction::isolation(&*txn),
182            IsolationLevel::ReadUncommitted
183        );
184        txn.abort().await.unwrap();
185        conn.close().await.unwrap();
186        assert!(conn.ensure_open().is_err());
187        assert!(conn.start_transaction(None).await.is_err());
188
189        let shared = Arc::new(SqliteConnection::new(Some(
190            "file:iso?mode=memory&cache=shared".into(),
191        )));
192        shared.open().await.unwrap();
193        let read_committed = shared
194            .start_transaction(Some(IsolationLevel::ReadCommitted))
195            .await
196            .unwrap();
197        assert_eq!(
198            datasources::Transaction::isolation(&*read_committed),
199            IsolationLevel::ReadCommitted
200        );
201        read_committed.abort().await.unwrap();
202        shared.close().await.unwrap();
203
204        let defaulted = SqliteConnection::new(None);
205        assert_eq!(defaulted.filename, ":memory:");
206        reset();
207        register("sqlite", Arc::new(With(Some(":memory:".into()))));
208        let ds = create(Some(&HashMap::from([("PROVIDER".into(), "sqlite".into())]))).unwrap();
209        assert_eq!(ds.provider(), "sqlite");
210        ds.create_connection().await.unwrap();
211        let command = ds.create_command(None).await.unwrap();
212        assert!(command.execute("DELETE FROM items", None).await.is_err());
213        command
214            .execute_reader_native(&datasources::Native::new("SELECT 1"), None)
215            .await
216            .unwrap();
217        ds.create_repository::<Row>(None, None);
218        let factory = BuiltDatasource::new("sqlite", Arc::new(With(None)), None);
219        factory.create_connection().await.unwrap();
220        let bound = factory
221            .create_command(Some(factory.create_connection().await.unwrap()))
222            .await
223            .unwrap();
224        assert!(bound.execute("DELETE FROM items", None).await.is_err());
225        assert!(bound
226            .execute_reader_native(&datasources::Native::new("SELECT 1"), None)
227            .await
228            .is_err());
229        (&factory as &dyn Datasource)
230            .create_repository::<Row>(Some(factory.create_connection().await.unwrap()), None);
231
232        let closed = Arc::new(SqliteConnection::new(None));
233        assert!(SqliteCommand::new(closed)
234            .execute_native(&datasources::Native::new("x"), None)
235            .await
236            .is_err());
237    }
238
239    #[tokio::test]
240    async fn return_statement_selects_scalar() {
241        use datasources::{Parameter, ReturnStatement};
242        let conn = Arc::new(SqliteConnection::new(Some(":memory:".into())));
243        conn.open().await.unwrap();
244        let command = SqliteCommand::new(conn);
245        let parameters = [Parameter::new("x", -3_i64)];
246        let response = command
247            .execute_statement(&ReturnStatement::new("abs", &parameters), Some(&parameters))
248            .await
249            .unwrap();
250        assert_eq!(response.result, datasources::Value::Int(3));
251    }
252}
datasources_sqlite/lib.rs

datasources_sqlite/
lib.rs
