def test_updateRegistration_simple(self):
listener = pb.Listener(self.master)
reg = yield listener.updateRegistration('example', 'pass', 'tcp:1234')
self.assertEqual(self.master.pbmanager._registrations,
[('tcp:1234', 'example', 'pass')])
self.assertEqual(listener._registrations['example'],
('pass', 'tcp:1234', reg))
def test_updateRegistration_port_changed(self):
listener = pb.Listener(self.master)
listener.updateRegistration('example', 'pass', 'tcp:1234')
reg1 = yield listener.updateRegistration('example', 'pass', 'tcp:4321')
self.assertEqual(listener._registrations['example'],
('pass', 'tcp:4321', reg1))
self.assertEqual(self.master.pbmanager._unregistrations,
[('tcp:1234', 'example')])
def __init__(self, master):
super().__init__()
self.pb = bbpb.Listener(master)
# WorkerRegistration instances keyed by worker name
self.registrations = {}
# connection objects keyed by worker name
self.connections = {}
def __init__(self, master):
service.AsyncMultiService.__init__(self)
self.pb = bbpb.Listener()
self.pb.setServiceParent(master)
# WorkerRegistration instances keyed by worker name
self.registrations = {}
# connection objects keyed by worker name
self.connections = {}
def test_create_connection(self):
listener = pb.Listener(self.master)
worker = mock.Mock()
worker.workername = 'test'
mind = mock.Mock()
listener.updateRegistration('example', 'pass', 'tcp:1234')
self.master.workers.register(worker)
conn = yield listener._create_connection(mind, worker.workername)
mind.broker.transport.setTcpKeepAlive.assert_called_with(1)
self.assertIsInstance(conn, pb.Connection)
def makeListener(self):
listener = pb.Listener()
listener.setServiceParent(self.master)
return listener
def test_constructor(self):
listener = pb.Listener(self.master)
self.assertEqual(listener.master, self.master)
self.assertEqual(listener._registrations, {})
def makeListener(self):
listener = pb.Listener(self.master)
return listener
