Updated pull to lease.

Author: kurtisvg

cloud-tasks/src/main/java/com.example.cloudtasks/PullQueue.java

@@ -25,17 +25,18 @@
 import com.google.api.services.cloudtasks.v2beta2.CloudTasksScopes;
 import com.google.api.services.cloudtasks.v2beta2.model.AcknowledgeTaskRequest;
 import com.google.api.services.cloudtasks.v2beta2.model.CreateTaskRequest;
+import com.google.api.services.cloudtasks.v2beta2.model.LeaseTasksRequest;
+import com.google.api.services.cloudtasks.v2beta2.model.LeaseTasksResponse;
 import com.google.api.services.cloudtasks.v2beta2.model.PullMessage;
-import com.google.api.services.cloudtasks.v2beta2.model.PullTasksRequest;
-import com.google.api.services.cloudtasks.v2beta2.model.PullTasksResponse;
 import com.google.api.services.cloudtasks.v2beta2.model.Task;
 import com.google.common.io.BaseEncoding;
-import java.io.IOException;
 import net.sourceforge.argparse4j.ArgumentParsers;
 import net.sourceforge.argparse4j.inf.ArgumentParser;
 import net.sourceforge.argparse4j.inf.Namespace;
 import net.sourceforge.argparse4j.inf.Subparsers;
 
+import java.io.IOException;
+
 
 public class PullQueue {
 
@@ -110,15 +111,15 @@ private static Task pullTask(
     CloudTasks client = createAuthorizedClient();
 
     // Create the PullTasksRequest
-    PullTasksRequest request = new PullTasksRequest().setMaxTasks(1).setLeaseDuration("600s");
+    LeaseTasksRequest request = new LeaseTasksRequest().setMaxTasks(1).setLeaseDuration("600s");
 
     //Execute the request and return the pulled task
-    PullTasksResponse response = client
+    LeaseTasksResponse response = client
         .projects()
         .locations()
         .queues()
         .tasks()
-        .pull(queueName, request)
+        .lease(queueName, request)
         .execute();
     return response.getTasks().get(0);
   }

cloud-tasks/src/test/java/com/example/cloudtasks/PullQueueIT.java

@@ -16,17 +16,18 @@
 
 package com.example.cloudtasks;
 
-import static org.junit.Assert.assertNotNull;
-import static org.junit.Assert.assertTrue;
-
-import java.io.ByteArrayOutputStream;
-import java.io.PrintStream;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
 
+import java.io.ByteArrayOutputStream;
+import java.io.PrintStream;
+
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertTrue;
+
 /**
  * Integration (system) tests for {@link PullQueue}.
  */