fix

Author: JingsongLi

paimon-core/src/test/java/org/apache/paimon/table/SimpleTableTestBase.java

@@ -1848,7 +1848,7 @@ protected void assertRollbackTo(
     @Test
     public void testRollbackSchemaSuccess() throws Exception {
         FileStoreTable table = createFileStoreTable();
-        SchemaManager schemaManager = new SchemaManager(LocalFileIO.create(), tablePath);
+        SchemaManager schemaManager = table.schemaManager();
         long firstSchemaId = schemaManager.latest().get().id();
 
         // evolve schema twice
@@ -1867,7 +1867,7 @@ public void testRollbackSchemaSuccess() throws Exception {
     @Test
     public void testRollbackSchemaFailedWithSnapshotReference() throws Exception {
         FileStoreTable table = createFileStoreTable();
-        SchemaManager schemaManager = new SchemaManager(LocalFileIO.create(), tablePath);
+        SchemaManager schemaManager = table.schemaManager();
         long firstSchemaId = schemaManager.latest().get().id();
 
         // write data to create a snapshot referencing firstSchemaId
@@ -1880,10 +1880,9 @@ public void testRollbackSchemaFailedWithSnapshotReference() throws Exception {
 
         // evolve schema
         schemaManager.commitChanges(SchemaChange.setOption("aa", "bb"));
-        long secondSchemaId = schemaManager.latest().get().id();
 
         // write data to create a snapshot referencing secondSchemaId
-        table = createFileStoreTable();
+        table = table.copyWithLatestSchema();
         write = table.newWrite(commitUser);
         commit = table.newCommit(commitUser);
         write.write(rowData(1, 10, 100L));
@@ -1894,7 +1893,6 @@ public void testRollbackSchemaFailedWithSnapshotReference() throws Exception {
         // rollback should fail because snapshot references secondSchemaId
         FileStoreTable finalTable = table;
         assertThatThrownBy(() -> finalTable.rollbackSchema(firstSchemaId))
-                .hasMessageContaining("Cannot rollback to schema " + firstSchemaId)
-                .hasMessageContaining("schema " + secondSchemaId + " is still referenced");
+                .hasMessageContaining("Cannot rollback to schema");
     }
 }