How to use

getStorageUnitByStorageName

method

in

org.finra.herd.service.helper.BusinessObjectDataHelper

@Test
public void getStorageUnitByStorageNameStorageUnitNoExists()
{
  String testStorageName = "I_DO_NOT_EXIST";
  // Try to get a non-existing storage unit.
  try
  {
    businessObjectDataHelper.getStorageUnitByStorageName(
      new BusinessObjectData(INTEGER_VALUE, BDEF_NAMESPACE, BDEF_NAME, FORMAT_USAGE_CODE, FORMAT_FILE_TYPE_CODE, FORMAT_VERSION, PARTITION_KEY,
        PARTITION_VALUE, SUBPARTITION_VALUES, DATA_VERSION, LATEST_VERSION_FLAG_SET, BusinessObjectDataStatusEntity.VALID, NO_STORAGE_UNITS,
        NO_ATTRIBUTES, NO_BUSINESS_OBJECT_DATA_PARENTS, NO_BUSINESS_OBJECT_DATA_CHILDREN, NO_BUSINESS_OBJECT_DATA_STATUS_HISTORY,
        NO_RETENTION_EXPIRATION_DATE), testStorageName);
    fail("Should throw a IllegalStateException when storage unit does not exist.");
  }
  catch (IllegalStateException e)
  {
    assertEquals(String.format("Business object data has no storage unit with storage name \"%s\".", testStorageName), e.getMessage());
  }
}

StorageUnit storageUnit = businessObjectDataHelper.getStorageUnitByStorageName(businessObjectData, storageName);

@Test
public void getStorageUnitByStorageName()
{
  // Create business object data with several test storage units.
  BusinessObjectData businessObjectData = new BusinessObjectData();
  List<String> testStorageNames = Arrays.asList("Storage_1", "storage-2", "STORAGE3");
  List<StorageUnit> storageUnits = new ArrayList<>();
  businessObjectData.setStorageUnits(storageUnits);
  for (String testStorageName : testStorageNames)
  {
    StorageUnit storageUnit = new StorageUnit();
    storageUnits.add(storageUnit);
    Storage storage = new Storage();
    storageUnit.setStorage(storage);
    storage.setName(testStorageName);
  }
  // Validate that we can find all storage units regardless of the storage name case.
  for (String testStorageName : testStorageNames)
  {
    assertEquals(testStorageName, businessObjectDataHelper.getStorageUnitByStorageName(businessObjectData, testStorageName).getStorage().getName());
    assertEquals(testStorageName,
      businessObjectDataHelper.getStorageUnitByStorageName(businessObjectData, testStorageName.toUpperCase()).getStorage().getName());
    assertEquals(testStorageName,
      businessObjectDataHelper.getStorageUnitByStorageName(businessObjectData, testStorageName.toLowerCase()).getStorage().getName());
  }
}

when(mockBusinessObjectDataHelper.getStorageUnitByStorageName(any(), any())).thenReturn(storageUnit);
when(mockDownloaderWebClient.getS3KeyPrefix(any())).thenReturn(s3KeyPrefixInformation);
when(mockDownloaderWebClient.getStorageUnitDownloadCredential(any(), any())).thenReturn(new StorageUnitDownloadCredential(

when(mockBusinessObjectDataHelper.getStorageUnitByStorageName(any(), any())).thenReturn(storageUnit);
when(mockDownloaderWebClient.getS3KeyPrefix(any())).thenReturn(s3KeyPrefixInformation);
when(mockS3Service.downloadDirectory(any())).then(new Answer<S3FileTransferResultsDto>()

when(mockBusinessObjectDataHelper.getStorageUnitByStorageName(any(), any())).thenReturn(storageUnit);
when(mockDownloaderWebClient.getS3KeyPrefix(any())).thenReturn(s3KeyPrefixInformation);
when(mockS3Service.downloadDirectory(any())).then(new Answer<S3FileTransferResultsDto>()