HoovesSocial/hippofish
2
0
Fork 0
Code Issues 3 Pull requests 1 Projects Releases Packages Wiki Activity Actions

add mock database

Browse source
This commit is contained in:
Namekuji 2023-05-25 09:42:59 -04:00
parent d453eaf4ae
commit 535eb3540b
No known key found for this signature in database
GPG key ID: B541BD6E646CABC7
2 changed files with 20 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
packages/backend/native-utils/crates/database/Cargo.toml
View file

@ -7,5 +7,5 @@ edition = "2021"
[dependencies] [dependencies]
once_cell = "1.17.1" once_cell = "1.17.1"
sea-orm = { version = "0.11.3", features = ["sqlx-postgres", "runtime-tokio-rustls"] } sea-orm = { version = "0.11.3", features = ["sqlx-postgres", "runtime-tokio-rustls", "mock"] }
thiserror = "1.0.40" thiserror = "1.0.40"

22
packages/backend/native-utils/crates/database/src/lib.rs
View file

@ -1,11 +1,13 @@
pub mod error; pub mod error;
use once_cell::sync::OnceCell; use once_cell::sync::{Lazy, OnceCell};
use sea_orm::{Database, DatabaseConnection}; use sea_orm::{Database, DatabaseBackend, DatabaseConnection, MockDatabase};
use crate::error::Error; use crate::error::Error;
static DB_CONN: OnceCell<DatabaseConnection> = OnceCell::new(); static DB_CONN: OnceCell<DatabaseConnection> = OnceCell::new();
static DB_MOCK: Lazy<DatabaseConnection> =
Lazy::new(|| MockDatabase::new(DatabaseBackend::Postgres).into_connection());
pub async fn init_database(connection_uri: impl Into<String>) -> Result<(), Error> { pub async fn init_database(connection_uri: impl Into<String>) -> Result<(), Error> {
let conn = Database::connect(connection_uri.into()).await?; let conn = Database::connect(connection_uri.into()).await?;
@ -14,5 +16,19 @@ pub async fn init_database(connection_uri: impl Into<String>) -> Result<(), Erro
} }
pub fn get_database() -> Result<&'static DatabaseConnection, Error> { pub fn get_database() -> Result<&'static DatabaseConnection, Error> {
DB_CONN.get().ok_or(Error::Uninitialized) if cfg!(test) {
Ok(&DB_MOCK)
} else {
DB_CONN.get().ok_or(Error::Uninitialized)
}
}
#[cfg(test)]
mod tests {
use super::get_database;
#[test]
fn can_get_mock_without_initialization() {
assert!(get_database().is_ok());
}
} }