@Test
  public void testRemoveBootstrapSession() {
    // Arrange
    reset();

    bootstrapSessionManagerWithCompactionHandler.addAndStartBootstrapSession(mockPartitionId, "snapshot2", "node2");

    // Act
    bootstrapSessionManagerWithCompactionHandler.removeBootstrapSession(mockPartitionId, "node2");

    // Assert
    assertNull(bootstrapSessionManagerWithCompactionHandler.getBootstrapSession(mockPartitionId, "node2"));
    assertTrue("Compaction should be re-enabled after session removal", compactionEnabled.get());
  }
Adds & removes a bootstrap session and checks that the session is removed correctly.
ambry-store/src/test/java/com/github/ambry/store/BootstrapSessionManagerTest.java