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
|
use std::io::{Cursor, Error, ErrorKind};
use zagan_bet::protocol::{io_failure_message, LineBuffer};
#[test]
fn keeps_data_after_the_first_newline() {
let mut buffer = LineBuffer::new();
buffer.push(b"PONG\nGAME_RESULT session 60 10 990 OSS two\n");
assert_eq!(buffer.next_line().as_deref(), Some("PONG"));
assert_eq!(
buffer.next_line().as_deref(),
Some("GAME_RESULT session 60 10 990 OSS two")
);
assert_eq!(buffer.next_line(), None);
}
#[test]
fn handles_crlf_and_empty_lines() {
let mut buffer = LineBuffer::new();
buffer.push(b"PONG\r\n\r\n");
assert_eq!(buffer.next_line().as_deref(), Some("PONG"));
assert_eq!(buffer.next_line().as_deref(), Some(""));
assert_eq!(buffer.next_line(), None);
}
#[test]
fn waits_for_a_split_line() {
let mut buffer = LineBuffer::new();
buffer.push(b"PO");
assert_eq!(buffer.next_line(), None);
buffer.push(b"NG\n");
assert_eq!(buffer.next_line().as_deref(), Some("PONG"));
}
#[test]
fn reads_multiple_lines_from_a_reader() {
let mut buffer = LineBuffer::new();
let mut reader = Cursor::new(b"one\ntwo\n".to_vec());
assert_eq!(buffer.read_line(&mut reader).unwrap().as_deref(), Some("one"));
assert_eq!(buffer.read_line(&mut reader).unwrap().as_deref(), Some("two"));
assert_eq!(buffer.read_line(&mut reader).unwrap(), None);
}
#[test]
fn returns_a_partial_line_at_eof() {
let mut buffer = LineBuffer::new();
let mut reader = Cursor::new(b"partial".to_vec());
assert_eq!(
buffer.read_line(&mut reader).unwrap().as_deref(),
Some("partial")
);
assert_eq!(buffer.read_line(&mut reader).unwrap(), None);
}
#[test]
fn timeout_errors_get_a_timeout_message() {
assert_eq!(
io_failure_message(&Error::from(ErrorKind::TimedOut)),
"ERROR Request timed out"
);
assert_eq!(
io_failure_message(&Error::from(ErrorKind::WouldBlock)),
"ERROR Request timed out"
);
assert_eq!(
io_failure_message(&Error::from(ErrorKind::UnexpectedEof)),
"ERROR Connection lost"
);
assert_eq!(
io_failure_message(&Error::from(ErrorKind::ConnectionReset)),
"ERROR Connection lost"
);
}
|
