pub mod common;
pub use common::{features::*, setup::*, TestContext};
use pretty_assertions::assert_eq;
use sea_orm::{entity::prelude::*, entity::*, DatabaseConnection};
#[sea_orm_macros::test]
#[cfg(all(feature = "sqlx-postgres", feature = "postgres-array"))]
async fn main() -> Result<(), DbErr> {
let ctx = TestContext::new("collection_tests").await;
create_tables(&ctx.db).await?;
insert_collection(&ctx.db).await?;
update_collection(&ctx.db).await?;
ctx.delete().await;
Ok(())
}
pub async fn insert_collection(db: &DatabaseConnection) -> Result<(), DbErr> {
use collection::*;
assert_eq!(
Model {
id: 1,
integers: vec![1, 2, 3],
integers_opt: Some(vec![1, 2, 3]),
}
.into_active_model()
.insert(db)
.await?,
Model {
id: 1,
integers: vec![1, 2, 3],
integers_opt: Some(vec![1, 2, 3]),
}
);
assert_eq!(
Model {
id: 2,
integers: vec![10, 9],
integers_opt: None,
}
.into_active_model()
.insert(db)
.await?,
Model {
id: 2,
integers: vec![10, 9],
integers_opt: None,
}
);
assert_eq!(
Model {
id: 3,
integers: vec![],
integers_opt: Some(vec![]),
}
.into_active_model()
.insert(db)
.await?,
Model {
id: 3,
integers: vec![],
integers_opt: Some(vec![]),
}
);
Ok(())
}
pub async fn update_collection(db: &DatabaseConnection) -> Result<(), DbErr> {
use collection::*;
let model = Entity::find_by_id(1).one(db).await?.unwrap();
ActiveModel {
integers: Set(vec![4, 5, 6]),
integers_opt: Set(Some(vec![4, 5, 6])),
..model.into_active_model()
}
.update(db)
.await?;
ActiveModel {
id: Unchanged(3),
integers: Set(vec![3, 1, 4]),
integers_opt: Set(None),
}
.update(db)
.await?;
Ok(())
}