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')