def set_db_cfg(self, cfg_name='mydb', sec_name='mysql'):
"""
set_db_cfg_type 设置类型
.. doctest::
mysql_o.set_db_cfg_type('debug')
mysql_o.set_db_cfg()
"""
# 获取数据库配置
app_loc = cofFile.get_app_loc()
if not self.db_cfg_type:
cfg_obj = cofConf.MyCfg(app_loc + os.sep + 'cfgtype.ini')
cfg_obj.set_section('cfg')
cfg_type = cfg_obj.get('type')
else:
cfg_type = self.db_cfg_type
db_cfg_obj = cofConf.MyCfg(os.sep + cfg_name + '.ini')
db_cfg_obj.set_section(sec_name)
self.host = db_cfg_obj.get('host')
self.port = int(db_cfg_obj.get('port'))
self.user = db_cfg_obj.get('user')
self.passwd = db_cfg_obj.get('passwd')
self.dbname = db_cfg_obj.get('db')
self.db = MySQLdb.connect(host=self.host, port=self.port, user=self.user, passwd=self.passwd, db=self.dbname)
self.cursor = self.db.cursor()
pre_sql = "SET NAMES 'utf8'"
self.exec_sql(pre_sql)
def __init__(self, path='params.ini'):
app_loc = cofFile.get_app_loc()
path = app_loc + 'config' + os.sep + 'lc_params_cfg' + os.sep + path
ex_path = cofFile.expand_links(path)
if not os.path.exists(ex_path):
print "config file is not exist (" + ex_path + ")"
exit()
self.path = ex_path
self.sec = ""
self.cfg_obj = ConfigParser.ConfigParser()
self.cfg_obj.read(self.path)
def __init__(self, path='params.ini'):
app_loc = cofFile.get_app_loc()
path = app_loc + 'config' + os.sep + 'lc_params_cfg' + os.sep + path
ex_path = cofFile.expand_links(path)
if not os.path.exists(ex_path):
print "config file is not exist (" + ex_path + ")"
exit()
self.path = ex_path
self.sec = ""
self.cfg_obj = ConfigParser.ConfigParser()
self.cfg_obj.read(self.path)
def judge_sensitive_word(words):
app_loc = cofFile.get_app_loc()
path = app_loc + 'api_call/sensitive_word/sensitiveWord.txt'
ex_path = cofFile.expand_links(path)
f = open(ex_path)
s = f.read()
s1 = re.split(' ', s)
for i in s1:
if i != "":
if i in words:
return False
return True
def set_db_cfg(self, cfg_name='mydb', sec_name='mysql'):
"""
set_db_cfg_type 设置类型
.. doctest::
mysql_o.set_db_cfg_type('debug')
mysql_o.set_db_cfg()
"""
# 获取数据库配置
app_loc = cofFile.get_app_loc()
if not self.db_cfg_type:
cfg_obj = cofConf.MyCfg(app_loc + os.sep + 'cfgtype.ini')
cfg_obj.set_section('cfg')
cfg_type = cfg_obj.get('type')
else:
cfg_type = self.db_cfg_type
db_cfg_obj = cofConf.MyCfg(os.sep + cfg_name + '.ini')
db_cfg_obj.set_section(sec_name)
self.host = db_cfg_obj.get('host')
self.port = int(db_cfg_obj.get('port'))
self.user = db_cfg_obj.get('user')
self.passwd = db_cfg_obj.get('passwd')
self.dbname = db_cfg_obj.get('db')
self.db = MySQLdb.connect(host=self.host,
port=self.port,
user=self.user,
passwd=self.passwd,
db=self.dbname)
self.cursor = self.db.cursor()
pre_sql = "SET NAMES 'utf8'"
self.exec_sql(pre_sql)
from nose import suite
from nose import loader
from cof.conf import MyCfg
opt_parser = OptionParser()
opt_parser.add_option("-s",
"--suite",
dest="suite",
help=u"指定运行的测试套件",
metavar="SUITE")
(options, opt_args) = opt_parser.parse_args()
suite_opt = options.suite
app_loc = CoFileM.get_app_loc()
app_cfg = CoConfM.get_cfg_type()
app_gbl_cfg_file = CoConfM.get_cfg_type_path()
if not suite_opt:
suite_config = "suites.json"
else:
# 到指定的目录获取对应的测试集配置
suite_config = app_loc + "config" + os.sep + app_cfg + os.sep + suite_opt + ".json"
# 获取当前时间
date = cofTime.get_date_ymd() # 日期【20150213】
timestamp = str(cofTime.get_ts()) # 时间戳【1423813170】
# 当前路径
path = os.path.abspath(__file__)
import datetime import StringIO import sys import unittest from xml.sax import saxutils import os import re import time import cof.co_time as CoTimeM import cof.file as CoFileM import types # by linsixiao app_loc = CoFileM.get_app_loc() # ------------------------------------------------------------------------ # The redirectors below are used to capture output during testing. Output # sent to sys.stdout and sys.stderr are automatically captured. However # in some cases sys.stdout is already cached before HTMLTestRunner is # invoked (e.g. calling logging.basicConfig). In order to capture those # output, use the redirectors for the cached stream. # # e.g. # >>> logging.basicConfig(stream=HTMLTestRunner.stdout_redirector) # >>> class OutputRedirector(object):
def get_cfg_type_path():
filepath = cofFile.get_app_loc() + 'cfgtype.ini'
return filepath