v_windows/v/vlib/io/multi_writer_test.v


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

module io

fn test_multi_writer_write_successful() {
	w0 := TestWriter{}
	w1 := TestWriter{}
	mut mw := new_multi_writer(w0, w1)
	n := mw.write('0123456789'.bytes()) or {
		assert false
		return
	}
	assert n == 10
	assert w0.bytes == '0123456789'.bytes()
	assert w1.bytes == '0123456789'.bytes()
}

fn test_multi_writer_write_incomplete() {
	w0 := TestWriter{}
	w1 := TestIncompleteWriter{}
	mut mw := new_multi_writer(w0, w1)
	n := mw.write('0123456789'.bytes()) or {
		assert w0.bytes == '0123456789'.bytes()
		assert w1.bytes == '012345678'.bytes()
		return
	}
	assert false
}

fn test_multi_writer_write_error() {
	w0 := TestWriter{}
	w1 := TestErrorWriter{}
	w2 := TestWriter{}
	mut mw := new_multi_writer(w0, w1, w2)
	n := mw.write('0123456789'.bytes()) or {
		assert w0.bytes == '0123456789'.bytes()
		assert w2.bytes == []
		return
	}
	assert false
}

struct TestWriter {
pub mut:
	bytes []byte
}

fn (mut w TestWriter) write(buf []byte) ?int {
	w.bytes << buf
	return buf.len
}

struct TestIncompleteWriter {
pub mut:
	bytes []byte
}

fn (mut w TestIncompleteWriter) write(buf []byte) ?int {
	b := buf[..buf.len - 1]
	w.bytes << b
	return b.len
}

struct TestErrorWriter {}

fn (mut w TestErrorWriter) write(buf []byte) ?int {
	return error('error writer errored')
}