jojochuang commented on code in PR #8551: URL: https://github.com/apache/ozone/pull/8551#discussion_r2126902377
########## hadoop-ozone/ozone-manager/src/test/java/org/apache/hadoop/ozone/om/snapshot/filter/AbstractReclaimableFilterTest.java: ########## @@ -159,19 +161,28 @@ protected void teardown() throws IOException { private void mockOzoneManager(BucketLayout bucketLayout) throws IOException { OMMetadataManager metadataManager = mock(OMMetadataManager.class); + BucketManager bucketManager = mock(BucketManager.class); when(ozoneManager.getMetadataManager()).thenReturn(metadataManager); + when(ozoneManager.getBucketManager()).thenReturn(bucketManager); long volumeCount = 0; - long bucketCount = 0; for (String volume : volumes) { when(metadataManager.getVolumeId(eq(volume))).thenReturn(volumeCount); - for (String bucket : buckets) { - when(ozoneManager.getBucketInfo(eq(volume), eq(bucket))) - .thenReturn(OmBucketInfo.newBuilder().setVolumeName(volume).setBucketName(bucket) - .setObjectID(bucketCount).setBucketLayout(bucketLayout).build()); - bucketCount++; - } volumeCount++; } + + when(bucketManager.getBucketInfo(anyString(), anyString())).thenAnswer(i -> { + String volume = i.getArgument(0, String.class); + String bucket = i.getArgument(1, String.class); + if (!volumes.contains(volume)) { + throw new OMException("Volume " + volume + " already exists", OMException.ResultCodes.VOLUME_NOT_FOUND); Review Comment: Looks like the exception message needs to update to " not found". See Gemini's review comment: https://github.com/jojochuang/ozone/pull/568#discussion_r2125748884 -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: issues-unsubscr...@ozone.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: issues-unsubscr...@ozone.apache.org For additional commands, e-mail: issues-h...@ozone.apache.org