def test_api_keys_regex(self, mock_config, mock_tempdir, mock_strftime):
f = Flare()
file_path, _ = f._strip_credentials(
os.path.join(get_mocked_temp(), 'apikeys.conf'),
f.MAIN_CREDENTIALS
)
with open(file_path) as f:
contents = f.read()
self.assertEqual(
contents,
"api_key: **************************aaaaa, **************************bbbbb,"
" **************************ccccc, **************************ddddd\n"
)
f = Flare()
file_path, _ = f._strip_credentials(
os.path.join(get_mocked_temp(), 'apikey.conf'),
f.MAIN_CREDENTIALS
)
with open(file_path) as f:
contents = f.read()
self.assertEqual(
contents,
"api_key: **************************aaaaa\n"
)
def test_api_keys_regex(self, mock_config, mock_tempdir, mock_strftime, mock_os_remove):
f = Flare()
file_path, _ = f._strip_credentials(
os.path.join(get_mocked_temp(), 'apikeys.conf'),
f.MAIN_CREDENTIALS
)
with open(file_path) as f:
contents = f.read()
self.assertEqual(
contents,
"api_key: **************************aaaaa, **************************bbbbb,"
" **************************ccccc, **************************ddddd\n"
)
f = Flare()
file_path, _ = f._strip_credentials(
os.path.join(get_mocked_temp(), 'apikey.conf'),
f.MAIN_CREDENTIALS
)
with open(file_path) as f:
contents = f.read()
self.assertEqual(
contents,
"api_key: **************************aaaaa\n"
)
def test_uri_password(self, mock_config, mock_tempdir, mock_strftime):
f = Flare()
_, credentials_log = f._strip_credentials(
os.path.join(get_mocked_temp(), mock_cfgs['uri_password']),
f.CHECK_CREDENTIALS)
self.assertEqual(
credentials_log,
" - this file contains a credential (password in a uri) which has been removed in the collected version"
)
def test_uri_password(self, mock_config, mock_tempdir, mock_strftime):
f = Flare()
_, credentials_log = f._strip_credentials(
os.path.join(get_mocked_temp(), mock_cfgs['uri_password']),
f.CHECK_CREDENTIALS
)
self.assertEqual(
credentials_log,
" - this file contains a credential (password in a uri) which has been removed in the collected version"
)
def test_proxy_user_pass_regex(self, mock_config, mock_tempdir,
mock_strftime, mock_os_remove):
f = Flare()
file_path, _ = f._strip_credentials(
os.path.join(get_mocked_temp(), 'proxy.conf'), f.MAIN_CREDENTIALS)
with open(file_path) as f:
contents = f.read()
self.assertEqual(contents, "proxy_user: ********\n"
"proxy_password: ********\n")
def test_password_regex(self, mock_config, mock_tempdir, mock_strftime,
mock_os_remove):
f = Flare()
file_path, credentials_log = f._strip_credentials(
os.path.join(get_mocked_temp(), 'password.yaml'),
f.CHECK_CREDENTIALS)
with open(file_path) as f:
contents = f.read()
self.assertEqual(
contents, "instances:\n"
" - pass: ********\n"
" other_password: ********\n")
def test_whitespace_proxy_user_pass_regex(self, mock_config, mock_tempdir, mock_strftime, mock_os_remove):
f = Flare()
file_path, _ = f._strip_credentials(
os.path.join(get_mocked_temp(), 'whitespace_proxy.conf'),
f.MAIN_CREDENTIALS
)
with open(file_path) as f:
contents = f.read()
self.assertEqual(
contents,
"proxy_user: ********\n"
"proxy_password: ********\n"
)
