146 lines
3.6 KiB
Rust
146 lines
3.6 KiB
Rust
/*
|
|
* Copyright (c) 2023 Stalwart Labs Ltd.
|
|
*
|
|
* This file is part of the Stalwart Mail Server.
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as
|
|
* published by the Free Software Foundation, either version 3 of
|
|
* the License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
* in the LICENSE file at the top-level directory of this distribution.
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
* You can be released from the requirements of the AGPLv3 license by
|
|
* purchasing a commercial license. Please contact licensing@stalw.art
|
|
* for more details.
|
|
*/
|
|
|
|
pub mod assign_id;
|
|
pub mod blob;
|
|
pub mod lookup;
|
|
pub mod ops;
|
|
pub mod query;
|
|
|
|
use std::io::Read;
|
|
|
|
use store::{FtsStore, Stores};
|
|
use utils::config::Config;
|
|
|
|
use crate::AssertConfig;
|
|
|
|
pub struct TempDir {
|
|
pub path: std::path::PathBuf,
|
|
}
|
|
|
|
const CONFIG: &str = r#"
|
|
[store."s3"]
|
|
type = "s3"
|
|
access-key = "minioadmin"
|
|
secret-key = "minioadmin"
|
|
region = "eu-central-1"
|
|
endpoint = "http://localhost:9000"
|
|
bucket = "tmp"
|
|
|
|
[store."fs"]
|
|
type = "fs"
|
|
path = "{TMP}"
|
|
|
|
[store."rocksdb"]
|
|
type = "rocksdb"
|
|
path = "{TMP}/rocksdb"
|
|
|
|
[store."foundationdb"]
|
|
type = "foundationdb"
|
|
|
|
[store."sqlite"]
|
|
type = "sqlite"
|
|
path = "{TMP}/sqlite.db"
|
|
|
|
[store."postgresql"]
|
|
type = "postgresql"
|
|
host = "localhost"
|
|
port = 5432
|
|
database = "stalwart"
|
|
user = "postgres"
|
|
password = "mysecretpassword"
|
|
|
|
[store."mysql"]
|
|
type = "mysql"
|
|
host = "localhost"
|
|
port = 3307
|
|
database = "stalwart"
|
|
user = "root"
|
|
password = "password"
|
|
|
|
[store."redis"]
|
|
type = "redis"
|
|
urls = "redis://127.0.0.1"
|
|
redis-type = "single"
|
|
|
|
"#;
|
|
|
|
#[tokio::test(flavor = "multi_thread")]
|
|
pub async fn store_tests() {
|
|
let insert = true;
|
|
let temp_dir = TempDir::new("store_tests", insert);
|
|
let mut config = Config::new(CONFIG.replace("{TMP}", &temp_dir.path.to_string_lossy()))
|
|
.unwrap()
|
|
.assert_no_errors();
|
|
let stores = Stores::parse_all(&mut config).await;
|
|
|
|
let store_id = std::env::var("STORE")
|
|
.expect("Missing store type. Try running `STORE=<store_type> cargo test`");
|
|
let store = stores
|
|
.stores
|
|
.get(&store_id)
|
|
.expect("Store not found")
|
|
.clone();
|
|
|
|
println!("Testing store {}...", store_id);
|
|
if insert {
|
|
store.destroy().await;
|
|
}
|
|
ops::test(store.clone()).await;
|
|
query::test(store.clone(), FtsStore::Store(store.clone()), insert).await;
|
|
assign_id::test(store).await;
|
|
|
|
if insert {
|
|
temp_dir.delete();
|
|
}
|
|
}
|
|
|
|
pub fn deflate_test_resource(name: &str) -> Vec<u8> {
|
|
let mut csv_path = std::path::PathBuf::from(env!("CARGO_MANIFEST_DIR"));
|
|
csv_path.push("resources");
|
|
csv_path.push(name);
|
|
|
|
let mut decoder = flate2::bufread::GzDecoder::new(std::io::BufReader::new(
|
|
std::fs::File::open(csv_path).unwrap(),
|
|
));
|
|
let mut result = Vec::new();
|
|
decoder.read_to_end(&mut result).unwrap();
|
|
result
|
|
}
|
|
|
|
impl TempDir {
|
|
pub fn new(name: &str, delete_if_exists: bool) -> Self {
|
|
let mut path = std::env::temp_dir();
|
|
path.push(name);
|
|
if delete_if_exists && path.exists() {
|
|
std::fs::remove_dir_all(&path).unwrap();
|
|
}
|
|
std::fs::create_dir_all(&path).unwrap();
|
|
Self { path }
|
|
}
|
|
|
|
pub fn delete(&self) {
|
|
std::fs::remove_dir_all(&self.path).unwrap();
|
|
}
|
|
}
|