| # Copyright 2019 The LUCI Authors. All rights reserved. |
| # Use of this source code is governed under the Apache License, Version 2.0 |
| # that can be found in the LICENSE file. |
| |
| from recipe_engine import post_process |
| |
| PYTHON_VERSION_COMPATIBILITY = "PY2+3" |
| |
| DEPS = [ |
| 'assertions', |
| 'properties', |
| 'step', |
| ] |
| |
| def RunSteps(api): |
| if 'maxDiff' in api.properties: |
| api.assertions.maxDiff = api.properties['maxDiff'] |
| try: |
| api.assertions.assertEqual(list(range(100)), list(range(99)) + [0]) |
| except AssertionError as e: |
| api.step('AssertionError', []) |
| assert 'Set self.maxDiff to None to see it.' not in str(e), ( |
| 'Did not expect self.maxDiff to appear in exception message:\n' + |
| str(e)) |
| modified_message = 'Set assertions.maxDiff to None to see it.' |
| if api.properties.get('diff_omitted', False): |
| assert modified_message in str(e), ( |
| 'Expected diff to be omitted. Exception message:\n' + str(e)) |
| else: |
| assert modified_message not in str(e), ( |
| 'Expected diff not to be omitted. Exception message:\n' + str(e)) |
| |
| |
| def GenTests(api): |
| yield api.test( |
| 'basic', |
| api.properties(maxDiff=None), |
| api.post_process(post_process.MustRun, 'AssertionError'), |
| api.post_process(post_process.StatusSuccess), |
| api.post_process(post_process.DropExpectation), |
| ) |
| |
| yield api.test( |
| 'diff-omitted', |
| api.properties(diff_omitted=True), |
| api.post_process(post_process.MustRun, 'AssertionError'), |
| api.post_process(post_process.StatusSuccess), |
| api.post_process(post_process.DropExpectation), |
| ) |
