def test_bump(self): with self.client as c: token = generate_key(SystemRandom()) with c.session_transaction() as s: s['token'] = token c.post('/admin/tip/action/', data={'action': 'Bump', 'rowid': '1', 't': token}) with c.session_transaction(method="POST", data={'t': token}) as s: self.assertIn(' tips have been bumped.', s['_flashes'][0][1])
def test_reply_to_reflection_passthrough(self): with self.client as c: token = generate_key(SystemRandom()) with c.session_transaction() as s: s['token'] = token s['last-request'] = datetime.now() - timedelta(hours=1) response2 = c.post('/reflection/1/reply', data={ 'p': 'test4', 'r': 'testmessage', 't': token, }, headers={ 'User-Agent': 'Flask test client', 'Referer': 'unittest', }) self.assertEqual(response2.status_code, 200) self.assertIn('json', response2.mimetype) response_data = json.loads(response2.data) self.assertEqual(response_data['status'], 'success') self.assertEqual(response_data['token'], session['token']) token = session['token'] with c.session_transaction(method="POST", data={'t':token}) as s: s['token'] = token s['last-request'] = datetime.now() - timedelta(milliseconds=1001) response3 = c.post('/reflection/1/reply', data={ 'p': 'test4', 'r': 'testmessage', 't': token, }, headers={ 'User-Agent': 'Flask test client', 'Referer': 'unittest', }) self.assertEqual(response3.status_code, 200) self.assertIn('json', response3.mimetype) response_data = json.loads(response3.data) self.assertEqual(response_data['token'], session['token']) self.assertEqual(response_data['status'], 'captcha')