← Back to branch summary

~justin-fathomdb/nova/justinsb-openstack-api-volumes

« back to all changes in this revision

Viewing changes to vendor/Twisted-10.0.0/twisted/internet/test/test_fdset.py

  • Committer: Jesse Andrews
  • Date: 2010-05-28 06:05:26 UTC
  • Revision ID: git-v1:bf6e6e718cdc7488e2da87b21e258ccc065fe499
initial commit

Show diffs side-by-side

added added

removed removed

Lines of Context:
vendor/Twisted-10.0.0/twisted/internet/test/test_fdset.py
 
1
# Copyright (c) 2009 Twisted Matrix Laboratories.
 
2
# See LICENSE for details.
 
3
 
 
4
"""
 
5
Tests for implementations of L{IReactorFDSet}.
 
6
"""
 
7
 
 
8
__metaclass__ = type
 
9
 
 
10
import socket
 
11
 
 
12
from twisted.internet.interfaces import IReactorFDSet
 
13
from twisted.internet.abstract import FileDescriptor
 
14
from twisted.internet.test.reactormixins import ReactorBuilder
 
15
 
 
16
# twisted.internet.tcp nicely defines some names with proper values on
 
17
# several different platforms.
 
18
from twisted.internet.tcp import EINPROGRESS, EWOULDBLOCK
 
19
 
 
20
 
 
21
class ReactorFDSetTestsBuilder(ReactorBuilder):
 
22
    """
 
23
    Builder defining tests relating to L{IReactorFDSet}.
 
24
    """
 
25
    requiredInterface = IReactorFDSet
 
26
 
 
27
    def _connectedPair(self):
 
28
        """
 
29
        Return the two sockets which make up a new TCP connection.
 
30
        """
 
31
        serverSocket = socket.socket()
 
32
        serverSocket.bind(('127.0.0.1', 0))
 
33
        serverSocket.listen(1)
 
34
        self.addCleanup(serverSocket.close)
 
35
 
 
36
        client = socket.socket()
 
37
        self.addCleanup(client.close)
 
38
        client.setblocking(False)
 
39
        try:
 
40
            client.connect(('127.0.0.1', serverSocket.getsockname()[1]))
 
41
        except socket.error, e:
 
42
            self.assertIn(e.args[0], (EINPROGRESS, EWOULDBLOCK))
 
43
        else:
 
44
            self.fail("Connect should have raised EINPROGRESS or EWOULDBLOCK")
 
45
        server, addr = serverSocket.accept()
 
46
        self.addCleanup(server.close)
 
47
 
 
48
        return client, server
 
49
 
 
50
 
 
51
    def _simpleSetup(self):
 
52
        reactor = self.buildReactor()
 
53
 
 
54
        client, server = self._connectedPair()
 
55
 
 
56
        fd = FileDescriptor(reactor)
 
57
        fd.fileno = client.fileno
 
58
 
 
59
        return reactor, fd, server
 
60
 
 
61
 
 
62
    def test_addReader(self):
 
63
        """
 
64
        C{reactor.addReader()} accepts an L{IReadDescriptor} provider and calls
 
65
        its C{doRead} method when there may be data available on its C{fileno}.
 
66
        """
 
67
        reactor, fd, server = self._simpleSetup()
 
68
 
 
69
        fd.doRead = reactor.stop
 
70
        reactor.addReader(fd)
 
71
        server.sendall('x')
 
72
 
 
73
        # The reactor will only stop if it calls fd.doRead.
 
74
        self.runReactor(reactor)
 
75
        # Nothing to assert, just be glad we got this far.
 
76
 
 
77
 
 
78
    def test_removeReader(self):
 
79
        """
 
80
        L{reactor.removeReader()} accepts an L{IReadDescriptor} provider
 
81
        previously passed to C{reactor.addReader()} and causes it to no longer
 
82
        be monitored for input events.
 
83
        """
 
84
        reactor, fd, server = self._simpleSetup()
 
85
 
 
86
        def fail():
 
87
            self.fail("doRead should not be called")
 
88
        fd.doRead = fail
 
89
 
 
90
        reactor.addReader(fd)
 
91
        reactor.removeReader(fd)
 
92
        server.sendall('x')
 
93
 
 
94
        # Give the reactor two timed event passes to notice that there's I/O
 
95
        # (if it is incorrectly watching for I/O).
 
96
        reactor.callLater(0, reactor.callLater, 0, reactor.stop)
 
