caisin/sea-orm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Testing composite primary keys

Browse Source
This commit is contained in:
Billy Chan 2021-09-02 11:47:57 +08:00
parent e72d8a9e04
commit 8e2b7e561e
No known key found for this signature in database
GPG Key ID: A2D690CAC7DF3CC7
5 changed files with 53 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/crud/create_cake.rs
View File

@ -53,10 +53,18 @@ pub async fn test_create_cake(db: &DbConn) {
baker_id: Set(baker_insert_res.last_insert_id as i32),
..Default::default()
};
let _cake_baker_res = CakesBakers::insert(cake_baker)
let cake_baker_res = CakesBakers::insert(cake_baker.clone())
.exec(db)
.await
.expect("could not insert cake_baker");
assert_eq!(
cake_baker_res.last_insert_id,
if cfg!(feature = "sqlx-postgres") {
(cake_baker.cake_id.unwrap(), cake_baker.baker_id.unwrap())
} else {
Default::default()
}
);
assert!(cake.is_some());
let cake_model = cake.unwrap();

10
tests/crud/create_lineitem.rs
View File

@ -52,10 +52,18 @@ pub async fn test_create_lineitem(db: &DbConn) {
baker_id: Set(baker_insert_res.last_insert_id as i32),
..Default::default()
};
let _cake_baker_res = CakesBakers::insert(cake_baker)
let cake_baker_res = CakesBakers::insert(cake_baker.clone())
.exec(db)
.await
.expect("could not insert cake_baker");
assert_eq!(
cake_baker_res.last_insert_id,
if cfg!(feature = "sqlx-postgres") {
(cake_baker.cake_id.unwrap(), cake_baker.baker_id.unwrap())
} else {
Default::default()
}
);
// Customer
let customer_kate = customer::ActiveModel {

10
tests/crud/create_order.rs
View File

@ -52,10 +52,18 @@ pub async fn test_create_order(db: &DbConn) {
baker_id: Set(baker_insert_res.last_insert_id as i32),
..Default::default()
};
let _cake_baker_res = CakesBakers::insert(cake_baker)
let cake_baker_res = CakesBakers::insert(cake_baker.clone())
.exec(db)
.await
.expect("could not insert cake_baker");
assert_eq!(
cake_baker_res.last_insert_id,
if cfg!(feature = "sqlx-postgres") {
(cake_baker.cake_id.unwrap(), cake_baker.baker_id.unwrap())
} else {
Default::default()
}
);
// Customer
let customer_kate = customer::ActiveModel {

14
tests/primary_key_tests.rs
View File

@ -28,12 +28,14 @@ async fn create_metadata(db: &DatabaseConnection) -> Result<(), DbErr> {
let res = Metadata::insert(metadata.clone()).exec(db).await?;
let expected_uuid = if cfg!(feature = "sqlx-postgres") {
metadata.uuid.unwrap()
} else {
Uuid::default()
};
assert_eq!(res.last_insert_id, expected_uuid);
assert_eq!(
res.last_insert_id,
if cfg!(feature = "sqlx-postgres") {
metadata.uuid.unwrap()
} else {
Default::default()
}
);
Ok(())
}

20
tests/sequential_op_tests.rs
View File

@ -78,10 +78,18 @@ async fn init_setup(db: &DatabaseConnection) {
..Default::default()
};
let _cake_baker_res = CakesBakers::insert(cake_baker)
let cake_baker_res = CakesBakers::insert(cake_baker.clone())
.exec(db)
.await
.expect("could not insert cake_baker");
assert_eq!(
cake_baker_res.last_insert_id,
if cfg!(feature = "sqlx-postgres") {
(cake_baker.cake_id.unwrap(), cake_baker.baker_id.unwrap())
} else {
Default::default()
}
);
let customer_kate = customer::ActiveModel {
name: Set("Kate".to_owned()),
@ -211,10 +219,18 @@ async fn create_cake(db: &DatabaseConnection, baker: baker::Model) -> Option<cak
..Default::default()
};
let _cake_baker_res = CakesBakers::insert(cake_baker)
let cake_baker_res = CakesBakers::insert(cake_baker.clone())
.exec(db)
.await
.expect("could not insert cake_baker");
assert_eq!(
cake_baker_res.last_insert_id,
if cfg!(feature = "sqlx-postgres") {
(cake_baker.cake_id.unwrap(), cake_baker.baker_id.unwrap())
} else {
Default::default()
}
);
Cake::find_by_id(cake_insert_res.last_insert_id)
.one(db)