KNOX-3069 - Fix Flakey Test - TokenServiceResourceTest (apache#938)

lmccay · Nov 9, 2024 · e3c11b6 · e3c11b6
1 parent 43f6185
commit e3c11b6
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/...ken/src/test/java/org/apache/knox/gateway/service/knoxtoken/TokenServiceResourceTest.java b/...ken/src/test/java/org/apache/knox/gateway/service/knoxtoken/TokenServiceResourceTest.java
@@ -1136,8 +1136,13 @@ private void testLimitingTokensPerUser(int configuredLimit, int numberOfTokens,
       tss.addMetadata(tokenId, tokenMetadata);
     }
 
+    ArrayList<String> tokenIDs = new ArrayList<>();
     for (int i = 0; i < numberOfTokens; i++) {
-      acquireToken(tr);
+      final Response tokenResponse = acquireToken(tr);
+      final String tokenId = getTagValue(tokenResponse.getEntity().toString(), "token_id");
+      assertNotNull("TokenID should not be null", tokenId);
+      assertFalse("TokenID must be unique", tokenIDs.contains(tokenId));
+      tokenIDs.add(tokenId);
     }
     final Response getKnoxTokensResponse = getUserTokensResponse(tr);
     final Collection<String> tokens = ((Map<String, Collection<String>>) JsonUtils.getObjectFromJsonString(getKnoxTokensResponse.getEntity().toString()))