Use capsys to test tasks

Author: andrewsg

appengine/flexible/tasks/main.py

@@ -14,8 +14,6 @@
 
 """App Engine app to serve as an endpoint for App Engine queue samples."""
 
-import logging
-
 from flask import Flask, request
 
 app = Flask(__name__)
@@ -25,8 +23,7 @@
 def log_payload():
     """Log the request payload."""
     payload = request.get_data(as_text=True) or '(empty payload)'
-    # Send our test logging message as a "warn" to ensure visibility.
-    logging.warn('Received task with payload: {}'.format(payload))
+    print('Received task with payload: {}'.format(payload))
     return 'Printed task payload: {}'.format(payload)
 
 

appengine/flexible/tasks/main_test.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import mock
 import pytest
 
 
@@ -28,19 +27,19 @@ def test_index(app):
     assert r.status_code == 200
 
 
-@mock.patch('logging.warn')
-def test_log_payload(logging_mock, app):
-    payload = 'hello'
+def test_log_payload(capsys, app):
+    payload = 'test_payload'
 
     r = app.post('/log_payload', data=payload)
     assert r.status_code == 200
 
-    assert logging_mock.called
+    out, _ = capsys.readouterr()
+    assert payload in out
 
 
-@mock.patch('logging.warn')
-def test_empty_payload(logging_mock, app):
+def test_empty_payload(capsys, app):
     r = app.post('/log_payload')
     assert r.status_code == 200
 
-    assert logging_mock.called
+    out, _ = capsys.readouterr()
+    assert 'empty payload' in out