use mongodb::bson::{doc, oid::ObjectId};
use oximod::Model;
use serde::{Deserialize, Serialize};
use testresult::TestResult;
mod common;
use common::init;
#[tokio::test]
async fn updates_multiple_documents_correctly() -> TestResult {
init().await?;
#[derive(Model, Serialize, Deserialize, Debug)]
#[db("test")]
#[collection("update")]
pub struct User {
#[serde(skip_serializing_if = "Option::is_none")]
_id: Option<ObjectId>,
name: String,
age: i32,
active: bool,
}
User::clear().await?;
let users = vec![
User::default().name("User1").age(70).active(true),
User::default().name("User2").age(65).active(true),
User::default().name("User3").age(40).active(true),
];
for user in users {
user.save().await?;
}
let result = User::update(
doc! { "age": { "$gte": 65 } },
doc! { "$set": { "active": false } },
)
.await?;
assert_eq!(result.matched_count, 2);
assert_eq!(result.modified_count, 2);
Ok(())
}
#[tokio::test]
async fn updates_multiple_documents_invalid_update_fails() -> TestResult {
init().await?;
#[derive(Model, Serialize, Deserialize, Debug)]
#[db("test")]
#[collection("update_invalid")]
pub struct User {
#[serde(skip_serializing_if = "Option::is_none")]
_id: Option<ObjectId>,
name: String,
age: i32,
active: bool,
}
User::clear().await?;
let users = vec![
User::default().name("User1").age(70).active(true),
User::default().name("User2").age(65).active(true),
User::default().name("User3").age(40).active(true),
];
for user in users {
user.save().await?;
}
let result = User::update(doc! { "age": { "$gte": 65 } }, doc! { "$set": "invalid" }).await;
assert!(result.is_err());
Ok(())
}
#[tokio::test]
async fn updates_optional_email_to_valid() -> TestResult {
init().await?;
#[derive(Model, Serialize, Deserialize, Debug)]
#[db("test")]
#[collection("update_optional_email_valid")]
pub struct User {
#[serde(skip_serializing_if = "Option::is_none")]
_id: Option<ObjectId>,
name: String,
age: i32,
active: bool,
#[validate(email)]
#[serde(skip_serializing_if = "Option::is_none")]
email: Option<String>,
}
User::clear().await?;
let users = vec![
User::default()
.name("User1")
.age(70)
.active(true)
.email("u1@example.com"),
User::default().name("User2").age(65).active(true),
User::default()
.name("User3")
.age(40)
.active(true)
.email("u3@example.com"),
];
for user in users {
user.save().await?;
}
let result = User::update(
doc! { "age": { "$gte": 65 } },
doc! { "$set": { "email": "user@example.com" } },
)
.await?;
assert_eq!(result.matched_count, 2);
assert_eq!(result.modified_count, 2);
Ok(())
}