blob: 54d8b112eca4b8c49fdb08c20441c8c07771b122 [file] [log] [blame]
from OnosLog import OnosLog
from scapy.all import log
from onosclidriver import OnosCliDriver
from OnosCtrl import OnosCtrl, get_mac
from docker import Client
from CordContainer import *
import json
import requests
import unittest
import os
import time
def get_controller_names(controllers):
controller_names = [ 'cord-onos' if controllers.index(c) == 0 else 'cord-onos-{}'.format(controllers.index(c)+1) for c in controllers ]
return controller_names
def get_controller_map(controllers):
controller_map = ( ('cord-onos' if controllers.index(c) == 0 else 'cord-onos-{}'.format(controllers.index(c)+1),c) for c in controllers )
return dict(controller_map)
class CordLogger(unittest.TestCase):
controllers = os.getenv('ONOS_CONTROLLER_IP', '').split(',')
controller_names = get_controller_names(controllers)
controller_map = get_controller_map(controllers)
cliSessions = {}
onosLogLevel = 'INFO'
curLogLevel = onosLogLevel
testLogLevel = os.getenv('LOG_LEVEL', onosLogLevel)
setup_dir = os.path.join(os.path.dirname(os.path.realpath(__file__)), '../setup')
archive_dir = os.path.join(setup_dir, 'test_logs')
onos_data_dir = os.path.join(setup_dir, 'cord-onos-data')
@classmethod
def cliSessionEnter(cls):
try:
for controller in cls.controllers:
if not controller:
continue
retries = 0
while retries < 30:
cli = OnosCliDriver(controller = controller, connect = True)
if cli.handle:
cls.cliSessions[controller] = cli
break
else:
retries += 1
time.sleep(2)
except:
pass
@classmethod
def cliSessionExit(cls):
try:
for controller, cli in cls.cliSessions.items():
if cli:
cli.disconnect()
except:
pass
def setUp(self):
'''Read the log buffer'''
self.logSet()
try:
onosLog = OnosLog()
st, output = onosLog.get_log()
except: pass
def tearDown(self):
'''Dump the log buffer for ERRORS/warnings'''
#reset the log level back to default log level after a test
self.logSet(level = self.onosLogLevel)
try:
onosLog = OnosLog()
st, output = onosLog.get_log( ('ERROR','WARN') )
if st and output:
log.info('\nTest %s has errors and warnings\n' %self._testMethodName)
log.info('%s' %output)
else:
log.info('\nTest %s has no errors and warnings in the logs' %self._testMethodName)
except: pass
try:
self.archive_results(self._testMethodName)
except: pass
@classmethod
def archive_results(cls, testName, controllers = None, iteration = None, archive_partition = False):
if not os.path.exists(cls.onos_data_dir):
return cls.archive_results_unshared(testName, controllers = controllers, iteration = iteration)
if not os.path.exists(cls.archive_dir):
os.mkdir(cls.archive_dir)
if controllers is None:
controllers = cls.controllers
controller_map = cls.controller_map
else:
controller_map = get_controller_map(controllers)
iteration_str = '' if iteration is None else '_{}'.format(iteration)
if archive_partition is False:
archive_target = 'log'
tar_options = ''
else:
archive_target = ''
tar_options = '--exclude=cache --exclude=tmp'
for c in controller_map.keys():
archive_file = os.path.join(cls.archive_dir,
'logs_{}_{}{}.tar.gz'.format(controller_map[c], testName, iteration_str))
archive_path = os.path.join(cls.setup_dir, '{}-data'.format(c), archive_target)
cmd = 'cd {} && tar cvzf {} . {}'.format(archive_path, archive_file, tar_options)
try:
os.system(cmd)
except: pass
@classmethod
def archive_results_unshared(cls, testName, controllers = None, iteration = None, cache_result = False):
log_map = {}
if controllers is None:
controllers = cls.controllers
else:
if type(controllers) in [ str, unicode ]:
controllers = [ controllers ]
try:
for controller in controllers:
onosLog = OnosLog(host = controller)
st, output = onosLog.get_log(cache_result = cache_result)
log_map[controller] = (st, output)
except:
return
if not os.path.exists(cls.archive_dir):
os.mkdir(cls.archive_dir)
for controller, results in log_map.items():
st, output = results
if st and output:
iteration_str = '' if iteration is None else '_{}'.format(iteration)
archive_file = os.path.join(cls.archive_dir,
'logs_{}_{}{}'.format(controller, testName, iteration_str))
archive_cmd = 'gzip -9 -f {}'.format(archive_file)
if os.access(archive_file, os.F_OK):
os.unlink(archive_file)
with open(archive_file, 'w') as fd:
fd.write(output)
try:
os.system(archive_cmd)
except: pass
@classmethod
def logSet(cls, level = None, app = 'org.onosproject', controllers = None, forced = False):
#explicit override of level is allowed to reset log levels
if level is None:
level = cls.testLogLevel
#if we are already at current/ONOS log level, there is nothing to do
if forced is False and level == cls.curLogLevel:
return
if controllers is None:
controllers = cls.controllers
else:
if type(controllers) in [str, unicode]:
controllers = [ controllers ]
cls.cliSessionEnter()
try:
for controller in controllers:
if cls.cliSessions.has_key(controller):
cls.cliSessions[controller].logSet(level = level, app = app)
cls.curLogLevel = level
except:
pass
cls.cliSessionExit()
@classmethod
def stat_option(cls, stat = None, serverDetails = None):
# each stat option we can do some specific functions
if stat is None:
stat = cls.statOptionsList
if serverDetails is None:
serverDetails = cls.serverOptionsList
stat_choice = 'COLLECTD'
test_name = cls.testHostName
test_image = 'cordtest/nose'
if stat_choice in stat:
onos_ctrl = OnosCtrl('org.onosproject.cpman')
status, _ = onos_ctrl.activate()
if serverDetails is '':
## default Test Container is used to install CollectD
pass
elif serverDetails in 'NEW':
test_image = 'cord-test/exserver'
test_name ='cord-collectd'
else:
pass
# cls.connect_server(serverDetails)
## TO-DO for already up and running server, install collectd agent etc...
cls.start_collectd_agent_in_server(name = test_name, image = test_image)
for controller in cls.controllers:
if not controller:
continue
url_mem_stats = 'http://%s:8181/onos/cpman/controlmetrics/memory_metrics'%(controller)
url_cpu_stats = 'http://%s:8181/onos/cpman/controlmetrics/cpu_metrics'%(controller)
auth = ('karaf', 'karaf')
cls.collectd_agent_metrics(controller, auth, url = url_cpu_stats)
cls.collectd_agent_metrics(controller, auth, url = url_mem_stats)
return
@classmethod
def collectd_agent_metrics(cls,controller=None, auth =None, url = None):
'''This function is getting rules from ONOS with json format'''
if url:
resp = requests.get(url, auth = auth)
log.info('Collectd agent has provided metrics via ONOS controller, url = %s \nand status = %s' %(url,resp.json()))
return resp
@classmethod
def start_collectd_agent_in_server(cls, name = None, image = None):
container_cmd_exec = Container(name = name, image = image)
tty = False
dckr = Client()
cmd = 'sudo /etc/init.d/collectd start'
i = container_cmd_exec.execute(cmd = cmd, tty= tty, stream = True)
return
@classmethod
def disable_onos_apps(cls, stat = None, app = None):
stat_choice = 'COLLECTD'
if stat is None:
stat = cls.statOptionsList
if stat_choice in stat:
onos_ctrl = OnosCtrl('org.onosproject.cpman')
status, _ = onos_ctrl.deactivate()