morethantext-web/src/morethantext/mod.rs

pub mod error;

use async_std::sync::{Arc, RwLock};
use error::DBError;
use std::collections::HashMap;

#[derive(Clone)]
pub struct MoreThanText {
    databases: Arc<RwLock<HashMap<String, Database>>>,
}

impl MoreThanText {
    pub async fn new() -> Self {
        Self {
            databases: Arc::new(RwLock::new(HashMap::new())),
        }
    }

    async fn create_database(&self, name: &str) -> Result<(), DBError> {
        let mut databases = self.databases.write().await;
        match databases.get(name) {
            Some(_) => Err(DBError::new("duplicate database name")),
            None => {
                let db = Database::new().await;
                databases.insert(name.to_string(), db);
                Ok(())
            }
        }
    }

    async fn use_database(&self, name: &str) -> Result<Database, DBError> {
        let databases = self.databases.read().await;
        match databases.get(name) {
            Some(db) => Ok(db.clone()),
            None => Err(DBError::new("database name not found")),
        }
    }
}

#[derive(Clone)]
struct Database;

impl Database {
    async fn new() -> Self {
        Self {}
    }

    async fn add_table(&self, _name: &str) {}
}

#[cfg(test)]
mod engine_functions {
    use super::*;

    #[async_std::test]
    async fn create_database() {
        let mtt = MoreThanText::new().await;
        mtt.create_database("smith").await.unwrap();
    }

    #[async_std::test]
    async fn database_names_must_be_unique() -> Result<(), DBError> {
        let mtt = MoreThanText::new().await;
        let msg = "duplicate database name";
        mtt.create_database("john").await.unwrap();
        match mtt.create_database("john").await {
            Ok(_) => Err(DBError::new("Duplicate names should cause error")),
            Err(err) => {
                if err.to_string() == msg {
                    Ok(())
                } else {
                    Err(DBError::new(format!(
                        "incorrect err message: got: '{}' want: '{}'",
                        err.to_string(),
                        msg
                    )))
                }
            }
        }
    }

    #[async_std::test]
    async fn use_database() -> Result<(), DBError> {
        let mtt = MoreThanText::new().await;
        let dbname = "Johnson";
        mtt.create_database(dbname).await.unwrap();
        mtt.use_database(dbname).await.unwrap();
        Ok(())
    }

    #[async_std::test]
    async fn use_missing_database() -> Result<(), DBError> {
        let error = "database name not found";
        let mtt = MoreThanText::new().await;
        match mtt.use_database("ssmith").await {
            Ok(_) => Err(DBError::new("Should raise database missing error")),
            Err(err) => {
                if err.to_string() == error {
                    Ok(())
                } else {
                    Err(DBError::new(format!(
                        "Incorrect error message: Got '{}' Want '{}'",
                        err.to_string(),
                        error
                    )))
                }
            }
        }
    }

    #[async_std::test]
    async fn create_get_table() {
        let db = "thedatabase";
        let mtt = MoreThanText::new().await;
        mtt.create_database(db).await.unwrap();
        let dbase = mtt.use_database(db).await.unwrap();
        dbase.add_table("melvin").await;
    }
}

#[cfg(test)]
mod database_functions {
    use super::*;

    #[async_std::test]
    async fn new_database() {
        Database::new().await;
    }

    #[async_std::test]
    async fn new_table() {
        let db = Database::new().await;
        db.add_table("fred").await;
    }
}