@@ -422,7 +422,7 @@ public void testRemoveSnapshotFromJson() {
422422 String action = MetadataUpdateParser .REMOVE_SNAPSHOTS ;
423423 long snapshotId = 2L ;
424424 String json = String .format ("{\" action\" :\" %s\" ,\" snapshot-ids\" :[2]}" , action );
425- MetadataUpdate expected = new MetadataUpdate .RemoveSnapshot (snapshotId );
425+ MetadataUpdate expected = new MetadataUpdate .RemoveSnapshots (snapshotId );
426426 assertEquals (action , expected , MetadataUpdateParser .fromJson (json ));
427427 }
428428
@@ -431,7 +431,7 @@ public void testRemoveSnapshotToJson() {
431431 String action = MetadataUpdateParser .REMOVE_SNAPSHOTS ;
432432 long snapshotId = 2L ;
433433 String expected = String .format ("{\" action\" :\" %s\" ,\" snapshot-ids\" :[2]}" , action );
434- MetadataUpdate update = new MetadataUpdate .RemoveSnapshot (snapshotId );
434+ MetadataUpdate update = new MetadataUpdate .RemoveSnapshots (snapshotId );
435435 String actual = MetadataUpdateParser .toJson (update );
436436 assertThat (actual )
437437 .as ("Remove snapshots should serialize to the correct JSON value" )
@@ -1030,15 +1030,9 @@ public void assertEquals(
10301030 (MetadataUpdate .AddSnapshot ) expectedUpdate , (MetadataUpdate .AddSnapshot ) actualUpdate );
10311031 break ;
10321032 case MetadataUpdateParser .REMOVE_SNAPSHOTS :
1033- if (actualUpdate instanceof MetadataUpdate .RemoveSnapshot ) {
1034- assertEqualsRemoveSnapshot (
1035- (MetadataUpdate .RemoveSnapshot ) expectedUpdate ,
1036- (MetadataUpdate .RemoveSnapshot ) actualUpdate );
1037- } else {
1038- assertEqualsRemoveSnapshots (
1039- (MetadataUpdate .RemoveSnapshots ) expectedUpdate ,
1040- (MetadataUpdate .RemoveSnapshots ) actualUpdate );
1041- }
1033+ assertEqualsRemoveSnapshot (
1034+ (MetadataUpdate .RemoveSnapshots ) expectedUpdate ,
1035+ (MetadataUpdate .RemoveSnapshots ) actualUpdate );
10421036 break ;
10431037 case MetadataUpdateParser .REMOVE_SNAPSHOT_REF :
10441038 assertEqualsRemoveSnapshotRef (
@@ -1240,10 +1234,10 @@ private static void assertEqualsAddSnapshot(
12401234 }
12411235
12421236 private static void assertEqualsRemoveSnapshot (
1243- MetadataUpdate .RemoveSnapshot expected , MetadataUpdate .RemoveSnapshot actual ) {
1244- assertThat (actual .snapshotId ())
1237+ MetadataUpdate .RemoveSnapshots expected , MetadataUpdate .RemoveSnapshots actual ) {
1238+ assertThat (actual .snapshotIds ())
12451239 .as ("Snapshot to remove should be the same" )
1246- .isEqualTo (expected .snapshotId ());
1240+ .isEqualTo (expected .snapshotIds ());
12471241 }
12481242
12491243 private static void assertEqualsRemoveSnapshots (
0 commit comments