def wrapper(task):
"""Wrapper."""
environment.set_value('TASK_PAYLOAD', task.payload())
try:
return func(task)
except: # Truly catch *all* exceptions.
e = sys.exc_info()[1]
e.extras = {'task_payload': environment.get_value('TASK_PAYLOAD')}
raise
finally:
environment.remove_key('TASK_PAYLOAD')
def _setup_memory_tools_environment(testcase):
"""Set up environment for various memory tools used."""
env = testcase.get_metadata('env')
if not env:
environment.reset_current_memory_tool_options(
redzone_size=testcase.redzone,
disable_ubsan=testcase.disable_ubsan)
return
for options_name, options_value in six.iteritems(env):
if not options_value:
environment.remove_key(options_name)
continue
environment.set_memory_tool_options(options_name, options_value)
def setUp(self):
test_helpers.patch_environ(self)
environment.remove_key('PROJECT_NAME')
environment.remove_key('ISSUE_TRACKER')
environment.remove_key('UPDATE_WEB_TESTS')
self.config = local_config.ProjectConfig()
def test_custom_binary(self):
"""Test custom binary."""
environment.set_value('CUSTOM_BINARY', True)
testcase = test_utils.create_generic_testcase()
expected_result = {
'result': {
'error_message': 'Not applicable to custom binaries.',
'feedback_url': '',
'found': False,
'project': '',
'suspected_components': '',
'suspected_cls': '',
}
}
self.assertIsNone(blame_task._prepare_predator_message(testcase))
testcase = data_handler.get_testcase_by_id(testcase.key.id())
actual_result = testcase.get_metadata('predator_result')
self.assertEqual(actual_result, expected_result)
environment.remove_key('CUSTOM_BINARY')
def test_is_directory_on_nfs_without_nfs(self):
"""Test is_directory_on_nfs without nfs."""
environment.remove_key('NFS_ROOT')
self.assertFalse(setup.is_directory_on_nfs('/nfs/dir1'))
def tearDown(self):
environment.remove_key('NFS_ROOT')
def _clear_env():
"""Clear build env vars."""
environment.remove_key('APP_PATH')
environment.remove_key('APP_REVISION')
environment.remove_key('APP_PATH_DEBUG')
environment.remove_key('APP_DIR')
environment.remove_key('BUILD_DIR')
environment.remove_key('BUILD_URL')
environment.remove_key('FUZZ_TARGET')