97
 
 
98
        self.runReactor(reactor)
 
99
        # Getting here means the right thing happened probably.
 
100
 
 
101
 
 
102
    def test_addWriter(self):
 
103
        """
 
104
        C{reactor.addWriter()} accepts an L{IWriteDescriptor} provider and
 
105
        calls its C{doWrite} method when it may be possible to write to its
 
106
        C{fileno}.
 
107
        """
 
108
        reactor, fd, server = self._simpleSetup()
 
109
 
 
110
        fd.doWrite = reactor.stop
 
111
        reactor.addWriter(fd)
 
112
 
 
113
        self.runReactor(reactor)
 
114
        # Getting here is great.
 
115
 
 
116
 
 
117
    def _getFDTest(self, kind):
 
118
        """
 
119
        Helper for getReaders and getWriters tests.
 
120
        """
 
121
        reactor = self.buildReactor()
 
122
        get = getattr(reactor, 'get' + kind + 's')
 
123
        add = getattr(reactor, 'add' + kind)
 
124
        remove = getattr(reactor, 'remove' + kind)
 
125
 
 
126
        client, server = self._connectedPair()
 
127
 
 
128
        self.assertNotIn(client, get())
 
129
        self.assertNotIn(server, get())
 
130
 
 
131
        add(client)
 
132
        self.assertIn(client, get())
 
133
        self.assertNotIn(server, get())
 
134
 
 
135
        remove(client)
 
136
        self.assertNotIn(client, get())
 
137
        self.assertNotIn(server, get())
 
138
 
 
139
 
 
140
    def test_getReaders(self):
 
141
        """
 
142
        L{IReactorFDSet.getReaders} reflects the additions and removals made
 
143
        with L{IReactorFDSet.addReader} and L{IReactorFDSet.removeReader}.
 
144
        """
 
145
        self._getFDTest('Reader')
 
146
 
 
147
 
 
148
    def test_removeWriter(self):
 
149
        """
 
150
        L{reactor.removeWriter()} accepts an L{IWriteDescriptor} provider
 
151
        previously passed to C{reactor.addWriter()} and causes it to no longer
 
152
        be monitored for outputability.
 
153
        """
 
154
        reactor, fd, server = self._simpleSetup()
 
155
 
 
156
        def fail():
 
157
            self.fail("doWrite should not be called")
 
158
        fd.doWrite = fail
 
159
 
 
160
        reactor.addWriter(fd)
 
161
        reactor.removeWriter(fd)
 
162
 
 
163
        # Give the reactor two timed event passes to notice that there's I/O
 
164
        # (if it is incorrectly watching for I/O).
 
165
        reactor.callLater(0, reactor.callLater, 0, reactor.stop)
 
166
 
 
167
        self.runReactor(reactor)
 
168
        # Getting here means the right thing happened probably.
 
169
 
 
170
 
 
171
    def test_getWriters(self):
 
172
        """
 
173
        L{IReactorFDSet.getWriters} reflects the additions and removals made
 
174
        with L{IReactorFDSet.addWriter} and L{IReactorFDSet.removeWriter}.
 
175
        """
 
176
        self._getFDTest('Writer')
 
177
 
 
178
 
 
179
    def test_removeAll(self):
 
180
        """
 
181
        C{reactor.removeAll()} removes all registered L{IReadDescriptor}
 
182
        providers and all registered L{IWriteDescriptor} providers and returns
 
183
        them.
 
184
        """
 
185
        reactor = self.buildReactor()
 
186
 
 
187
        reactor, fd, server = self._simpleSetup()
 
188
 
 
189
        fd.doRead = lambda: self.fail("doRead should not be called")
 
190
        fd.doWrite = lambda: self.fail("doWrite should not be called")
 
191
 
 
192
        server.sendall('x')
 
193
 
 
194
        reactor.addReader(fd)
 
195
        reactor.addWriter(fd)
 
196
 
 
197
        removed = reactor.removeAll()
 
198
 
 
199
        # Give the reactor two timed event passes to notice that there's I/O
 
200
        # (if it is incorrectly watching for I/O).
 
201
        reactor.callLater(0, reactor.callLater, 0, reactor.stop)
 
202
 
 
203
        self.runReactor(reactor)
 
204
        # Getting here means the right thing happened probably.
 
205
 
 
206
        self.assertEqual(removed, [fd])
 
207
 
 
208
 
 
209
globals().update(ReactorFDSetTestsBuilder.makeTestCaseClasses())