Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -16,136 +16,82 @@ pub struct PersonEntity { | |
| pub created_at: i64, | ||
| } | ||
|
|
||
| #[cfg(test)] | ||
| mod test { | ||
| use super::PersonEntity; | ||
| use crate::entities::person::model::UpsertPerson; | ||
| use pretty_assertions::assert_eq; | ||
| use scyllax::prelude::*; | ||
|
|
||
| #[test] | ||
| fn test_pks() { | ||
| assert_eq!(PersonEntity::pks(), vec!["id".to_string()]); | ||
| } | ||
|
|
||
| #[test] | ||
| fn test_keys() { | ||
| assert_eq!( | ||
| PersonEntity::keys(), | ||
| vec![ | ||
| "id".to_string(), | ||
| "email".to_string(), | ||
| "age".to_string(), | ||
| "\"createdAt\"".to_string() | ||
| ] | ||
| ); | ||
| } | ||
|
|
||
| #[test] | ||
| fn test_upsert_v1() { | ||
| let upsert = UpsertPerson { | ||
| id: v1_uuid(), | ||
| email: MaybeUnset::Set("[email protected]".to_string()), | ||
| age: MaybeUnset::Set(Some(21)), | ||
| created_at: MaybeUnset::Unset, | ||
| }; | ||
|
|
||
| let (query, values) = upsert.query().expect("failed to parse into query"); | ||
|
|
||
| assert_eq!( | ||
| query, | ||
| r#"update "person" set "email" = ?, "age" = ? where "id" = ?;"# | ||
| ); | ||
|
|
||
| let mut result_values = SerializedValues::new(); | ||
| result_values | ||
| .add_value(&upsert.email) | ||
| .expect("failed to add value"); | ||
| result_values | ||
| .add_value(&upsert.age) | ||
| .expect("failed to add value"); | ||
| result_values | ||
| .add_value(&upsert.id) | ||
| .expect("failed to add value"); | ||
|
|
||
| assert_eq!(values, result_values); | ||
| } | ||
|
|
||
| #[test] | ||
| fn test_upsert_v2() { | ||
| let upsert = UpsertPerson { | ||
| id: v1_uuid(), | ||
| email: MaybeUnset::Set("[email protected]".to_string()), | ||
| age: MaybeUnset::Unset, | ||
| created_at: MaybeUnset::Unset, | ||
| }; | ||
|
|
||
| let (query, values) = upsert.query().expect("failed to parse into query"); | ||
|
|
||
| assert_eq!(query, r#"update "person" set "email" = ? where "id" = ?;"#); | ||
|
|
||
| let mut result_values = SerializedValues::new(); | ||
| result_values | ||
| .add_value(&upsert.email) | ||
| .expect("failed to add value"); | ||
| result_values | ||
| .add_value(&upsert.id) | ||
| .expect("failed to add value"); | ||
|
|
||
| assert_eq!(values, result_values); | ||
| } | ||
| } | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,10 +1,13 @@ | ||
| use scyllax::{delete_query, prelude::*}; | ||
| use uuid::Uuid; | ||
|
|
||
| /// Load all queries for this entity | ||
| #[tracing::instrument(skip(db))] | ||
| pub async fn load(db: &mut Executor) -> anyhow::Result<()> { | ||
| let _ = GetPersonById::prepare(db).await; | ||
| let _ = GetPeopleByIds::prepare(db).await; | ||
| let _ = GetPersonByEmail::prepare(db).await; | ||
| let _ = DeletePersonById::prepare(db).await; | ||
|
|
||
| Ok(()) | ||
| } | ||
|
|
@@ -40,3 +43,64 @@ pub struct GetPersonByEmail { | |
| /// The email address of the [`super::model::PersonEntity`] to get | ||
| pub email: String, | ||
| } | ||
|
|
||
| /// Get a [`super::model::PersonEntity`] by its [`uuid::Uuid`] | ||
| #[delete_query( | ||
| query = "delete from person where id = ?", | ||
| entity_type = "super::model::PersonEntity" | ||
| )] | ||
| pub struct DeletePersonById { | ||
| /// The [`uuid::Uuid`] of the [`super::model::PersonEntity`] to get | ||
| pub id: Uuid, | ||
| } | ||
|
|
||
| #[cfg(test)] | ||
| mod test { | ||
| use super::*; | ||
| use scyllax::prelude::*; | ||
|
Check warning on line 60 in example/src/entities/person/queries.rs
|
||
|
|
||
| #[test] | ||
| fn test_get_person_by_id() { | ||
| let _query = GetPersonById { id: v1_uuid() }; | ||
|
|
||
| assert_eq!( | ||
| GetPersonById::query(), | ||
| r#"select id, email, age, "createdAt" from person where id = ? limit 1"# | ||
| ); | ||
| } | ||
|
|
||
| #[test] | ||
| fn test_get_people_by_ids() { | ||
| let _query = GetPeopleByIds { | ||
| ids: vec![v1_uuid(), v1_uuid()], | ||
| limit: 10, | ||
| }; | ||
|
|
||
| assert_eq!( | ||
| GetPeopleByIds::query(), | ||
| r#"select id, email, age, "createdAt" from person where id in ? limit ?"# | ||
| ); | ||
| } | ||
|
|
||
| #[test] | ||
| fn test_get_person_by_email() { | ||
| let _query = GetPersonByEmail { | ||
| email: "[email protected]".to_string(), | ||
| }; | ||
|
|
||
| assert_eq!( | ||
| GetPersonByEmail::query(), | ||
| r#"select id, email, age, "createdAt" from person_by_email where email = ? limit 1"# | ||
| ); | ||
| } | ||
|
|
||
| #[test] | ||
| fn test_delete_person_by_id() { | ||
| let _query = DeletePersonById { id: v1_uuid() }; | ||
|
|
||
| assert_eq!( | ||
| DeletePersonById::query(), | ||
| r#"delete from person where id = ?"# | ||
| ); | ||
| } | ||
| } | ||
Oops, something went wrong.