feat(test): add even more db query tests (#806)

* Test 'get_block_children'

* Test 'get_spending_transaction'

* Add test 'test_ledger_updates_by_address'

* Add test 'test_ledger_updates_by_milestone'

* Add test 'test_spent_unspent_ledger_updates'

* Use 'count_documents'

* Prep milestone tests

* Add 'len' to 'MongoDbCollectionExt' trait with enabled 'rand' feature

* Use 'MongoDbCollectionExt' count

* Clippy

* Fmt

* Go away now

* Rename claiming test module

* Make MongoDbCollection trait async

* Fix merge

* Simplify collection creation in launch code

* Format

* Slight refactor

* Use generic setup helper

* Improve spending transaction test

* Temporarily create only 1 unspent output for the spending tx

* Ensure all inputs fetch the spending block

* Format

* Fix docs

* PR comments 1

* Simplify

* Cleanup

* PR comments 2

* Format

* Add protocol parameters test

* Format and clean up

* PR comments 3

* Fix

* Add treasury update tests

Co-authored-by: Jochen Görtler <[email protected]>

Author: Alex6323

src/types/stardust/block/payload/milestone/milestone_id.rs

@@ -10,7 +10,7 @@ use serde::{Deserialize, Serialize};
 use crate::types::util::bytify;
 
 /// Uniquely identifies a milestone.
-#[derive(Copy, Clone, Debug, PartialEq, Eq, Serialize, Deserialize)]
+#[derive(Copy, Clone, Debug, PartialEq, Eq, Serialize, Deserialize, Hash)]
 #[serde(transparent)]
 pub struct MilestoneId(#[serde(with = "bytify")] pub [u8; Self::LENGTH]);
 

src/types/tangle/milestone_index.rs

@@ -10,7 +10,7 @@ use serde::{Deserialize, Serialize};
 
 /// The index of a given milestone.
 #[derive(
-    Clone, Copy, PartialOrd, Ord, PartialEq, Eq, Debug, Default, Serialize, Deserialize, Add, Sub, Deref, DerefMut,
+    Clone, Copy, PartialOrd, Ord, PartialEq, Eq, Debug, Default, Serialize, Deserialize, Add, Sub, Deref, DerefMut, Hash,
 )]
 #[serde(transparent)]
 pub struct MilestoneIndex(pub u32);

tests/outputs.rs

@@ -5,7 +5,6 @@ mod common;
 
 #[cfg(feature = "rand")]
 mod test_rand {
-
     use chronicle::{
         db::collections::{OutputCollection, OutputMetadataResult, OutputWithMetadataResult},
         types::{

tests/outputs_claiming.rs

@@ -5,7 +5,6 @@ mod common;
 
 #[cfg(feature = "rand")]
 mod test_rand {
-
     use chronicle::{
         db::collections::OutputCollection,
         types::{

tests/protocol_updates.rs

@@ -5,7 +5,6 @@ mod common;
 
 #[cfg(feature = "rand")]
 mod test_rand {
-
     use bee_block_stardust::rand::number::rand_number_range;
     use chronicle::{
         db::{collections::ProtocolUpdateCollection, MongoDbCollectionExt},

tests/treasury_updates.rs

@@ -0,0 +1,92 @@
+// Copyright 2022 IOTA Stiftung
+// SPDX-License-Identifier: Apache-2.0
+
+mod common;
+
+#[cfg(feature = "rand")]
+mod test_rand {
+    use std::collections::HashMap;
+
+    use bee_block_stardust::rand::number::rand_number_range;
+    use chronicle::{
+        db::{collections::TreasuryCollection, MongoDbCollectionExt},
+        types::{
+            stardust::block::payload::{MilestoneId, TreasuryTransactionPayload},
+            tangle::MilestoneIndex,
+        },
+    };
+
+    use super::common::{setup_collection, setup_database, teardown};
+
+    #[tokio::test]
+    async fn test_insert_treasury_updates() {
+        let db = setup_database("test-insert-treasury-updates").await.unwrap();
+        let update_collection = setup_collection::<TreasuryCollection>(&db).await.unwrap();
+
+        let ctx = bee_block_stardust::protocol::protocol_parameters();
+        let mut milestones = HashMap::new();
+
+        for (milestone_index, payload) in
+            (0..10u32).map(|milestone_index| (milestone_index, TreasuryTransactionPayload::rand(&ctx)))
+        {
+            milestones.insert(milestone_index, payload.input_milestone_id);
+
+            update_collection
+                .insert_treasury(milestone_index.into(), &payload)
+                .await
+                .unwrap();
+        }
+
+        assert_eq!(update_collection.count().await.unwrap(), 10);
+        assert_eq!(
+            &update_collection
+                .get_latest_treasury()
+                .await
+                .unwrap()
+                .unwrap()
+                .milestone_id,
+            milestones.get(&9).unwrap()
+        );
+
+        teardown(db).await;
+    }
+
+    #[tokio::test]
+    async fn test_insert_many_treasury_updates() {
+        let db = setup_database("test-insert-many-treasury-updates").await.unwrap();
+        let update_collection = setup_collection::<TreasuryCollection>(&db).await.unwrap();
+
+        let mut milestones = HashMap::new();
+
+        let treasury_updates = (0..10u32)
+            .map(|milestone_index| {
+                (
+                    MilestoneIndex::from(milestone_index),
+                    MilestoneId::rand(),
+                    rand_number_range(1000..10000000u64),
+                )
+            })
+            .inspect(|(milestone_index, milestone_id, _)| {
+                milestones.insert(milestone_index.0, *milestone_id);
+            })
+            .collect::<Vec<_>>();
+
+        update_collection
+            .insert_treasury_payloads(treasury_updates)
+            .await
+            .unwrap();
+
+        assert_eq!(update_collection.count().await.unwrap(), 10);
+        assert_eq!(
+            &update_collection
+                .get_latest_treasury()
+                .await
+                .unwrap()
+                .unwrap()
+                .milestone_id,
+            milestones.get(&9).unwrap()
+        );
+
+        teardown(db).await;
+    }
+}