rustygram 0.1.4

RustyGram is a minimal and blazing fast telegram notification framework for Rust
Documentation 
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

#[cfg(test)]
mod test {
    use crate::{
        types::{SendMessageOption, SendMessageParseMode},
        *,
    }; // import lib.rs
    use std::{env, sync::Arc};

    /// Reading bot token, and chat id for inteacting with telegram bot.
    ///
    /// Set environment variable as follows before running test
    /// `TEST_BOT_TOKEN` - telegram bot's token
    /// `TEST_CHAT_ID` - telegram bot's chat id
    ///
    /// These environment variable names are used for testing purpose only.
    fn get_keys() -> Result<(String, String), String> {
        dotenv::dotenv().ok();

        let bot_token = env::var("TEST_BOT_TOKEN")
            .map_err(|_| "TEST_BOT_TOKEN environment variable is not set")?;
        let chat_id =
            env::var("TEST_CHAT_ID").map_err(|_| "TEST_CHAT_ID environment variable is not set")?;

        Ok((bot_token, chat_id))
    }

    /// Create test bot instance with given bot token, and chat id.
    fn get_bot() -> Bot {
        let keys = get_keys().expect("Failed to read environment variables");
        create_bot(&keys.0, &keys.1)
    }

    #[tokio::test]
    async fn test_send_message_simple() {
        let bot = get_bot();

        let res = bot.send_message("test_send_message_simple", None).await;
        assert!(res.is_ok());
    }

    #[tokio::test]
    async fn test_send_markdown_message() {
        let bot = get_bot();
        let m1 = bot.send_message(
            r#"\[rustygram\] __MarkdownV2__  
                *async msg 1*  
                `Tap to copy this text`\.  
                You can visit my [website](https://github.com/extremelySunnyYK)\.  
                Woot\!"#,
            Some(SendMessageOption {
                parse_mode: Some(SendMessageParseMode::MarkdownV2),
            }),
        );

        let m2 = bot.send_message(
            r#"\[rustygram\] __MarkdownV2__  
                *async msg 2*  
                `Tap to copy this text`\.  
                You can visit my [website](https://github.com/extremelySunnyYK)\.  
                Woot\!"#,
            Some(SendMessageOption {
                parse_mode: Some(SendMessageParseMode::MarkdownV2),
            }),
        );

        assert!(m1.await.is_ok());
        assert!(m2.await.is_ok());
    }

    #[tokio::test]
    async fn test_send_html_message() {
        let bot = get_bot();
        let m1 = bot.send_message(
            r#"[rustygram] <u>HTML style</u> - <b>async msg 1</b>
                <code>Tap to copy this text</code>.
                You can visit my <a href="https://github.com/ExtremelySunnyYK">website</a>.
                Woot!"#,
            Some(SendMessageOption {
                parse_mode: Some(SendMessageParseMode::HTML),
            }),
        );
        let m2 = bot.send_message(
            r#"[rustygram] <u>HTML style</u> - <b>async msg 2</b>
                    <code>Tap to copy this text</code>.
                    You can visit my <a href="https://github.com/ExtremelySunnyYK">website</a>.
                    Woot!"#,
            Some(SendMessageOption {
                parse_mode: Some(SendMessageParseMode::HTML),
            }),
        );

        assert!(m1.await.is_ok());
        assert!(m2.await.is_ok());
    }
    #[tokio::test]
    async fn test_send_csv() {
        use std::fs::File;
        use std::io::Write;
        use tempfile::tempdir;

        // Create a temporary directory and file for testing
        let temp_dir = tempdir().unwrap();
        let file_path = temp_dir.path().join("test.csv");
        let mut file = File::create(&file_path).unwrap();
        writeln!(file, "name,age\nJohn,30\nJane,25").unwrap();

        let bot = get_bot();
        let result = bot.send_csv(file_path.to_str().unwrap(), "test_csv").await;

        // If the result is an error, print the error message
        if let Err(e) = &result {
            println!("Error: {}", e);
        }

        // Assert that the result is Ok
        assert!(result.is_ok());
    }
}