feat: add snapshot diff test with object tags

Author: YutaLin

hadoop-ozone/ozone-manager/src/test/java/org/apache/hadoop/ozone/om/snapshot/TestSnapshotDiffManager.java

@@ -1651,4 +1651,95 @@ public void testGetSnapshotDiffReportJob() throws Exception {
       }
     }
   }
+
+  @Test
+  public void testSnapshotDiffReportWithObjectTags() throws IOException {
+    Map<String, String> originalTags = new HashMap<>();
+    originalTags.put("Environment", "Production");
+    originalTags.put("Team", "Backend");
+
+    Map<String, String> modifiedTags = new HashMap<>();
+    modifiedTags.put("Environment", "Staging");
+    modifiedTags.put("Team", "Backend");
+
+    OmKeyInfo keyWithOriginalTags = new OmKeyInfo.Builder()
+        .setObjectID(100)
+        .setParentObjectID(200)
+        .setVolumeName(VOLUME_NAME)
+        .setBucketName(BUCKET_NAME)
+        .setKeyName("key-with-tags")
+        .setReplicationConfig(new ECReplicationConfig(3, 2))
+        .addAllTags(originalTags)
+        .build();
+
+    OmKeyInfo keyWithModifiedTags = new OmKeyInfo.Builder()
+        .setObjectID(100)
+        .setParentObjectID(200)
+        .setVolumeName(VOLUME_NAME)
+        .setBucketName(BUCKET_NAME)
+        .setKeyName("key-with-tags")
+        .setReplicationConfig(new ECReplicationConfig(3, 2))
+        .addAllTags(modifiedTags)
+        .build();
+
+    assertFalse(keyWithOriginalTags.isKeyInfoSame(keyWithModifiedTags,
+            false, false, false, false, true));
+
+    PersistentMap<byte[], byte[]> oldObjectIdKeyMap = new StubbedPersistentMap<>();
+    PersistentMap<byte[], byte[]> newObjectIdKeyMap = new StubbedPersistentMap<>();
+    PersistentMap<byte[], Boolean> objectIdToIsDirMap = new StubbedPersistentMap<>();
+
+    byte[] objectId100 = codecRegistry.asRawData(100L);
+    byte[] keyNameBytes = codecRegistry.asRawData("key-with-tags");
+    oldObjectIdKeyMap.put(objectId100, keyNameBytes);
+    newObjectIdKeyMap.put(objectId100, keyNameBytes);
+    objectIdToIsDirMap.put(objectId100, false);
+
+    Map<String, WithParentObjectId> fromSnapshotTableMap = new HashMap<>();
+    Map<String, WithParentObjectId> toSnapshotTableMap = new HashMap<>();
+    fromSnapshotTableMap.put("key-with-tags", keyWithOriginalTags);
+    toSnapshotTableMap.put("key-with-tags", keyWithModifiedTags);
+
+    try (MockedStatic<SnapshotUtils> mockedSnapshotUtils = mockStatic(SnapshotUtils.class)) {
+      mockedSnapshotUtils.when(() ->
+              SnapshotUtils.isBlockLocationInfoSame(any(OmKeyInfo.class), any(OmKeyInfo.class)))
+          .thenCallRealMethod();
+
+      String tableName = "KeyTable";
+      Table<String, OmKeyInfo> fromSnapshotTable = mock(Table.class);
+      when(fromSnapshotTable.get(anyString())).thenAnswer(i ->
+          fromSnapshotTableMap.get(i.getArgument(0)));
+      when(fromSnapshotTable.getName()).thenReturn(tableName);
+
+      Table<String, OmKeyInfo> toSnapshotTable = mock(Table.class);
+      when(toSnapshotTable.get(anyString())).thenAnswer(i ->
+          toSnapshotTableMap.get(i.getArgument(0)));
+      when(fromSnapshotTable.getName()).thenReturn(tableName);
+
+      Map<String, String> tablePrefixes = mock(Map.class);
+      when(tablePrefixes.get(anyString())).thenReturn("");
+
+      SnapshotDiffManager spy = spy(snapshotDiffManager);
+      doReturn(true).when(spy)
+          .areDiffJobAndSnapshotsActive(VOLUME_NAME, BUCKET_NAME, "fromSnap", "toSnap");
+
+      long totalDiffEntries = spy.generateDiffReport("jobId",
+          fromSnapshotTable, toSnapshotTable, null, null,
+          objectIdToIsDirMap, oldObjectIdKeyMap, newObjectIdKeyMap,
+          VOLUME_NAME, BUCKET_NAME, "fromSnap", "toSnap", false,
+          Optional.empty(), Optional.empty(), tablePrefixes);
+
+      assertEquals(1, totalDiffEntries);
+
+      SnapshotDiffJob snapshotDiffJob = new SnapshotDiffJob(0, "jobId",
+          JobStatus.DONE, VOLUME_NAME, BUCKET_NAME, "fromSnap", "toSnap",
+          false, true, 1, null, 0.0);
+      SnapshotDiffReportOzone diffReport = snapshotDiffManager.createPageResponse(
+          snapshotDiffJob, VOLUME_NAME, BUCKET_NAME, "fromSnap", "toSnap",
+          0, Integer.MAX_VALUE);
+
+      DiffReportEntry entry = diffReport.getDiffList().get(0);
+      assertEquals(SnapshotDiffReport.DiffType.MODIFY, entry.getType());
+    }
+  }
 }