67 lines
1.8 KiB
Rust
67 lines
1.8 KiB
Rust
pub mod error;
|
|
use async_std::{fs::create_dir, path::Path};
|
|
|
|
use error::DBError;
|
|
|
|
const DATA: &str = "data";
|
|
|
|
#[derive(Clone)]
|
|
pub struct MoreThanText;
|
|
|
|
impl MoreThanText {
|
|
pub async fn new(dir: &str) -> Result<Self, DBError> {
|
|
let data_dir = Path::new(dir).join(DATA);
|
|
if !data_dir.is_dir().await {
|
|
match create_dir(data_dir).await {
|
|
Ok(_) => (),
|
|
Err(err) => {
|
|
let mut error = DBError::new("failed to initialize");
|
|
error.add_source(err);
|
|
return Err(error);
|
|
}
|
|
}
|
|
}
|
|
Ok(Self {})
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod init {
|
|
use super::*;
|
|
use std::error::Error;
|
|
use tempfile::tempdir;
|
|
|
|
#[async_std::test]
|
|
async fn create_data_dir() {
|
|
let dir = tempdir().unwrap();
|
|
MoreThanText::new(dir.path().to_str().unwrap())
|
|
.await
|
|
.unwrap();
|
|
let data_dir = dir.path().join(DATA);
|
|
assert!(data_dir.is_dir(), "Did not create the data directory.");
|
|
dir.close().unwrap();
|
|
}
|
|
|
|
#[async_std::test]
|
|
async fn existing_data_dir() {
|
|
let dir = tempdir().unwrap();
|
|
let data_dir = dir.path().join(DATA);
|
|
create_dir(data_dir).await.unwrap();
|
|
MoreThanText::new(dir.path().to_str().unwrap())
|
|
.await
|
|
.unwrap();
|
|
dir.close().unwrap();
|
|
}
|
|
|
|
#[async_std::test]
|
|
async fn bad_data_dir() {
|
|
match MoreThanText::new("kljsdgfhslkfrh").await {
|
|
Ok(_) => assert!(false, "This test should fail to create a data directory"),
|
|
Err(err) => {
|
|
assert_eq!(err.to_string(), "failed to initialize");
|
|
assert!(err.source().is_some(), "Must include the source error.");
|
|
}
|
|
};
|
|
}
|
|
}
|