@Test
public void testDiskSizeIsChangedToMin() {

    JobDescriptor<BatchJobExt> jobDescriptor = newJobDescriptorWithDiskSize(100);
    when(configuration.getMinDiskSizeMB()).thenReturn(MIN_DISK_SIZE);
    when(entitySanitizer.sanitize(any())).thenReturn(Optional.of(jobDescriptor));
    ExtendedJobSanitizer sanitizer = new ExtendedJobSanitizer(configuration, jobAssertions, entitySanitizer, disruptionBudgetSanitizer, jd -> false, jd -> false, titusRuntime);
    Optional<JobDescriptor<BatchJobExt>> sanitizedJobDescriptorOpt = sanitizer.sanitize(jobDescriptor);
    JobDescriptor<BatchJobExt> sanitizedJobDescriptor = sanitizedJobDescriptorOpt.get();
    assertThat(sanitizedJobDescriptor).isNotNull();
    assertThat(sanitizedJobDescriptor.getContainer().getContainerResources().getDiskMB()).isEqualTo(MIN_DISK_SIZE);
    String nonCompliant = sanitizedJobDescriptor.getAttributes().get(TITUS_NON_COMPLIANT_FEATURES);
    assertThat(nonCompliant).contains(FeatureRolloutPlans.MIN_DISK_SIZE_STRICT_VALIDATION_FEATURE);
}