def test_version(self): args = ['--version'] with self.capture_stdout() as output: main(args) self.assertEqual(output.read().strip(), __version__)
def test_config(self): args = [ '--config', config, 'loads.examples.test_blog.TestWebSite.test_something', '--quiet' ] main(args)
def test_help(self): args = [] with self.capture_stdout() as output: main(args) self.assertTrue(output.read().strip().startswith('usage'))
def test_check_cluster(self): args = ['--check-cluster'] with self.capture_stdout() as output: main(args) self.assertTrue('Success: 3' in output.read().strip())
def test_ping_broker(self): args = ['--ping-broker'] with self.capture_stdout() as output: main(args) output = output.read().strip() self.assertTrue(re.search(_WANTED, output) is not None, output)
def test_purge_broker(self): args = ['--purge-broker'] with self.capture_stdout() as output: main(args) wanted = ['Nothing to purge.', 'We have 1 run(s) right now:\nPurged.'] self.assertTrue(output.read().strip() in wanted)
def test_ping_broker(self): args = ['--ping-broker'] with self.capture_stdout() as output: main(args) res = [line.strip() for line in output.read().strip().split('\n') if line.strip() != ''] wanted = ['Broker running on pid ', '3 agents registered', 'endpoints:', '- backend: ipc:///tmp/loads-back.ipc', '- publisher: ipc:///tmp/loads-publisher.ipc', '- register: ipc:///tmp/loads-reg.ipc', '- frontend: ipc:///tmp/loads-front.ipc', '- receiver: ipc:///tmp/loads-broker-receiver.ipc'] for index, line in enumerate(wanted): self.assertTrue(res[index].startswith(line))
def test_ping_broker(self): args = ['--ping-broker'] with self.capture_stdout() as output: main(args) res = [ line.strip() for line in output.read().strip().split('\n') if line.strip() != '' ] wanted = [ 'Broker running on pid ', '3 agents registered', 'endpoints:', '- backend: ipc:///tmp/loads-back.ipc', '- publisher: ipc:///tmp/loads-publisher.ipc', '- register: ipc:///tmp/loads-reg.ipc', '- frontend: ipc:///tmp/loads-front.ipc', '- receiver: ipc:///tmp/loads-broker-receiver.ipc' ] for index, line in enumerate(wanted): self.assertTrue(res[index].startswith(line))
def test_config(self): args = ['--config', config, 'loads.examples.test_blog.TestWebSite.test_something', '--quiet'] main(args)