def test_discovery_module(self):
"""HealthCheckProgram discovers tests in package."""
stream = StringIO()
cli.main(
args=['hospital.healthchecks'],
stream=stream)
output = stream.getvalue()
self.assertTrue(output.startswith(
'..\n'
'-------------------------------------'
'---------------------------------\n'
'Ran 2 tests in '))
self.assertTrue(output.endswith('s\n\nOK\n'))
def test_output(self):
"""HealthCheckProgram runs healthchecks."""
stream = StringIO()
cli.main(
args=['hospital.healthchecks.predictable'],
stream=stream)
output = stream.getvalue()
self.assertTrue(output.startswith(
'.\n'
'-------------------------------------'
'---------------------------------\n'
'Ran 1 test in '))
self.assertTrue(output.endswith('s\n\nOK\n'))
def test_failure(self, exit_mock, healthcheck_mock):
"""HealthCheckProgram exits with code 1 on failure."""
healthcheck_mock.side_effect = AssertionError
stream = StringIO()
cli.main(
args=['hospital.healthchecks.predictable'],
stream=stream)
output = stream.getvalue()
self.assertTrue(output.startswith('Fs\n'))
self.assertIn('Ran 2 tests in ', output)
self.assertTrue(output.endswith(
's\n\nFAILED (failures=1, skipped=1)\n'))
exit_mock.assert_called_once_with(1)
def test_discovery_module(self, exit_mock):
"""HealthCheckProgram discovers tests in package."""
stream = StringIO()
cli.main(
args=['hospital.healthchecks'],
stream=stream)
output = stream.getvalue()
self.assertTrue(output.startswith(
'..s\n'
'-------------------------------------'
'---------------------------------\n'
'Ran 3 tests in '))
self.assertTrue(output.endswith('s\n\nOK (skipped=1)\n'))
exit_mock.assert_called_once_with(0)
def test_output(self, exit_mock):
"""HealthCheckProgram runs healthchecks."""
stream = StringIO()
cli.main(
args=['hospital.healthchecks.predictable'],
stream=stream)
output = stream.getvalue()
self.assertTrue(output.startswith(
'.s\n'
'-------------------------------------'
'---------------------------------\n'
'Ran 2 tests in '))
self.assertTrue(output.endswith('s\n\nOK (skipped=1)\n'))
exit_mock.assert_called_once_with(0)
