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)))