def test_getPipe(self):
"""
You can get the Pipe itself using the src endpoint as a key.
"""
listen = 'unix:'+self.mktemp()
connect = 'unix:path='+self.mktemp()
p = Plumber()
p.addPipe(listen, connect)
pipe = p.getPipe(listen)
self.assertTrue(isinstance(pipe, Pipe))
self.assertEqual(pipe, list(p.pipe_services)[0].factory)
def test_rmPipe(self):
"""
You can remove pipes by endpoint name
"""
listen = 'unix:'+self.mktemp()
connect = 'unix:path='+self.mktemp()
p = Plumber()
s = p.addPipe(listen, connect)
d = p.rmPipe(listen)
def check(result):
self.assertEqual(list(p.pipe_services), [])
self.assertEqual(s.parent, None)
return d.addCallback(check)
def test_addPipe(self):
"""
Adding a pipe should create a new service using the pipeFactory and
add the service to the list of services.
"""
listen = 'unix:'+self.mktemp()
connect = 'unix:path='+self.mktemp()
p = Plumber()
s = p.addPipe(listen, connect)
services = list(p.pipe_services)
self.assertEqual(len(services), 1, "Should have added a service")
self.assertEqual(s, services[0], "Should have returned the service")
self.assertTrue(isinstance(s.factory, Pipe))
self.assertEqual(s.factory.dst, connect, "Should have set the "
"destination on the Pipe")
def test_stop(self):
"""
You can stop the whole process.
"""
class FakeReactor:
called = []
def stop(self):
self.called.append('stop')
fake_reactor = FakeReactor()
p = Plumber(_reactor=fake_reactor)
self.assertEqual(p._reactor, fake_reactor)
p.stop()
self.assertEqual(fake_reactor.called, ['stop'])
def test_pipeCommand(self):
"""
You can execute things on the Pipe with a key
"""
listen = 'unix:'+self.mktemp()
connect = 'unix:path='+self.mktemp()
p = Plumber()
p.addPipe(listen, connect)
pipe = p.getPipe(listen)
called = []
def fake(*a, **kw):
called.append((a, kw))
return 'result'
pipe.foo = fake
r = p.pipeCommand(listen, 'foo', 'arg1', 'arg2', kw1='foo', kw2='bar')
self.assertEqual(r, 'result', "Should return whatever the Pipe's "
"method returned")
self.assertEqual(called, [
(('arg1', 'arg2'), {'kw1':'foo', 'kw2':'bar'}),
], "Should have passed all the appropriate args through")
def test_ls(self):
"""
You can list the details of all pipes
"""
p = Plumber()
p.addPipe('unix:foo', 'unix:foo2')
p.addPipe('unix:bar', 'unix:bar2')
r = list(p.ls())
expected = [tuple(['unix:bar']+list(x)) for x in p.pipeCommand('unix:bar', 'ls')]
expected += [tuple(['unix:foo']+list(x)) for x in p.pipeCommand('unix:foo', 'ls')]
self.assertEqual(r, expected)
def __init__(self, results=None):
Plumber.__init__(self)
self.called = []
self._results = results or {}