def tearDown(self):
try:
tasks.stop(ctx=self._create_context('stop'))
except BaseException as e:
logger.error(e.message)
try:
tasks.uninstall(ctx=self._create_context('uninstall'))
except BaseException as e:
logger.warning(e.message)
self.runner.delete(path=tasks.AGENT_FOLDER_NAME,
ignore_missing=True)
self.runner.delete(
path='C:\\{0}'.format(tasks.AGENT_EXEC_FILE_NAME),
ignore_missing=True)
def tearDown(self):
try:
tasks.stop(ctx=self._create_context('stop'))
except BaseException as e:
logger.error(e.message)
try:
tasks.uninstall(ctx=self._create_context('uninstall'))
except BaseException as e:
logger.warning(e.message)
self.runner.delete(path=tasks.AGENT_FOLDER_NAME,
ignore_missing=True)
self.runner.delete(
path='C:\\{0}'.format(tasks.AGENT_EXEC_FILE_NAME),
ignore_missing=True)
def test_full_lifecycle(self):
tasks.install(ctx=self._create_context('install'))
tasks.start(ctx=self._create_context('start'))
def _create_session():
return {
'host': os.environ[TEST_MACHINE_IP_ENV_VARIABLE],
'user': '******',
'password': '******'
}
# Retrieve registered plugins
from windows_agent_installer.winrm_runner import WinRMRunner
from windows_agent_installer.tasks import RUNTIME_AGENT_PATH
from windows_agent_installer.tasks import AGENT_EXEC_FILE_NAME
runner = WinRMRunner(session_config=_create_session())
response = runner.run(
'{0}\Scripts\celery.exe inspect registered'
.format(RUNTIME_AGENT_PATH))
# Assert agent has necessary includes
self.assertTrue(AGENT_INCLUDES in response.std_out)
# Restart the service
tasks.restart(ctx=self._create_context('restart'))
# Stop the service
tasks.stop(ctx=self._create_context('stop'))
# Uninstall, delete the files
tasks.uninstall(ctx=self._create_context('uninstall'))
# Assert files are gone
self.assertFalse(runner.exists(path=RUNTIME_AGENT_PATH))
self.assertFalse(
runner.exists(
path='C:\{0}'.format(AGENT_EXEC_FILE_NAME)))
def test_full_lifecycle(self):
tasks.install(ctx=self._create_context('install'))
tasks.start(ctx=self._create_context('start'))
def _create_session():
return {
'host': os.environ[TEST_MACHINE_IP_ENV_VARIABLE],
'user': '******',
'password': '******'
}
# Retrieve registered plugins
from windows_agent_installer.winrm_runner import WinRMRunner
from windows_agent_installer.tasks import RUNTIME_AGENT_PATH
from windows_agent_installer.tasks import AGENT_EXEC_FILE_NAME
runner = WinRMRunner(session_config=_create_session())
response = runner.run(
'{0}\Scripts\celery.exe inspect registered'
.format(RUNTIME_AGENT_PATH))
# Assert agent has necessary includes
self.assertTrue(AGENT_INCLUDES in response.std_out)
# Restart the service
tasks.restart(ctx=self._create_context('restart'))
# Stop the service
tasks.stop(ctx=self._create_context('stop'))
# Uninstall, delete the files
tasks.uninstall(ctx=self._create_context('uninstall'))
# Assert files are gone
self.assertFalse(runner.exists(path=RUNTIME_AGENT_PATH))
self.assertFalse(
runner.exists(
path='C:\{0}'.format(AGENT_EXEC_FILE_NAME)))
