fix(test): max test specific for single query

datahub-project · Oct 8, 2024 · 1c10fac · 1c10fac
1 parent 5335454
commit 1c10fac
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/metadata-io/src/test/java/com/linkedin/metadata/entity/ebean/EbeanAspectDaoTest.java b/metadata-io/src/test/java/com/linkedin/metadata/entity/ebean/EbeanAspectDaoTest.java
@@ -56,7 +56,8 @@ public void testGetLatestAspectsForUpdate() throws JsonProcessingException {
 
     testDao.runInTransactionWithRetryUnlocked(
         (txContext) -> {
-          testDao.getLatestAspects(Map.of("urn:li:corpuser:test", Set.of("status")), true);
+          testDao.getLatestAspects(
+              Map.of("urn:li:corpuser:testGetLatestAspectsForUpdate", Set.of("status")), true);
           return "";
         },
         mock(AspectsBatch.class),
@@ -65,7 +66,7 @@ public void testGetLatestAspectsForUpdate() throws JsonProcessingException {
     // Get the captured SQL statements
     List<String> sql =
         LoggedSql.stop().stream()
-            .filter(str -> str.contains("(t0.urn,t0.aspect,t0.version)"))
+            .filter(str -> str.contains("testGetLatestAspectsForUpdate"))
             .toList();
     assertEquals(
         sql.size(), 1, String.format("Found: %s", new ObjectMapper().writeValueAsString(sql)));