-
Notifications
You must be signed in to change notification settings - Fork 1
/
flask_handler_test.py
57 lines (39 loc) · 1.37 KB
/
flask_handler_test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
import pytest
beforeEach = pytest.mark.usefixtures
from flask import Flask
import FlaskDeferredHandler
from google.appengine.ext import deferred
DEFAULT_URL = FlaskDeferredHandler._DEFAULT_URL
@pytest.fixture
def app():
app = Flask(__name__)
app.config['TESTING'] = True
FlaskDeferredHandler.register(app)
return app
messages = []
@pytest.fixture
def clear_messages():
while messages:
messages.pop()
def A(data):
messages.append(data)
@beforeEach("clear_messages")
class TestFlaskDeferredHandler:
def testCatchRoot(self, app, taskqueue):
task = deferred.defer(A, 'A')
headers = [('X-AppEngine-TaskName', task.name)]
with app.test_client() as c:
rv = c.post(DEFAULT_URL, data=task.payload,
headers=[('X-AppEngine-TaskName', task.name)],
content_type='application/octet-stream')
assert ['A'] == messages
assert rv.status_code == 200
def testCatchAll(self, app, taskqueue):
task = deferred.defer(A, 'A')
with app.test_client() as c:
rv = c.post(DEFAULT_URL + '/foo/bar',
data=task.payload,
headers=[('X-AppEngine-TaskName', task.name)],
content_type='application/octet-stream')
assert ['A'] == messages
assert rv.status_code == 200