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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112

use chain_impl_mockchain::config::ConfigParam;
use serde::{Deserialize, Serialize};
use std::{convert::TryFrom, fmt, str::FromStr};
use thiserror::Error;

/// the settings for the fees to be redistributed to
///
#[derive(Copy, Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
#[serde(rename_all = "snake_case")]
pub enum FeesGoTo {
    /// the fees will be added to the epoch's reward pot to then be distributed amongst
    /// the pools.
    Rewards,
    /// the pools don't receive any rewards to add transactions in the blocks
    /// it is instead given entirely to the treasury.
    Treasury,
}

/* Display ****************************************************************** */

impl fmt::Display for FeesGoTo {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        match self {
            Self::Rewards => "rewards".fmt(f),
            Self::Treasury => "treasury".fmt(f),
        }
    }
}

#[derive(Debug, Error)]
#[error("Invalid fees go to setting. Expect \"rewards\" or \"treasury\" ")]
pub struct FromStrFeesGoToError;

impl FromStr for FeesGoTo {
    type Err = FromStrFeesGoToError;
    fn from_str(s: &str) -> Result<Self, Self::Err> {
        match s {
            "rewards" => Ok(Self::Rewards),
            "treasury" => Ok(Self::Treasury),
            _ => Err(FromStrFeesGoToError),
        }
    }
}

/* Conversion *************************************************************** */

#[derive(Debug, Error)]
#[error("Invalid fees go to config parameter. Expect \"rewards\" or \"treasury\" ")]
pub struct TryFromFeesGoToError;

impl TryFrom<ConfigParam> for FeesGoTo {
    type Error = TryFromFeesGoToError;
    fn try_from(config_param: ConfigParam) -> Result<Self, Self::Error> {
        match config_param {
            ConfigParam::FeesInTreasury(to_treasury) => {
                let fees_go_to = if to_treasury {
                    Self::Treasury
                } else {
                    Self::Rewards
                };
                Ok(fees_go_to)
            }
            _ => Err(TryFromFeesGoToError),
        }
    }
}

impl From<FeesGoTo> for ConfigParam {
    fn from(fees_go_to: FeesGoTo) -> Self {
        let to_treasury = fees_go_to == FeesGoTo::Treasury;
        ConfigParam::FeesInTreasury(to_treasury)
    }
}

#[cfg(test)]
mod test {
    use super::*;
    use quickcheck::{Arbitrary, Gen};

    impl Arbitrary for FeesGoTo {
        fn arbitrary<G: Gen>(g: &mut G) -> Self {
            if bool::arbitrary(g) {
                Self::Rewards
            } else {
                Self::Treasury
            }
        }
    }

    quickcheck! {
        fn serde_encode_decode(fees_go_to: FeesGoTo) -> bool {
            let s = serde_yaml::to_string(&fees_go_to).unwrap();
            let fees_go_to_dec: FeesGoTo = serde_yaml::from_str(&s).unwrap();

            fees_go_to == fees_go_to_dec
        }

        fn display_from_str(fees_go_to: FeesGoTo) -> bool {
            let s = fees_go_to.to_string();
            let fees_go_to_dec: FeesGoTo = s.parse().unwrap();

            fees_go_to == fees_go_to_dec
        }

        fn convert_from_to_config_param(fees_go_to: FeesGoTo) -> bool {
            let cp = ConfigParam::from(fees_go_to);
            let fees_go_to_dec = FeesGoTo::try_from(cp).unwrap();

            fees_go_to == fees_go_to_dec
        }
    }
}