def test_wait_multi_switch(self):
"""
If you switch between lots of things, wait should wait for all previous
connections -- not just the most recent.
"""
p = Pipe('foo')
proto1 = object()
p.addConnection('foo', proto1)
p.switch('bar')
proto2 = object()
p.addConnection('bar', proto2)
p.switch('coo')
proto3 = object()
p.addConnection('coo', proto3)
w = p.wait()
self.assertFalse(w.called)
p.removeConnection('bar', proto2)
self.assertFalse(w.called)
p.removeConnection('foo', proto1)
self.assertTrue(w.called)
def test_switch_back(self):
"""
If you switch forwarding to one dst, then switch back, the original
destination's Deferred should not fire, but the intermediate destination
should.
"""
# X -> foo
p = Pipe('foo')
foo_d = p.alive['foo']
proto1 = object()
p.addConnection('foo', proto1)
# X -> bar
p.switch('bar')
bar_d = p.alive['bar']
proto2 = object()
p.addConnection('bar', proto2)
# X -> foo
p.switch('foo')
p.removeConnection('foo', proto1)
self.assertFalse(foo_d.called, "Should not have called"
" since that is where the current forwarding is")
self.assertEqual(p._connections['foo'], 0)
self.assertFalse(bar_d.called)
p.removeConnection('bar', proto2)
self.assertTrue(bar_d.called)
def test_noswitch_nocallback(self):
"""
If the Pipe is still forwarding to an endpoint, the
alive Deferred should not fire even when the last
connection is done.
"""
p = Pipe('foo')
proto1 = object()
p.addConnection('foo', proto1)
p.removeConnection('foo', proto1)
d = p.alive['foo']
self.assertEqual(d.called, False, "Should not have called "
"the Deferred because it's still "
"forwarding to that endpoint")
def test_ls(self):
"""
You should be able to list the endpoints currently
forwarding
"""
p = Pipe('foo')
r = set(p.ls())
self.assertEqual(r, set([
# endpoint, connections, active/not
('foo', 0, True),
]))
proto1 = object()
p.addConnection('foo', proto1)
self.assertEqual(set(p.ls()), set([
('foo', 1, True),
]))
p.switch('bar')
proto2 = object()
p.addConnection('bar', proto2)
self.assertEqual(set(p.ls()), set([
('foo', 1, False),
('bar', 1, True),
]))
p.removeConnection('bar', proto2)
self.assertEqual(set(p.ls()), set([
('foo', 1, False),
('bar', 0, True),
]))
p.removeConnection('foo', proto1)
self.assertEqual(set(p.ls()), set([
('bar', 0, True),
]))
self.assertEqual(p.alive.keys(), ['bar'], "Only active or"
" still working endpoints should be "
"listed")
p.switch('foo')
self.assertEqual(set(p.ls()), set([
('foo', 0, True),
]))
self.assertEqual(p.alive.keys(), ['foo'], "Only active or"
" still working endpoints should be listed")
def test_wait_wait(self):
"""
Calling wait multiple times should behave correctly
"""
pipe = Pipe('foo')
proto = object()
pipe.addConnection('foo', proto)
pipe.switch('bar')
wait = pipe.wait()
pipe.removeConnection('foo', proto)
self.assertTrue(wait.called)
pipe.switch('foo')
proto = object()
pipe.addConnection('foo', proto)
pipe.switch('bar')
wait = pipe.wait()
pipe.removeConnection('foo', proto)
self.assertTrue(wait.called)
def test_wait_conns(self):
"""
If there are pending connections, wait is only called once they have
finished.
"""
p = Pipe('foo')
proto1 = object()
p.addConnection('foo', proto1)
proto2 = object()
p.addConnection('foo', proto2)
p.switch('bar')
w = p.wait()
self.assertFalse(w.called, "Should not have called yet, because there "
"is an active connection to an old destination")
p.removeConnection('foo', proto1)
self.assertFalse(w.called)
p.removeConnection('foo', proto2)
self.assertTrue(w.called, "Should have called; the last connection "
"finished")
def test_wait_switch_back_and_forth(self):
"""
Switching back and forth should work
"""
pipe = Pipe('foo')
p1 = object()
pipe.addConnection('foo', p1)
foo_d = pipe.alive['foo']
pipe.switch('bar')
wait = pipe.wait()
p2 = object()
pipe.addConnection('bar', p2)
pipe.switch('foo')
p3 = object()
pipe.addConnection('foo', p3)
pipe.switch('bar')
p4 = object()
pipe.addConnection('bar', p4)
pipe.switch('foo')
p5 = object()
pipe.addConnection('foo', p5)
pipe.switch('bar')
p6 = object()
pipe.addConnection('bar', p6)
self.assertFalse(wait.called, "Should not have called wait yet")
self.assertFalse(foo_d.called)
pipe.removeConnection('bar', p6)
self.assertFalse(wait.called)
self.assertFalse(foo_d.called)
pipe.removeConnection('foo', p5)
self.assertFalse(wait.called)
self.assertFalse(foo_d.called)
pipe.removeConnection('bar', p4)
self.assertFalse(wait.called)
self.assertFalse(foo_d.called)
pipe.removeConnection('foo', p3)
self.assertFalse(wait.called)
self.assertFalse(foo_d.called)
pipe.removeConnection('bar', p2)
self.assertFalse(wait.called)
self.assertFalse(foo_d.called)
pipe.removeConnection('foo', p1)
self.assertTrue(wait.called)
self.assertTrue(foo_d.called)
