mithril-common 0.7.14

Common types, interfaces, and utilities for Mithril nodes.
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

//! A set of extension traits to add test utilities to this crate `messages`

use std::collections::BTreeSet;

use strum::IntoEnumIterator;

use crate::messages::{DiscontinuedSignedEntityType, SignedEntityTypeDiscriminantsMessage};

/// Extension trait adding test utilities to [SignedEntityTypeDiscriminantsMessage]
pub trait SignedEntityTypeDiscriminantsMessageTestExtension {
    /// `TEST ONLY` - Get all the discontinued discriminants
    fn all_discontinued() -> BTreeSet<SignedEntityTypeDiscriminantsMessage>;
}

impl SignedEntityTypeDiscriminantsMessageTestExtension for SignedEntityTypeDiscriminantsMessage {
    fn all_discontinued() -> BTreeSet<SignedEntityTypeDiscriminantsMessage> {
        DiscontinuedSignedEntityType::all()
            .into_iter()
            .map(SignedEntityTypeDiscriminantsMessage::Discontinued)
            .collect()
    }
}

/// Extension trait adding test utilities to [DiscontinuedSignedEntityType]
pub trait DiscontinuedSignedEntityTypeTestExtension {
    /// `TEST ONLY` - Get all the discontinued discriminants
    fn all() -> BTreeSet<DiscontinuedSignedEntityType>;
}

impl DiscontinuedSignedEntityTypeTestExtension for DiscontinuedSignedEntityType {
    fn all() -> BTreeSet<DiscontinuedSignedEntityType> {
        DiscontinuedSignedEntityType::iter().collect()
    }
}