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):
service.AsyncMultiService.__init__(self)
self.pb = bbpb.Listener()
self.pb.setServiceParent(master)
# BuildslaveRegistration instances keyed by buildslave name
self.registrations = {}
# connection objects keyed by buildslave name
self.connections = {}
def test_getPerspective(self):
listener = pb.Listener(self.master)
buildslave = mock.Mock()
buildslave.slavename = 'test'
mind = mock.Mock()
listener.updateRegistration('example', 'pass', 'tcp:1234')
self.master.buildslaves.register(buildslave)
conn = yield listener._getPerspective(mind, buildslave.slavename)
mind.broker.transport.setTcpKeepAlive.assert_called_with(1)
self.assertIsInstance(conn, pb.Connection)
def __init__(self, master):
service.MultiService.__init__(self)
self.setName('buildslaves')
self.master = master
self.pb = bbpb.Listener(self.master)
self.pb.setServiceParent(self)
# BuildslaveRegistration instances keyed by buildslave name
self.registrations = {}
# connection objects keyed by buildslave name
self.connections = {}
def __init__(self, master):
service.AsyncMultiService.__init__(self)
self.setName('buildslaves')
self.master = master
self.pb = bbpb.Listener(self.master)
self.pb.setServiceParent(self)
# BuildslaveRegistration instances keyed by buildslave name
self.registrations = {}
# connection objects keyed by buildslave name
self.connections = {}
# self.slaves contains a ready BuildSlave instance for each
# potential buildslave, i.e. all the ones listed in the config file.
# If the slave is connected, self.slaves[slavename].slave will
# contain a RemoteReference to their Bot instance. If it is not
# connected, that attribute will hold None.
self.slaves = {} # maps slavename to BuildSlave
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, {})
