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()