twitch_irc/message/commands/
pong.rs

1use crate::message::commands::ServerMessageParseError;
2use crate::message::IRCMessage;
3use std::convert::TryFrom;
4
5#[cfg(feature = "with-serde")]
6use {serde::Deserialize, serde::Serialize};
7
8/// A `PONG` connection-control message.
9#[derive(Debug, Clone, PartialEq, Eq)]
10#[cfg_attr(feature = "with-serde", derive(Serialize, Deserialize))]
11pub struct PongMessage {
12    /// The message that this `PongMessage` was parsed from.
13    pub source: IRCMessage,
14}
15
16impl TryFrom<IRCMessage> for PongMessage {
17    type Error = ServerMessageParseError;
18
19    fn try_from(source: IRCMessage) -> Result<PongMessage, ServerMessageParseError> {
20        if source.command != "PONG" {
21            return Err(ServerMessageParseError::MismatchedCommand(source));
22        }
23
24        Ok(PongMessage { source })
25    }
26}
27
28impl From<PongMessage> for IRCMessage {
29    fn from(msg: PongMessage) -> IRCMessage {
30        msg.source
31    }
32}
33
34#[cfg(test)]
35mod tests {
36    use crate::message::{IRCMessage, PongMessage};
37    use std::convert::TryFrom;
38
39    #[test]
40    pub fn test_basic() {
41        // this is what the Twitch servers answers "PING" with
42        let src = "PONG :tmi.twitch.tv";
43        let irc_message = IRCMessage::parse(src).unwrap();
44        let msg = PongMessage::try_from(irc_message.clone()).unwrap();
45
46        assert_eq!(
47            msg,
48            PongMessage {
49                source: irc_message
50            }
51        )
52    }
53
54    #[test]
55    pub fn test_with_argument() {
56        // this is the answer to "PING test"
57        let src = ":tmi.twitch.tv PONG tmi.twitch.tv :test";
58        let irc_message = IRCMessage::parse(src).unwrap();
59        let msg = PongMessage::try_from(irc_message.clone()).unwrap();
60
61        assert_eq!(
62            msg,
63            PongMessage {
64                source: irc_message
65            }
66        )
67    }
68}