Fix unit test error

This commit is contained in:
changzhi1990 2016-01-15 15:35:53 +08:00
parent 934b7bd70a
commit 9f32be83a6
3 changed files with 21 additions and 4 deletions

View File

@ -60,8 +60,8 @@ class IPerfDriver(object):
out_dict.pop()
out_data = out_dict[-1].split()
data = dict()
data['Bandwidth'] = out_data[-2] + out_data[-1]
data['Transfer'] = out_data[-4] + out_data[-3]
data['Bandwidth'] = out_data[-2] + ' ' + out_data[-1]
data['Transfer'] = out_data[-4] + ' ' + out_data[-3]
data['Interval'] = out_data[-6]
return data
raise Exception('Start iperf failed, please check on the node.')

View File

@ -21,7 +21,6 @@ import sys
from cliff.command import Command
from cliff.lister import Lister
from json import JSONDecoder
from stetho.stethoclient.constants import AGENT_INFOS
LISTEN_PORT = 9698
SETUP_LINK_IP_PRE = "192.168.100."
@ -48,6 +47,16 @@ class Logger():
print Logger.FAIL + info + Logger.ENDC
try:
from stetho.stethoclient.constants import AGENT_INFOS
except:
AGENT_INFOS = {
'agent-64': "127.0.0.1",
'agent-65': "127.0.0.1",
}
Logger.log_fail("Import stetho configure file fail. Use fake data!")
def setup_server(agent):
log = logging.getLogger(__name__)
if agent in AGENT_INFOS:

View File

@ -20,7 +20,6 @@ import socket
import sys
from cliff.lister import Lister
from stetho.stethoclient.constants import AGENT_INFOS
LISTEN_PORT = 9698
@ -45,6 +44,15 @@ class Logger():
def log_fail(info):
print Logger.FAIL + info + Logger.ENDC
try:
from stetho.stethoclient.constants import AGENT_INFOS
except:
AGENT_INFOS = {
'agent-64': "127.0.0.1",
'agent-65': "127.0.0.1",
}
Logger.log_fail("Import stetho configure file fail. Use fake data!")
def setup_server(agent):
log = logging.getLogger(__name__)