def setUp(self):
def _add_alias(self, command_log_dir, logger):
self.cli_ctx.logging.command_log_dir = command_log_dir
self.cmd("az extension add -n alias")
logger.warning("Adding whl ext alias")
super(TestCommandLogFile, self).setUp()
self.temp_command_log_dir = self.create_temp_dir()
# if alias is installed as a wheel extension. Remove it for now and re-install it later.
if "alias" in [ext.name for ext in get_extensions() if isinstance(ext, WheelExtension)]:
logger.warning("Removing whl ext alias, will reinstall it afterwards if it is a published whl extension.")
self.cmd("az extension remove -n alias")
self.addCleanup(_add_alias, self, self.cli_ctx.logging.command_log_dir, logger)
self.cli_ctx.logging.command_log_dir = self.temp_command_log_dir
# populate command dir with logs
self._run_cmd("extension remove -n alias", expect_failure=True)
time.sleep(2)
self._run_cmd("extension add -n alias")
time.sleep(2)
self._run_cmd("alias create -n foo123 -c bar123", expect_failure=True)
time.sleep(2)
self._run_cmd("alias list")
def tearDownClass(cls):
super(TestCommandLogFile, cls).tearDownClass()
for ext_name in [
ext.name for ext in get_extensions()
if isinstance(ext, WheelExtension)
and ext.name not in cls.original_extensions
]:
remove_extension(ext_name)
logger.warning(
"Removing ext %s, as it was not in the original list of installed wheel extensions..",
ext_name)
for ext_name in cls.original_extensions:
add_extension(extension_name=ext_name)
logger.warning("Adding whl ext %s", ext_name)
def setUpClass(cls):
super(TestCommandLogFile, cls).setUpClass()
cls.ext_to_add_post_test = None
# ensure that after tests are run we return to original extensions' state.
cls.original_extensions = [
ext.name for ext in get_extensions()
if isinstance(ext, WheelExtension)
]
for ext_name in cls.original_extensions:
remove_extension(ext_name)
logger.warning(
"Removing ext %s, will reinstall it afterwards if it is a published whl extension.",
ext_name)
def _ext_installed(ext):
return ext in [ext.name for ext in get_extensions()]