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
|
"""Channel testing"""
from test.testlib import *
from git.async.channel import *
import time
class TestChannels(TestBase):
def test_base(self):
# creating channel yields a write and a read channal
wc, rc = Channel()
assert isinstance(wc, WChannel)
assert isinstance(rc, RChannel)
# everything else fails
self.failUnlessRaises(ValueError, Channel, 1, "too many args")
# TEST UNLIMITED SIZE CHANNEL - writing+reading is FIFO
item = 1
item2 = 2
wc.write(item)
wc.write(item2)
# read all - it blocks as its still open for writing
to = 0.2
st = time.time()
assert rc.read(timeout=to) == [item, item2]
assert time.time() - st >= to
# next read blocks. it waits a second
st = time.time()
assert len(rc.read(1, True, to)) == 0
assert time.time() - st >= to
# writing to a closed channel raises
assert not wc.closed
wc.close()
assert wc.closed
wc.close() # fine
assert wc.closed
self.failUnlessRaises(IOError, wc.write, 1)
# reading from a closed channel never blocks
assert len(rc.read()) == 0
# TEST LIMITED SIZE CHANNEL
# channel with max-items set
wc, rc = Channel(1)
wc.write(item) # fine
# blocks for a a moment, its full
st = time.time()
self.failUnlessRaises(EOFError, wc.write, item, True, to)
assert time.time() - st >= to
# get our only one
assert rc.read(1)[0] == item
# its empty,can put one again
wc.write(item2)
wc.close()
# reading 10 will only yield one, it will not block as its closed
assert rc.read(10, timeout=1)[0] == item2
|