oximod 0.1.3

MongoDB ODM for Rust inspired by Mongoose
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

use mongodb::bson::{ doc, oid::ObjectId };
use oximod::{ set_global_client, Model };
use testresult::TestResult;
use serde::{ Deserialize, Serialize };
use futures_util::stream::StreamExt;

// Run test: cargo nextest run aggregates_documents_correctly
#[tokio::test]
async fn aggregates_documents_correctly() -> TestResult {
    dotenv::dotenv().ok();
    let mongodb_uri = std::env::var("MONGODB_URI").expect("Failed to find MONGODB_URI");

    set_global_client(mongodb_uri).await.unwrap_or_else(|e| panic!("{}", e));

    #[derive(Model, Serialize, Deserialize, Debug)]
    #[db("test")]
    #[collection("aggregate_test")]
    pub struct LogEntry {
        #[serde(skip_serializing_if = "Option::is_none")]
        _id: Option<ObjectId>,
        level: String,
        message: String,
    }

    LogEntry::clear().await?;

    let logs = vec![
        LogEntry {
            _id: None,
            level: "INFO".to_string(),
            message: "Startup complete".to_string(),
        },
        LogEntry {
            _id: None,
            level: "ERROR".to_string(),
            message: "Failed to connect".to_string(),
        },
        LogEntry {
            _id: None,
            level: "INFO".to_string(),
            message: "Listening on port 3000".to_string(),
        }
    ];

    for log in logs {
        log.save().await?;
    }

    let pipeline = vec![doc! { "$match": { "level": "INFO" } }, doc! { "$count": "info_count" }];

    let mut cursor = LogEntry::aggregate(pipeline).await?;
    let result = cursor.next().await.expect("Expected at least one document")?;
    let count = result.get_i32("info_count").expect("Expected 'info_count' field");

    assert_eq!(count, 2);

    Ok(())
}