Unlock 30% off on Manual Testing Annual Plans this Holiday Season.

Copied to Clipboard!

How to use _stop_agent method in yandex-tank

Best Python code snippet using yandex-tank

systems.py

Source:systems.py

...139    def install(self, params):140        if 'installer' in params:141            self.INSTALLER = params.installer142        self._update_system(params)143        self._stop_agent(params)144        response = self.s3.get_object(Bucket=params.bucket, Key=params.key)145        with open(self.INSTALLER, 'wb') as f:146            f.write(response['Body'].read())147        subprocess.check_call(148            ['chmod', '+x', './{0}'.format(self.INSTALLER)]149        )150        credentials = self.session.get_credentials()151        environment = os.environ.copy()152        environment['AWS_REGION'] = params.region153        environment['AWS_ACCESS_KEY_ID'] = credentials.access_key154        environment['AWS_SECRET_ACCESS_KEY'] = credentials.secret_key155        if credentials.token is not None:156            environment['AWS_SESSION_TOKEN'] = credentials.token157        subprocess.check_call(158            ['./{0}'.format(self.INSTALLER), 'auto'],159            env=environment160        )161    def uninstall(self, params):162        process = self._stop_agent(params)163        if process.returncode == 0:164            self._remove_agent(params)165    def _update_system(self, params):166        raise NotImplementedError('preinstall')167    def _remove_agent(self, params):168        raise NotImplementedError('remove_agent')169    def _stop_agent(self, params):170        process = subprocess.Popen(171            ['service', 'codedeploy-agent', 'stop'],172            stdout=subprocess.PIPE,173            stderr=subprocess.PIPE174        )175        (output, error) = process.communicate()176        if process.returncode != 0 and params.not_found_msg not in error:177            raise RuntimeError(178                'Failed to stop the AWS CodeDeploy Agent:\n{0}'.format(error)179            )180        return process181class Ubuntu(Linux):182    def _update_system(self, params):183        subprocess.check_call(['apt-get', '-y', 'update'])184        subprocess.check_call(['apt-get', '-y', 'install', 'ruby2.0'])185    def _remove_agent(self, params):186        subprocess.check_call(['dpkg', '-r', 'codedeploy-agent'])187    def _stop_agent(self, params):188        params.not_found_msg = 'codedeploy-agent: unrecognized service'189        return Linux._stop_agent(self, params)190class RHEL(Linux):191    def _update_system(self, params):192        subprocess.check_call(['yum', '-y', 'install', 'ruby'])193    def _remove_agent(self, params):194        subprocess.check_call(['yum', '-y', 'erase', 'codedeploy-agent'])195    def _stop_agent(self, params):196        params.not_found_msg = 'Redirecting to /bin/systemctl stop  codedeploy-agent.service'...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.