tests: Add real-world SQLite test for sea_orm integration

Author: angrynode

Cargo.toml

@@ -28,6 +28,9 @@ serde = { version = "1", features = [ "derive" ] }
 fluent-uri = { version = "0.4", features = [ "serde" ] }
 # For Sea-ORM integration
 sea-orm = { version = "2.0.0-rc.18", optional = true }
+sea-orm-migration = { version = "2.0.0-rc.18", optional = true }
+async-tempfile = { version = "0.7", optional = true }
+tokio = { version = "1", optional = true }
 
 [dev-dependencies]
 serde_json = "1"
@@ -36,7 +39,7 @@ serde_json = "1"
 magnet_force_name = []
 unknown_tracker_scheme = []
 sea_orm = [ "dep:sea-orm" ]
-test_sea_orm = [ "dep:sea-orm", "sea-orm/sqlx-sqlite" ]
+test_sea_orm = [ "dep:sea-orm", "sea-orm/sqlx-sqlite", "dep:tokio", "tokio/macros", "tokio/rt", "dep:async-tempfile", "dep:sea-orm-migration", "sea-orm-migration/runtime-tokio" ]
 
 [[test]]
 name = "magnet_force_name"
@@ -53,5 +56,5 @@ test = true
 [[test]]
 name = "sea_orm"
 path = "tests/sea_orm.rs"
-required-features = [ "sea_orm" ]
+required-features = [ "sea_orm", "test_sea_orm" ]
 test = true

tests/sea_orm.rs

@@ -40,6 +40,78 @@ mod magnet {
     impl ActiveModelBehavior for ActiveModel {}
 }
 
+mod mixed {
+    use super::*;
+
+    #[derive(Clone, Debug, PartialEq, DeriveEntityModel)]
+    #[sea_orm(table_name = "mixed")]
+    pub struct Model {
+        #[sea_orm(primary_key)]
+        pub id: i32,
+        #[sea_orm(unique)]
+        pub torrent_id: TorrentID,
+        #[sea_orm(unique)]
+        pub magnet: MagnetLink,
+    }
+
+    #[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+    pub enum Relation {}
+
+    #[async_trait::async_trait]
+    impl ActiveModelBehavior for ActiveModel {}
+
+    pub mod migration {
+        use sea_orm_migration::prelude::*;
+
+        mod m20251115_01_mixed {
+            use sea_orm_migration::{prelude::*, schema::*};
+
+            #[derive(DeriveMigrationName)]
+            pub struct Migration;
+
+            #[async_trait::async_trait]
+            impl MigrationTrait for Migration {
+                async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
+                    manager
+                        .create_table(
+                            Table::create()
+                                .table(Mixed::Table)
+                                .if_not_exists()
+                                .col(pk_auto(Mixed::Id))
+                                .col(string(Mixed::TorrentID).unique_key())
+                                .col(string(Mixed::Magnet).unique_key())
+                                .to_owned(),
+                        )
+                        .await
+                }
+
+                async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
+                    manager
+                        .drop_table(Table::drop().table(Mixed::Table).to_owned())
+                        .await
+                }
+            }
+
+            #[derive(DeriveIden)]
+            enum Mixed {
+                Table,
+                Id,
+                TorrentID,
+                Magnet,
+            }
+        }
+
+        pub struct Migrator;
+
+        #[async_trait::async_trait]
+        impl MigratorTrait for Migrator {
+            fn migrations() -> Vec<Box<dyn MigrationTrait>> {
+                vec![Box::new(m20251115_01_mixed::Migration)]
+            }
+        }
+    }
+}
+
 #[test]
 fn test_magnet_active_model() {
     let magnet =
@@ -65,3 +137,103 @@ fn test_torrentid_active_model() {
         ..Default::default()
     };
 }
+
+#[tokio::test]
+async fn test_torrent_real_db() {
+    use sea_orm_migration::*;
+
+    let tmpdir = async_tempfile::TempDir::new().await.unwrap();
+    let sqlite = tmpdir.join("db.sqlite");
+    let sqlite_str = sqlite.to_str().unwrap();
+
+    let db = sea_orm::Database::connect(&format!("sqlite://{}?mode=rwc", sqlite_str))
+        .await
+        .unwrap();
+    mixed::migration::Migrator::up(&db, None).await.unwrap();
+
+    let magnet =
+        MagnetLink::new(&std::fs::read_to_string("tests/bittorrent-v2-test.magnet").unwrap())
+            .unwrap();
+
+    let model = mixed::ActiveModel {
+        torrent_id: Set(magnet.id()),
+        magnet: Set(magnet.clone()),
+        ..Default::default()
+    }
+    .save(&db)
+    .await
+    .unwrap();
+
+    let magnet2 = MagnetLink::new(
+        &std::fs::read_to_string("tests/bittorrent-v2-hybrid-test.magnet").unwrap(),
+    )
+    .unwrap();
+
+    let model2 = mixed::ActiveModel {
+        torrent_id: Set(magnet2.id()),
+        magnet: Set(magnet2.clone()),
+        ..Default::default()
+    }
+    .save(&db)
+    .await
+    .unwrap();
+
+    let nonactive_model = model.try_into_model().unwrap();
+    let saved_model_by_id = mixed::Entity::find_by_id(nonactive_model.id)
+        .one(&db)
+        .await
+        .unwrap()
+        .unwrap();
+    assert_eq!(saved_model_by_id, nonactive_model);
+    assert_eq!(saved_model_by_id.magnet, magnet);
+    assert_eq!(nonactive_model.magnet, magnet);
+
+    let nonactive_model2 = model2.try_into_model().unwrap();
+    let saved_model_by_id2 = mixed::Entity::find_by_id(nonactive_model2.id)
+        .one(&db)
+        .await
+        .unwrap()
+        .unwrap();
+    assert_eq!(saved_model_by_id2, nonactive_model2);
+    assert_eq!(saved_model_by_id2.magnet, magnet2);
+    assert_eq!(nonactive_model2.magnet, magnet2);
+
+    // Try query by TorrentID
+    let saved_model_by_torrentid = mixed::Entity::find()
+        .filter(mixed::Column::TorrentId.eq(magnet.id()))
+        .one(&db)
+        .await
+        .unwrap()
+        .unwrap();
+    assert_eq!(saved_model_by_id, nonactive_model);
+    assert_eq!(saved_model_by_torrentid.magnet, magnet);
+
+    // Try query by MagnetLink
+    let saved_model_by_magnet = mixed::Entity::find()
+        .filter(mixed::Column::Magnet.eq(magnet.clone()))
+        .one(&db)
+        .await
+        .unwrap()
+        .unwrap();
+    assert_eq!(saved_model_by_id, nonactive_model);
+    assert_eq!(saved_model_by_magnet.magnet, magnet);
+
+    // Try listing torrents
+    let mut found_one = false;
+    let mut found_two = false;
+    let list = mixed::Entity::find().all(&db).await.unwrap();
+    println!("{:?}", list);
+    for entry in list {
+        if entry.magnet == magnet && entry.torrent_id == magnet.id() {
+            found_one = true;
+            continue;
+        }
+
+        if entry.magnet == magnet2 && entry.torrent_id == magnet2.id() {
+            found_two = true;
+        }
+    }
+
+    assert!(found_one);
+    assert!(found_two);
+}