From f09e18cdb82433f07963bde5bac67ffae169b553 Mon Sep 17 00:00:00 2001
From: david-leifker <114954101+david-leifker@users.noreply.github.com>
Date: Wed, 9 Oct 2024 09:35:42 -0500
Subject: [PATCH] fix(test): fix test specific for single query (#11567)

---
 .../linkedin/metadata/entity/ebean/EbeanAspectDaoTest.java   | 5 +++--
 1 file changed, 3 insertions(+), 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
index 109c9b5c44efb9..7d1fdd34026f9f 100644
--- 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)));