def test_disconnect(self): """ should go through to hub.disconnect """ shell = ShellProtocol() stop_d = defer.Deferred() shell.hub = FakeHub(disconnect=stop_d) sendLine = [] shell.sendLine = sendLine.append shell.cmd_disconnect('tcp:8080') self.assertIn(('disconnect', 'tcp:8080'), shell.hub.called)
def test_build(self): s = ShellProtocol() build_response = {'uid': 'something'} s.hub = FakeHub(build=build_response) sendLine_called = [] s.sendLine = sendLine_called.append s.cmd_build('project', 'version') self.assertNotEqual(sendLine_called, [], "Should have sent something back") self.assertTrue('something' in sendLine_called[0], "Should include the build request uid in the response") self.assertEqual(s.hub.called, [ ('build', dict(project='project', version='version', test_path=None)), ])
def test_connect(self): """ should go through to hub.connect """ shell = ShellProtocol() client = defer.Deferred() shell.hub = FakeHub(connect=client) sendLine = [] shell.sendLine = sendLine.append shell.cmd_connect('my endpoint') while sendLine: sendLine.pop() self.assertIn(('connect', 'my endpoint'), shell.hub.called) client.callback('foo') self.assertNotEqual(sendLine, [], "Once client connects, connectee " "should be notified.")
def test_stop(self): """ stop should go through to stopServer """ shell = ShellProtocol() stop_d = defer.Deferred() shell.hub = FakeHub(stopServer=stop_d) sendLine = [] shell.sendLine = sendLine.append shell.cmd_stop('tcp:8080') while sendLine: sendLine.pop() self.assertIn(('stopServer', 'tcp:8080'), shell.hub.called) stop_d.callback('foo') self.assertNotEqual(sendLine, [], "Once server stops, connectee " "should be notified.")
def test_start(self): """ start should go through to startServer """ shell = ShellProtocol() factory = object() server = defer.Deferred() shell.hub = FakeHub(getPBServerFactory=factory, startServer=server) sendLine = [] shell.sendLine = sendLine.append shell.cmd_start('tcp:8080') while sendLine: sendLine.pop() self.assertIn(('startServer', factory, 'tcp:8080'), shell.hub.called) server.callback('foo') self.assertNotEqual(sendLine, [], "Once server starts, connectee " "should be notified.")