def test_whitelist_disable(self):
# Keep a reference for use during mock
get_config = stubs.datadog_agent.get_config
def mock_get_config(key):
if key == 'tracemalloc_whitelist':
return 'test1,test2'
return get_config(key)
with mock.patch('datadog_checks.base.stubs.datadog_agent.get_config',
side_effect=mock_get_config):
assert should_profile_memory(stubs.datadog_agent, 'test') is False
def test_default(self):
with mock.patch('datadog_checks.base.stubs.datadog_agent.get_config',
return_value=''):
assert should_profile_memory(stubs.datadog_agent, 'test') is True
