def test_windows(self):
with mock.patch.object(os, "name", "nt"):
with mock.patch.dict(
os.environ,
{
"LOCALAPPDATA": "C:\\Users\\Alice\\AppData\\Local",
"APPDATA": "C:\\Users\\Alice\\AppData\\Roaming",
},
):
self.assertEqual(
"C:\\Users\\Alice\\AppData\\Local",
util.get_user_config_directory(),
)
with mock.patch.dict(
os.environ,
{
"LOCALAPPDATA": "",
"APPDATA": "C:\\Users\\Alice\\AppData\\Roaming",
},
):
self.assertEqual(
"C:\\Users\\Alice\\AppData\\Roaming",
util.get_user_config_directory(),
)
with mock.patch.dict(
os.environ,
{
"LOCALAPPDATA": "",
"APPDATA": "",
},
):
self.assertIsNone(util.get_user_config_directory())
def test_non_windows(self):
with mock.patch.dict(os.environ, {"HOME": "/home/alice"}):
self.assertEqual(
"/home/alice%s.config" % os.sep, util.get_user_config_directory())
with mock.patch.dict(
os.environ, {"XDG_CONFIG_HOME": "/home/alice/configz"}):
self.assertEqual(
"/home/alice/configz", util.get_user_config_directory())
def __init__(self, user_config_directory=_DEFAULT_CONFIG_DIRECTORY):
"""Creates a CredentialsStore.
Args:
user_config_directory: Optional absolute path to the root directory for
storing user configs, under which to store the credentials file. If not
set, defaults to a platform-specific location. If set to None, the
store is disabled (reads return None; write and clear are no-ops).
"""
if user_config_directory is CredentialsStore._DEFAULT_CONFIG_DIRECTORY:
user_config_directory = util.get_user_config_directory()
if user_config_directory is None:
logger.warning(
"Credentials caching disabled - no private config directory found"
)
if user_config_directory is None:
self._credentials_filepath = None
else:
self._credentials_filepath = os.path.join(
user_config_directory, *TENSORBOARD_CREDENTIALS_FILEPATH_PARTS)
