def test_expired(self):
expired_token = generate_token(b'buaa', -1)
b, result = certify_token('wechatterbot',
expired_token.encode('utf-8'))
temp = json.loads(result.data.decode('utf-8'))
self.assertEqual(temp['code'], 10000043)
self.assertEqual(b, False)
def test_wrong_token(self):
wrong_token = generate_token(b'what', 3600)
r = self.app.get('admin/search_rule?username=wechatterbot' +
'&token=' + wrong_token + '&id=1',
headers=self.myheaders)
result = json.loads(r.data.decode('utf-8'))
self.assertEqual(result['code'], 10000044)
self.assertEqual(r.status_code, 401)
def test_check_fail(self):
wrong_token = generate_token(b'what', 3600)
data = {'username': '******', 'token': wrong_token}
r = self.app.post('admin/certify',
data=json.dumps(data),
headers=self.myheaders)
result = json.loads(r.data.decode('utf-8'))
self.assertEqual(result['code'], 10000044)
self.assertEqual(r.status_code, 401)
def setUp(self):
self.myheaders = {'Content-Type': 'application/json'}
self.token = generate_token(b'buaa', 3600)
self.app = create_app().test_client()
def test_wrong_token(self):
wrong_token = generate_token(b'what', 3600)
b, result = certify_token('wechatterbot', wrong_token.encode('utf-8'))
temp = json.loads(result.data.decode('utf-8'))
self.assertEqual(temp['code'], 10000044)
self.assertEqual(b, False)
def setUp(self):
self.myheaders = {'Content-Type': 'application/json'}
self.token = generate_token(b'buaa', 3600)
