valitydev/yandex-tank
14
0
Fork 0
mirror of https://github.com/valitydev/yandex-tank.git synced 2024-11-06 10:25:17 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

bugfixes

Browse Source
This commit is contained in:
Timur Torubarov 2016-06-14 16:53:58 +03:00
parent 1d6b611e80
commit ba4522f7d5
1 changed files with 25 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
yandextank/plugins/BatteryHistorian/plugin.py
View File

@ -34,37 +34,36 @@ class Plugin(AbstractPlugin):
def configure(self):
self.device_id = self.get_option("device_id", None).strip()
if not self.device_id:
modify = ''
else:
if self.device_id:
modify = '-s {device_id}'.format(device_id=self.device_id)
for key, value in self.cmds.iteritems():
self.cmds[key] = value % modify
self.logfile = self.core.mkstemp(".log", "battery_historian_")
self.core.add_artifact_file(self.logfile)
for key, value in self.cmds.iteritems():
self.cmds[key] = value % modify
self.logfile = self.core.mkstemp(".log", "battery_historian_")
self.core.add_artifact_file(self.logfile)
def prepare_test(self):
try:
out = subprocess.check_output(self.cmds['enable_full_log'], shell=True)
logger.debug('Enabling full-log: %s', out)
out = subprocess.check_output(self.cmds['reset'], shell=True)
logger.debug('Reseting battery stats: %s', out)
except CalledProcessError:
logger.error('Error trying to prepare battery historian plugin', exc_info=True)
if self.device_id:
try:
out = subprocess.check_output(self.cmds['enable_full_log'], shell=True)
logger.debug('Enabling full-log: %s', out)
out = subprocess.check_output(self.cmds['reset'], shell=True)
logger.debug('Reseting battery stats: %s', out)
except CalledProcessError:
logger.error('Error trying to prepare battery historian plugin', exc_info=True)
def end_test(self, retcode):
try:
logger.debug('dumping battery stats')
dump = subprocess.Popen(self.cmds['dump'], stdout=subprocess.PIPE, shell=True).communicate()[0]
out = subprocess.check_output(self.cmds['disable_full_log'], shell=True)
logger.debug('Disabling fulllog: %s', out)
out = subprocess.check_output(self.cmds['reset'], shell=True)
logger.debug('Battery stats reset: %s', out)
except CalledProcessError:
logger.error('Error trying to collect battery historian plugin data', exc_info=True)
with open(self.logfile, 'w') as f:
f.write(dump)
if self.device_id:
try:
logger.debug('dumping battery stats')
dump = subprocess.Popen(self.cmds['dump'], stdout=subprocess.PIPE, shell=True).communicate()[0]
out = subprocess.check_output(self.cmds['disable_full_log'], shell=True)
logger.debug('Disabling fulllog: %s', out)
out = subprocess.check_output(self.cmds['reset'], shell=True)
logger.debug('Battery stats reset: %s', out)
except CalledProcessError:
logger.error('Error trying to collect battery historian plugin data', exc_info=True)
with open(self.logfile, 'w') as f:
f.write(dump)
return retcode
def is_test_finished(self):