This commit is contained in:
subcrip 2024-03-29 19:14:21 +08:00
parent 77243161f7
commit 8c8ce903db
Signed by: subcrip
SSH Key Fingerprint: SHA256:dFPFi68d8C87YkFkEBU4TkcrYRySWpekRR1hbnDWUCw
2 changed files with 49 additions and 6 deletions

View File

@ -29,20 +29,53 @@ impl OAuthDatabase {
pub mod db_tests { pub mod db_tests {
use super::*; use super::*;
pub struct DBTestEntryType {
name: String,
age: i32,
}
impl DBTestEntryType {
pub fn new(name: String, age: i32) -> Self {
Self { name, age }
}
}
pub struct DBTestComponents { pub struct DBTestComponents {
d: OAuthDatabase d: Option<OAuthDatabase>,
dropped: bool,
} }
impl DBTestComponents { impl DBTestComponents {
fn default() -> Self {
Self { d: None, dropped: false }
}
pub async fn new(mut d: OAuthDatabase) -> Result<Self, tokio_postgres::Error> { pub async fn new(mut d: OAuthDatabase) -> Result<Self, tokio_postgres::Error> {
d.psql_execute("create table rust_test(name text, age int)", &[]).await?; d.psql_execute("create table rust_test(name text, age int)", &[]).await?;
Ok(Self { d }) Ok(Self { d: Some(d), dropped: false })
}
pub async fn create(&mut self, item: DBTestEntryType) -> Result<(), tokio_postgres::Error> {
self.d.as_mut().unwrap().psql_execute("insert into rust_test (name, age) values ($1, $2)", &[&item.name, &item.age]).await?;
Ok(())
}
async fn destroy(&mut self) -> Result<(), tokio_postgres::Error> {
eprintln!("destroy() called.");
self.d.as_mut().unwrap().psql_execute("drop table rust_test", &[]).await?;
eprintln!("destroy() finished.");
Ok(())
} }
} }
impl Drop for DBTestComponents { impl Drop for DBTestComponents {
fn drop(&mut self) { fn drop(&mut self) {
futures::executor::block_on(self.d.psql_execute("drop table rust_test", &[])).unwrap(); eprintln!("drop() called.");
if !self.dropped {
let mut this = DBTestComponents::default();
std::mem::swap(&mut this, self);
this.dropped = true;
tokio::spawn(async move {this.destroy().await});
}
} }
} }
} }

View File

@ -24,8 +24,18 @@ async fn main() {
println!("{}", json!(a)); println!("{}", json!(a));
} }
#[tokio::test] // #[tokio::test]
async fn test_table_level_operations() { // async fn test_table_level_operations() {
// use db::db_tests;
// let _b = db_tests::DBTestComponents::new(db::OAuthDatabase::connect("localhost", 5432, "postgres", "configjson", "myoauth").await.unwrap()).await.unwrap();
// }
#[tokio::test(flavor = "multi_thread")]
async fn test_entry_level_operations() {
use db::db_tests; use db::db_tests;
let _b = db_tests::DBTestComponents::new(db::OAuthDatabase::connect("localhost", 5432, "postgres", "configjson", "myoauth").await.unwrap()); eprintln!("1");
let mut b = db_tests::DBTestComponents::new(db::OAuthDatabase::connect("localhost", 5432, "postgres", "configjson", "myoauth").await.unwrap()).await.unwrap();
eprintln!("2");
assert!(matches!(b.create(db_tests::DBTestEntryType::new("subcrip".to_string(), 19)).await, Ok(())));
eprintln!("3");
} }