greed 0.6.0

A rust tool to automate trades 📈
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
113
114
115
116
117
118

use crate::cli::history::output_format::OutputFormatArgs;
use crate::cli::history::range::HistoryRangeArgs;
use clap::Args;
use greed::asset::AssetSymbol;

#[derive(Args, Debug)]
pub struct QuoteHistoryArgs {
    /// Asset symbols to filter to; omit to include all symbols.
    #[arg(value_name = "SYMBOLS")]
    pub symbols: Vec<AssetSymbol>,
    #[command(flatten)]
    pub history_range: HistoryRangeArgs,
    #[command(flatten)]
    pub output_format: OutputFormatArgs,
}

#[cfg(test)]
mod test {
    use crate::cli::Cli;
    use clap::Parser;
    use greed::asset::AssetSymbol;
    use greed::output_format::OutputFormat;

    #[test]
    fn parses_with_no_symbols_defaults_to_empty_filter() {
        let cli = Cli::try_parse_from(["greed", "history", "quote-history"])
            .expect("should parse with no symbols");

        let crate::cli::Command::History(history_args) = cli.command else {
            panic!("expected History command");
        };
        let crate::cli::history::HistoryCommand::QuoteHistory(args) = history_args.command else {
            panic!("expected QuoteHistory subcommand");
        };

        assert_eq!(args.symbols, Vec::<AssetSymbol>::new());
        assert!(args.history_range.days.is_none());
    }

    #[test]
    fn parses_multiple_positional_symbols_before_range_flags() {
        let cli = Cli::try_parse_from([
            "greed",
            "history",
            "quote-history",
            "VTI",
            "AAPL",
            "--days",
            "3",
        ])
        .expect("should parse with symbols and a range flag");

        let crate::cli::Command::History(history_args) = cli.command else {
            panic!("expected History command");
        };
        let crate::cli::history::HistoryCommand::QuoteHistory(args) = history_args.command else {
            panic!("expected QuoteHistory subcommand");
        };

        assert_eq!(
            args.symbols,
            vec![AssetSymbol::new("VTI"), AssetSymbol::new("AAPL")]
        );
        assert_eq!(args.history_range.days, Some(3));
    }

    #[test]
    fn rejects_combining_days_and_last_hour() {
        let result = Cli::try_parse_from([
            "greed",
            "history",
            "quote-history",
            "--days",
            "3",
            "--last-hour",
        ]);

        assert!(result.is_err());
    }

    #[test]
    fn defaults_to_text_when_format_omitted() {
        let cli = Cli::try_parse_from(["greed", "history", "quote-history"])
            .expect("should parse with no format flag");

        let crate::cli::Command::History(history_args) = cli.command else {
            panic!("expected History command");
        };
        let crate::cli::history::HistoryCommand::QuoteHistory(args) = history_args.command else {
            panic!("expected QuoteHistory subcommand");
        };

        assert_eq!(args.output_format.format, OutputFormat::Text);
    }

    #[test]
    fn parses_explicit_json_format() {
        let cli = Cli::try_parse_from(["greed", "history", "quote-history", "--format", "json"])
            .expect("should parse with an explicit json format");

        let crate::cli::Command::History(history_args) = cli.command else {
            panic!("expected History command");
        };
        let crate::cli::history::HistoryCommand::QuoteHistory(args) = history_args.command else {
            panic!("expected QuoteHistory subcommand");
        };

        assert_eq!(args.output_format.format, OutputFormat::Json);
    }

    #[test]
    fn rejects_invalid_format_value() {
        let result =
            Cli::try_parse_from(["greed", "history", "quote-history", "--format", "bogus"]);

        assert!(result.is_err());
    }
}