fix: getAspectClassNames to handle union alias (#477)

Author: ybz1013

dao-api/src/main/java/com/linkedin/metadata/dao/utils/ModelUtils.java

@@ -929,7 +929,9 @@ public static <ASPECT_UNION extends UnionTemplate> List<String> getAspectClassNa
     try {
       final UnionTemplate unionTemplate = unionClass.newInstance();
       final UnionDataSchema unionDataSchema = (UnionDataSchema) unionTemplate.schema();
-      return unionDataSchema.getMembers().stream().map(UnionDataSchema.Member::getUnionMemberKey).collect(Collectors.toList());
+      return unionDataSchema.getMembers().stream().map(
+          member -> member.hasAlias() ? member.getType().getUnionMemberKey() : member.getUnionMemberKey()
+      ).collect(Collectors.toList());
     } catch (Exception e) {
       throw new RuntimeException(e);
     }

dao-api/src/test/java/com/linkedin/metadata/dao/utils/ModelUtilsTest.java

@@ -601,6 +601,13 @@ public void testGetAspectClassNames() {
     assertEquals(classNames.get(1), "com.linkedin.testing.EntityBar");
   }
 
+  @Test
+  public void testGetAspectClassNamesWithAlias() {
+    List<String> classNames = ModelUtils.getAspectClassNames(EntityUnionAlias.class);
+    assertEquals(classNames.get(0), "com.linkedin.testing.EntityFoo");
+    assertEquals(classNames.get(1), "com.linkedin.testing.EntityBar");
+  }
+
   @Test
   public void testGetUnionClassFromSnapshot() {
     Class<UnionTemplate> unionTemplate = ModelUtils.getUnionClassFromSnapshot(EntitySnapshot.class);