def create_guest():
db = DB()
for table, data in datas.items():
db.clear(table)
for d in data:
db.insert(table, d)
db.close()
def init_data():
db = DB()
for table, data in datas.items():
# db.clear(table)
for d in data:
db.insert(table, d)
db.close()
def delete_data():
db = DB()
for table, data in datas.items():
key = keys[table]['key']
value = keys[table]['value']
db.clear(table, key, value)
db.close()
def test_elm():
db = DB()
db.clear("order_push_record", "饿了么")
r = Elm_Order()
assert "ok" in r.elm_order()
db.update("饿了么")
db.close()
def insert_data():
db = DB()
for table, data in datas.items():
key = keys[table]['key']
value = keys[table]['value']
for d in data:
db.insert(table, d)
db.close()
def insert_data1():
db = DB()
for table, data in datas1.items():
key = keys[table]['key']
value = keys[table]['value']
db.clear(table, key, value)
for d in data:
db.insert(table, d)
db.close()
def init_data():
# 创建一个DB类的实例
db = DB()
# print(datas.items())
for table, data in datas.items():
db.clear(table)
for d in data:
db.insert(table, d)
db.close()
def test_meituan():
'''美团下单'''
db=DB()
db.clear("order_push_record","美团")
r = Meituan_Order()
assert "ok" in r.meituan_order()
db.update("美团")
db.close()
def test_baidu():
'''百度下单'''
db=DB()
db.clear("fosordertranslog","百度")
r=Baidu_Order()
assert "error" in r.baidu_order()
db.update("百度")
db.close()
class DialogueTest(unittest.TestCase):
'''测试对话管理'''
def setUp(self):
self.db = DB()
self.base_url = "http://39.96.87.250:8000/xiaohuang/"
self.table_name = "myusecase"
self.usecase = self.db.select(self.table_name)
self.result = {}
def tearDown(self):
self.db.close()
print(self.result)
def test_beauty_dialog(self):
'''测试闲聊机器人对话'''
pass
class Operate():
def __init__(self, db_name):
self.connection = DB(db_name).connection
"""
获取订单信息
:param username:
:return result:
"""
def get_invest_order(self, username):
sql = """SELECT * FROM `pj_test3_user`.`t_user` a LEFT JOIN `pj_test3_core`.`t_plan_invest` b ON a.`id`=b.`investor_id` WHERE a.`cellphone`='%s' AND b.`status`=%d;""" % (
username, 200)
with self.connection.cursor() as cursor:
cursor.execute(sql)
result = cursor.fetchall()
return result
"""
设置当天债转进池金额为0
"""
def set_debit_limit_zero(self):
self.connection.ping(reconnect=True)
sql = """UPDATE `pj_test3_other`.`t_debt_limit` a SET a.debt_limit=0 WHERE a.`effect_date`=DATE(NOW());"""
with self.connection.cursor() as cursor:
cursor.execute(sql)
self.connection.commit()
"""
获取订单状态
:param username:
:return result:
"""
def get_invest_order_status(self, username):
sql = """SELECT b.`status` FROM `pj_test3_user`.`t_user` a LEFT JOIN `pj_test3_core`.`t_plan_invest` b ON a.`id`=b.`investor_id` WHERE a.`cellphone`='%s' ORDER BY b.`id` DESC LIMIT 1;""" % (
username)
with self.connection.cursor() as cursor:
cursor.execute(sql)
result = cursor.fetchall()[0]
return result
def close_invest_order_check(self):
self.connection.close()
