Merge pull request GoogleCloudPlatform#1114 from GoogleCloudPlatform/tasks-tests

andrewsg · web-flow · commit fcd818610bf2 · 2017-09-14T15:53:54.000-07:00
Fix Tasks sample test issues.
diff --git a/appengine/flexible/tasks/main_test.py b/appengine/flexible/tasks/main_test.py
@@ -32,7 +32,7 @@ def test_index(app):
 def test_log_payload(logging_mock, app):
     payload = 'hello'
 
-    r = app.post('/log_payload', payload)
+    r = app.post('/log_payload', data=payload)
     assert r.status_code == 200
 
     assert logging_mock.called
diff --git a/tasks/pull_queue_snippets.py b/tasks/pull_queue_snippets.py
@@ -38,7 +38,7 @@ def create_task(project, queue, location):
     task = {
         'task': {
             'pull_task_target': {
-                'payload': base64.b64encode(payload)
+                'payload': base64.b64encode(payload.encode()).decode()
             }
         }
     }
diff --git a/tasks/pull_queue_snippets_test.py b/tasks/pull_queue_snippets_test.py
@@ -18,7 +18,7 @@
 
 TEST_PROJECT_ID = os.getenv('GCLOUD_PROJECT')
 TEST_LOCATION = os.getenv('TEST_QUEUE_LOCATION', 'us-central1')
-TEST_QUEUE_NAME = os.getenv('TEST_QUEUE_NAME', 'test-queue')
+TEST_QUEUE_NAME = os.getenv('TEST_QUEUE_NAME', 'my-pull-queue')
 
 
 def test_create_task():
diff --git a/tasks/queue.yaml b/tasks/queue.yaml

Original file line number	Diff line number	Diff line change
`@@ -38,7 +38,7 @@ def create_task(project, queue, location):`
`38`	`38`	`task = {`
`39`	`39`	`'task': {`
`40`	`40`	`'pull_task_target': {`
`41`		`- 'payload': base64.b64encode(payload)`
	`41`	`+ 'payload': base64.b64encode(payload.encode()).decode()`
`42`	`42`	`}`
`43`	`43`	`}`
`44`	`44`	`}`