def test_extends(self):
with main.app.app_context():
main.init()
self.insert_fake_key_list()
web_interface._add_extend('文档', 'http://www.baidu.com')
web_interface._add_extend('ss', 'http://www.ee.com')
c = main.app.test_client()
res = c.get('/extend')
res = json.loads(res.data.decode())
assert len(res['extend']) == 2
web_interface._delete_extend(1)
res = c.get('/extend')
res = json.loads(res.data.decode())
assert len(res['extend']) == 1
web_interface._delete_extend(2)
res = c.get('/extend')
res = json.loads(res.data.decode())
assert len(res['extend']) == 0
web_interface._add_extend('second', 'http://localhost:9001/action')
res = c.get('/extend')
res = json.loads(res.data.decode())
param = dict(
verify=main.app.config['verify'],
time=datetime.datetime.now().strftime('%Y-%m-%d %H.%M.%S'),
data=dict(
id=res['extend'][0]['passName'],
param='msg=hello'
)
)
res = self.send_request('send_extend', param)
res = json.loads(res.data.decode())
assert res.get('status') == 'success'
def test_add_extend(self):
with main.app.app_context():
main.init()
web_interface._add_extend('你好', 'http://www.baidu.com')
assert db.get_db().execute('select * from Extend').fetchone()['name'] == '你好'
