Implement Message::get_data_strategy()

A method that returns a proptest strategy that creates `Message::GetData` instances.
ZcashFoundation · Jul 6, 2021 · 494460b · 494460b
1 parent 8b94baf
commit 494460b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/zebra-network/src/protocol/external/arbitrary.rs b/zebra-network/src/protocol/external/arbitrary.rs
@@ -91,4 +91,11 @@ impl Message {
     pub fn inv_strategy() -> BoxedStrategy<Self> {
         any::<Vec<InventoryHash>>().prop_map(Message::Inv).boxed()
     }
+
+    /// Create a strategy that only generates [`Message::GetData`] messages.
+    pub fn get_data_strategy() -> BoxedStrategy<Self> {
+        any::<Vec<InventoryHash>>()
+            .prop_map(Message::GetData)
+            .boxed()
+    }
 }