akulij/gongbotrs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: use unwrap instead of is_ok in tests to get error messages

Browse Source
This commit is contained in:
Akulij 2025-04-18 13:16:41 +03:00
parent 95cc880b4f
commit e56b7284df
1 changed files with 13 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
src/db/tests.rs
View File

@ -16,31 +16,29 @@ async fn setup_db() -> DB {
async fn test_get_media() { async fn test_get_media() {
let mut db = setup_db().await; let mut db = setup_db().await;
let result = db.drop_media("test_get_media_literal").await; let result = db.drop_media("test_get_media_literal").await.unwrap();
assert!(result.is_ok());
let media_items = db.get_media("test_get_media_literal").await.unwrap(); let media_items = db.get_media("test_get_media_literal").await.unwrap();
assert_eq!(media_items.len(), 0); assert_eq!(media_items.len(), 0);
let result = db let result = db
.add_media("test_get_media_literal", "photo", "file_id_1") .add_media("test_get_media_literal", "photo", "file_id_1")
.await; .await
assert!(result.is_ok()); .unwrap();
let media_items = db.get_media("test_get_media_literal").await.unwrap(); let media_items = db.get_media("test_get_media_literal").await.unwrap();
assert_eq!(media_items.len(), 1); assert_eq!(media_items.len(), 1);
let result = db let result = db
.add_media("test_get_media_literal", "video", "file_id_2") .add_media("test_get_media_literal", "video", "file_id_2")
.await; .await
assert!(result.is_ok()); .unwrap();
let media_items = db.get_media("test_get_media_literal").await.unwrap(); let media_items = db.get_media("test_get_media_literal").await.unwrap();
assert_eq!(media_items.len(), 2); assert_eq!(media_items.len(), 2);
// Clean up after test // Clean up after test
let result = db.drop_media("test_get_media_literal").await; let result = db.drop_media("test_get_media_literal").await.unwrap();
assert!(result.is_ok());
} }
#[tokio::test] #[tokio::test]
@ -51,11 +49,9 @@ async fn test_add_media() {
let media_type = "photo"; let media_type = "photo";
let file_id = "LjaldhAOh"; let file_id = "LjaldhAOh";
let result = db.drop_media(literal).await; let result = db.drop_media(literal).await.unwrap();
assert!(result.is_ok());
let result = db.add_media(literal, media_type, file_id).await; let result = db.add_media(literal, media_type, file_id).await.unwrap();
assert!(result.is_ok());
// Verify that the media was added is correct // Verify that the media was added is correct
let media_items = db.get_media(literal).await.unwrap(); let media_items = db.get_media(literal).await.unwrap();
@ -65,8 +61,7 @@ async fn test_add_media() {
assert_eq!(media_items[0].file_id, file_id); assert_eq!(media_items[0].file_id, file_id);
// Clean up after test // Clean up after test
let result = db.drop_media(literal).await; let result = db.drop_media(literal).await.unwrap();
assert!(result.is_ok());
} }
#[tokio::test] #[tokio::test]
@ -75,21 +70,19 @@ async fn test_drop_media() {
let result = db let result = db
.add_media("test_drop_media_literal", "photo", "file_id_1") .add_media("test_drop_media_literal", "photo", "file_id_1")
.await; .await
assert!(result.is_ok()); .unwrap();
// Verify that the media was added // Verify that the media was added
let media_items = db.get_media("test_drop_media_literal").await.unwrap(); let media_items = db.get_media("test_drop_media_literal").await.unwrap();
assert_eq!(media_items.len(), 1); assert_eq!(media_items.len(), 1);
let result = db.drop_media("test_drop_media_literal").await; let result = db.drop_media("test_drop_media_literal").await.unwrap();
assert!(result.is_ok());
// Verify that the media has been dropped // Verify that the media has been dropped
let media_items = db.get_media("test_drop_media_literal").await.unwrap(); let media_items = db.get_media("test_drop_media_literal").await.unwrap();
assert_eq!(media_items.len(), 0); assert_eq!(media_items.len(), 0);
// Clean up after test // Clean up after test
let result = db.drop_media("test_drop_media_literal").await; let result = db.drop_media("test_drop_media_literal").await.unwrap();
assert!(result.is_ok());
} }