Split Cloud Tasks snippet methods (#2332)

Author: averikitsch

appengine/flexible/tasks/snippets.py

@@ -176,26 +176,44 @@ def create_task_with_name(project, location, queue, task_name):
 
 
 def delete_task(project, location, queue):
-    # [START taskqueues_setup]
+    # [START taskqueues_deleting_tasks]
     client = tasks.CloudTasksClient()
 
     # TODO(developer): Uncomment these lines and replace with your values.
     # project = 'my-project-id'
     # location = 'us- central1'
     # queue = 'queue1'
-    # [END taskqueues_setup]
 
-    # [START taskqueues_deleting_tasks]
     task_path = client.task_path(project, location, queue, 'foo')
     response = client.delete_task(task_path)
     # [END taskqueues_deleting_tasks]
+    return response
 
+
+def purge_queue(project, location, queue):
     # [START taskqueues_purging_tasks]
+    client = tasks.CloudTasksClient()
+
+    # TODO(developer): Uncomment these lines and replace with your values.
+    # project = 'my-project-id'
+    # location = 'us- central1'
+    # queue = 'queue1'
+
     queue_path = client.queue_path(project, location, queue)
     response = client.purge_queue(queue_path)
     # [END taskqueues_purging_tasks]
+    return response
+
 
+def pause_queue(project, location, queue):
     # [START taskqueues_pause_queue]
+    client = tasks.CloudTasksClient()
+
+    # TODO(developer): Uncomment these lines and replace with your values.
+    # project = 'my-project-id'
+    # location = 'us- central1'
+    # queue = 'queue1'
+
     queue_path = client.queue_path(project, location, queue)
     response = client.pause_queue(queue_path)
     # [END taskqueues_pause_queues]

appengine/flexible/tasks/snippets_test.py

@@ -71,14 +71,30 @@ def test_create_task_with_name():
 
 @pytest.mark.order6
 def test_delete_task():
+    result = snippets.delete_task(
+        TEST_PROJECT_ID, TEST_LOCATION, QUEUE_NAME_1)
+    assert result is None
+
+
+@pytest.mark.order7
+def test_purge_queue():
     name = "projects/{}/locations/{}/queues/{}".format(
         TEST_PROJECT_ID, TEST_LOCATION, QUEUE_NAME_1)
-    result = snippets.delete_task(
+    result = snippets.purge_queue(
         TEST_PROJECT_ID, TEST_LOCATION, QUEUE_NAME_1)
     assert name in result.name
 
 
-@pytest.mark.order7
+@pytest.mark.order8
+def test_pause_queue():
+    name = "projects/{}/locations/{}/queues/{}".format(
+        TEST_PROJECT_ID, TEST_LOCATION, QUEUE_NAME_1)
+    result = snippets.pause_queue(
+        TEST_PROJECT_ID, TEST_LOCATION, QUEUE_NAME_1)
+    assert name in result.name
+
+
+@pytest.mark.order9
 def test_delete_queue():
     result = snippets.delete_queue(
         TEST_PROJECT_ID, TEST_LOCATION, QUEUE_NAME_1)
@@ -89,7 +105,7 @@ def test_delete_queue():
     assert result is None
 
 
-@pytest.mark.order8
+@pytest.mark.order10
 def test_retry_task():
     QUEUE_SIZE = 3
     QUEUE_NAME = []