No public description

PiperOrigin-RevId: 636614826
googlemaps · May 23, 2024 · edcc9aa · edcc9aa
1 parent b784573
commit edcc9aa
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/...st/java/com/google/fleetengine/auth/sample/DeliveryConsumerTokenValidationScriptTest.java b/...st/java/com/google/fleetengine/auth/sample/DeliveryConsumerTokenValidationScriptTest.java
@@ -4,7 +4,6 @@
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.mock;
-import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 
@@ -49,17 +48,17 @@ public void setup() throws IOException {
   }
 
   @Test
-  public void run_callsSearchTasks_withTrackingId() throws Throwable {
+  public void run_callsGetTaskTrackingInfo_withTrackingId() throws Throwable {
     DeliveryConsumerTokenValidationScript script =
         new DeliveryConsumerTokenValidationScript(runtime, configuration, commandsFactory);
     script.run(TEST_TRACKING_ID);
 
     ArgumentCaptor<String> captor = ArgumentCaptor.forClass(String.class);
-    verify(deliveryServiceCommandsSuccess, times(1)).searchTasks(captor.capture());
+    verify(deliveryServiceCommandsSuccess).getTaskTrackingInfo(captor.capture());
     assertEquals(TEST_TRACKING_ID, captor.getValue());
 
     runtime.wasExpectPermissionDeniedCalled();
-    verify(deliveryServiceCommandsFails, times(1)).searchTasks(captor.capture());
+    verify(deliveryServiceCommandsFails).getTaskTrackingInfo(captor.capture());
     assertEquals(TEST_TRACKING_ID, captor.getValue());
   }
 }
diff --git a/...java/com/google/fleetengine/auth/sample/DeliveryFleetReaderTokenValidationScriptTest.java b/...java/com/google/fleetengine/auth/sample/DeliveryFleetReaderTokenValidationScriptTest.java
@@ -4,7 +4,6 @@
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.mock;
-import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 
@@ -46,13 +45,13 @@ public void setup() throws IOException {
   }
 
   @Test
-  public void run_callsSearchTasks_withTrackingId() throws Throwable {
+  public void run_callsGetTaskTrackingInfo_withTrackingId() throws Throwable {
     DeliveryFleetReaderTokenValidationScript script =
         new DeliveryFleetReaderTokenValidationScript(runtime, configuration, commandsFactory);
     script.run(TEST_TRACKING_ID);
 
     ArgumentCaptor<String> captor = ArgumentCaptor.forClass(String.class);
-    verify(deliveryServiceCommands, times(1)).searchTasks(captor.capture());
+    verify(deliveryServiceCommands).getTaskTrackingInfo(captor.capture());
     assertEquals(TEST_TRACKING_ID, captor.getValue());
   }
 }