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
use crate::message::commands::ServerMessageParseError;
use crate::message::IRCMessage;
use std::convert::TryFrom;
#[derive(Debug, Clone, PartialEq)]
pub struct PingMessage {
pub source: IRCMessage,
}
impl TryFrom<IRCMessage> for PingMessage {
type Error = ServerMessageParseError;
fn try_from(source: IRCMessage) -> Result<PingMessage, ServerMessageParseError> {
if source.command != "PING" {
return Err(ServerMessageParseError::MismatchedCommand(source));
}
Ok(PingMessage { source })
}
}
impl From<PingMessage> for IRCMessage {
fn from(msg: PingMessage) -> IRCMessage {
msg.source
}
}
#[cfg(test)]
mod tests {
use crate::message::{IRCMessage, PingMessage};
use std::convert::TryFrom;
#[test]
pub fn test_basic() {
let src = ":tmi.twitch.tv PING";
let irc_message = IRCMessage::parse(src).unwrap();
let msg = PingMessage::try_from(irc_message.clone()).unwrap();
assert_eq!(
msg,
PingMessage {
source: irc_message
}
)
}
#[test]
pub fn test_with_arguments() {
let src = ":tmi.twitch.tv PING test :abc def";
let irc_message = IRCMessage::parse(src).unwrap();
let msg = PingMessage::try_from(irc_message.clone()).unwrap();
assert_eq!(
msg,
PingMessage {
source: irc_message
}
)
}
}