OpenSource/teloxide
OpenSource/teloxide
1
0
Fork 0
mirror of https://github.com/teloxide/teloxide.git synced 2024-10-23 17:36:54 +02:00
Code Issues Projects Releases Packages Wiki Activity

Cleanup files when testing sqlite storage

Browse source
This commit is contained in:
Maybe Waffle 2022-03-14 22:05:15 +04:00
parent 8ae3222409
commit 3c97ed83f4
1 changed files with 20 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
tests/sqlite.rs
View file

@ -1,36 +1,47 @@
use std::{
fmt::{Debug, Display},
fs,
sync::Arc,
};
use teloxide::dispatching::dialogue::{Serializer, SqliteStorage, SqliteStorageError, Storage};
#[tokio::test(flavor = "multi_thread")]
async fn test_sqlite_json() {
let storage =
SqliteStorage::open("./test_db1.sqlite", teloxide::dispatching::dialogue::serializer::Json)
.await
.unwrap();
fs::create_dir("./test_db1").unwrap();
let storage = SqliteStorage::open(
"./test_db1/test_db1.sqlite",
teloxide::dispatching::dialogue::serializer::Json,
)
.await
.unwrap();
test_sqlite(storage).await;
fs::remove_dir_all("./test_db1").unwrap();
}
#[tokio::test(flavor = "multi_thread")]
async fn test_sqlite_bincode() {
fs::create_dir("./test_db2").unwrap();
let storage = SqliteStorage::open(
"./test_db2.sqlite",
"./test_db2/test_db2.sqlite",
teloxide::dispatching::dialogue::serializer::Bincode,
)
.await
.unwrap();
test_sqlite(storage).await;
fs::remove_dir_all("./test_db2").unwrap();
}
#[tokio::test(flavor = "multi_thread")]
async fn test_sqlite_cbor() {
let storage =
SqliteStorage::open("./test_db3.sqlite", teloxide::dispatching::dialogue::serializer::Cbor)
.await
.unwrap();
fs::create_dir("./test_db3").unwrap();
let storage = SqliteStorage::open(
"./test_db3/test_db3.sqlite",
teloxide::dispatching::dialogue::serializer::Cbor,
)
.await
.unwrap();
test_sqlite(storage).await;
fs::remove_dir_all("./test_db3").unwrap();
}
type Dialogue = String;