def test_select_best_ip(self):
best_ip = select_best_ip()
ip = best_ip['stock']['ip']
port = best_ip['stock']['port']
self.assertTrue(isinstance(ip, str), '未获取到ip')
self.assertTrue(isinstance(port, int), '未获取到端口号')
self.assertTrue(ping(ip, port, 'stock') < datetime.timedelta(0, 1, 0), '地址ping不通： {} {} {}'.format(ip, port, ping(ip, port, 'stock')))
# ip = best_ip['future']['ip']
# port = best_ip['future']['port']
# self.assertTrue(ping(ip, port, 'stock') < datetime.timedelta(0, 1, 0), '地址ping不通： {} {} {}'.format(ip, port, ping(ip, port, 'stock')))
code = '000001'
days = 300
start = datetime.datetime.now().date() - datetime.timedelta(days)
end = datetime.datetime.now().date() - datetime.timedelta(10)
data = QA_fetch_get_stock_day(code, start_date=start, end_date=end)
print(data)
self.assertTrue(len(data) > (end - start).days / 2,
'返回数据个数不匹配，数据长度：{},天数（包含节假日）：{}'.format(len(data), (end - start).days / 2))
default_ip = {'stock': {'ip': None, 'port': None},
'future': {'ip': None, 'port': None}}
qasetting = QA_Setting()
qasetting.set_config(
section='IPLIST', option='default', default_value=default_ip)
best_ip = select_best_ip()
ip = best_ip['stock']['ip']
port = best_ip['stock']['port']
self.assertTrue(isinstance(ip, str) or ip is None, '未获取到ip')
self.assertTrue(isinstance(port, int) or port is None, '未获取到端口号')
ip = best_ip['future']['ip']
port = best_ip['future']['port']
self.assertTrue(isinstance(ip, str) or ip is None, '未获取到ip')
self.assertTrue(isinstance(port, int) or port is None, '未获取到端口号')
data = QA_fetch_get_stock_day(code, start, end)
self.assertTrue(len(data) > (end - start).days / 2,
'返回数据个数不匹配，数据长度：{},天数（包含节假日）：{}'.format(len(data), (end - start).days / 2))
def test_select_best_ip(self):
best_ip = select_best_ip()
ip = best_ip['stock']['ip']
port = best_ip['stock']['port']
self.assertTrue(isinstance(ip, str), '未获取到ip')
self.assertTrue(isinstance(port, int), '未获取到端口号')
self.assertTrue(
ping(ip, port, 'stock') < datetime.timedelta(0, 1, 0),
'地址ping不通： {} {} {}'.format(ip, port, ping(ip, port, 'stock')))
type = 'future'
ip = best_ip[type]['ip']
port = best_ip[type]['port']
self.assertTrue(
ping(ip, port, type) < datetime.timedelta(0, 1, 0),
'地址ping不通： {} {} {}'.format(ip, port, ping(ip, port, 'stock')))
code = '000001'
days = 300
start = datetime.datetime.now().date() - datetime.timedelta(days)
end = datetime.datetime.now().date() - datetime.timedelta(10)
data = QA_fetch_get_stock_day(code, start_date=start, end_date=end)
# print(data)
self.assertTrue(
len(data) > (end - start).days / 2,
'返回数据个数不匹配，数据长度：{},天数（包含节假日）：{}'.format(len(data),
(end - start).days / 2))
# 恢复初始化ip，重新测试ip
default_ip = {
'stock': {
'ip': None,
'port': None
},
'future': {
'ip': None,
'port': None
}
}
qasetting = QA_Setting()
qasetting.set_config(section='IPLIST',
option='default',
default_value=default_ip)
filenames = ['stock_ip_list', 'stock_ip_list_MP']
for filename in filenames:
# 删除保存ip的pickle文件
filename = '{}{}{}.pickle'.format(
qasetting.get_config(section='LOG',
option='path',
default_value=""), os.sep, filename)
if os.path.isfile(filename):
os.remove(filename)
best_ip = select_best_ip()
ip = best_ip['stock']['ip']
port = best_ip['stock']['port']
self.assertTrue(isinstance(ip, str), '未获取到ip')
self.assertTrue(isinstance(port, int), '未获取到端口号')
ip = best_ip['future']['ip']
port = best_ip['future']['port']
self.assertTrue(isinstance(ip, str), '未获取到ip')
self.assertTrue(isinstance(port, int), '未获取到端口号')
data = QA_fetch_get_stock_day(code, start, end)
self.assertTrue(
len(data) > (end - start).days / 2,
'返回数据个数不匹配，数据长度：{},天数（包含节假日）：{}'.format(len(data),
(end - start).days / 2))