Explore grace period renewal (#328)

quarkiverse · Jan 23, 2025 · 5f6ad27 · 5f6ad27
1 parent da77bf6
commit 5f6ad27
Showing 1 changed file with 35 additions and 3 deletions.
diff --git a/client/src/test/java/io/quarkus/vault/client/VaultCachingTokenProviderTest.java b/client/src/test/java/io/quarkus/vault/client/VaultCachingTokenProviderTest.java
@@ -147,7 +147,7 @@ public CompletionStage<VaultToken> apply(VaultAuthRequest authRequest) {
     public void testExpiringTokensRenew(VaultClient client) throws Exception {
 
         var token = client.auth().token().create(new VaultAuthTokenCreateTokenParams()
-                .setTtl(Duration.ofMinutes(1))
+                .setTtl(Duration.ofHours(1))
                 .setRenewable(true))
                 .thenApply(
                         a -> VaultToken.from(a.getClientToken(), a.isRenewable(), a.getLeaseDuration(), tickableInstanceSource))
@@ -182,14 +182,46 @@ public CompletionStage<VaultToken> apply(VaultAuthRequest authRequest) {
         verify(cachingTokenProvider, times(0))
                 .extend(any(), any());
 
-        tickableInstanceSource.tick(Duration.ofSeconds(45));
+        tickableInstanceSource.tick(Duration.ofMinutes(45));
 
         for (int i = 0; i < 10; i++) {
             testClient.secrets().kv2().listSecrets("/")
                     .toCompletableFuture().get();
         }
 
-        verify(executor, times(21))
+        verify(executor, times(20))
+                .execute(any());
+        verify(tokenProvider, times(1))
+                .apply(any());
+        verify(cachingTokenProvider, times(1))
+                .request(any());
+        verify(cachingTokenProvider, times(0))
+                .extend(any(), any());
+
+        tickableInstanceSource.tick(Duration.ofMinutes(14).plus(Duration.ofSeconds(20)));
+
+        for (int i = 0; i < 10; i++) {
+            testClient.secrets().kv2().listSecrets("/")
+                    .toCompletableFuture().get();
+        }
+
+        verify(executor, times(30))
+                .execute(any());
+        verify(tokenProvider, times(1))
+                .apply(any());
+        verify(cachingTokenProvider, times(1))
+                .request(any());
+        verify(cachingTokenProvider, times(0))
+                .extend(any(), any());
+
+        tickableInstanceSource.tick(Duration.ofSeconds(15));
+
+        for (int i = 0; i < 10; i++) {
+            testClient.secrets().kv2().listSecrets("/")
+                    .toCompletableFuture().get();
+        }
+
+        verify(executor, times(41))
                 .execute(any());
         verify(tokenProvider, times(1))
                 .apply(any());