use of org.finra.herd.model.api.xml.EmrClusterDefinitionEbsConfiguration in project herd by FINRAOS.
the class EmrDaoImplTest method testGetInstanceTypeConfigs.
@Test
public void testGetInstanceTypeConfigs() {
// Create objects required for testing.
final String instanceType = STRING_VALUE;
final Integer weightedCapacity = INTEGER_VALUE;
final String bidPrice = STRING_VALUE_2;
final Double bidPriceAsPercentageOfOnDemandPrice = DOUBLE_VALUE;
final EmrClusterDefinitionEbsConfiguration emrClusterDefinitionEbsConfiguration = null;
final List<EmrClusterDefinitionConfiguration> emrClusterDefinitionConfigurations = null;
final EmrClusterDefinitionInstanceTypeConfig emrClusterDefinitionInstanceTypeConfig = new EmrClusterDefinitionInstanceTypeConfig(instanceType, weightedCapacity, bidPrice, bidPriceAsPercentageOfOnDemandPrice, emrClusterDefinitionEbsConfiguration, emrClusterDefinitionConfigurations);
// Call the method under test.
List<InstanceTypeConfig> result = emrDaoImpl.getInstanceTypeConfigs(Arrays.asList(emrClusterDefinitionInstanceTypeConfig));
// Verify the external calls.
verifyNoMoreInteractionsHelper();
// Validate the results.
final List<Configuration> expectedConfigurations = null;
assertEquals(Arrays.asList(new InstanceTypeConfig().withInstanceType(instanceType).withWeightedCapacity(weightedCapacity).withBidPrice(bidPrice).withBidPriceAsPercentageOfOnDemandPrice(bidPriceAsPercentageOfOnDemandPrice).withEbsConfiguration(null).withConfigurations(expectedConfigurations)), result);
}
use of org.finra.herd.model.api.xml.EmrClusterDefinitionEbsConfiguration in project herd by FINRAOS.
the class EmrDaoImpl method getEbsConfiguration.
/**
* Creates an instance of {@link EbsConfiguration} from a given instance of {@link EmrClusterDefinitionEbsConfiguration}.
*
* @param emrClusterDefinitionEbsConfiguration the instance of {@link EmrClusterDefinitionEbsConfiguration}
*
* @return the instance of {@link EbsConfiguration}
*/
protected EbsConfiguration getEbsConfiguration(EmrClusterDefinitionEbsConfiguration emrClusterDefinitionEbsConfiguration) {
EbsConfiguration ebsConfiguration = null;
if (emrClusterDefinitionEbsConfiguration != null) {
ebsConfiguration = new EbsConfiguration();
ebsConfiguration.setEbsBlockDeviceConfigs(getEbsBlockDeviceConfigs(emrClusterDefinitionEbsConfiguration.getEbsBlockDeviceConfigs()));
ebsConfiguration.setEbsOptimized(emrClusterDefinitionEbsConfiguration.isEbsOptimized());
}
return ebsConfiguration;
}
use of org.finra.herd.model.api.xml.EmrClusterDefinitionEbsConfiguration in project herd by FINRAOS.
the class EmrDaoImplTest method testGetInstanceGroupConfig.
@Test
public void testGetInstanceGroupConfig() throws Exception {
// Call the method under test.
InstanceGroupConfig result = emrDaoImpl.getInstanceGroupConfig(InstanceRoleType.MASTER, EC2_INSTANCE_TYPE, INSTANCE_COUNT, BID_PRICE, new EmrClusterDefinitionEbsConfiguration(Collections.singletonList(new EmrClusterDefinitionEbsBlockDeviceConfig(new EmrClusterDefinitionVolumeSpecification(VOLUME_TYPE, IOPS, SIZE_IN_GB), VOLUMES_PER_INSTANCE)), EBS_OPTIMIZED));
// Verify the external calls.
verifyNoMoreInteractionsHelper();
// Validate the results.
assertEquals(new InstanceGroupConfig(InstanceRoleType.MASTER, EC2_INSTANCE_TYPE, INSTANCE_COUNT).withMarket(MarketType.SPOT).withBidPrice(BID_PRICE.toPlainString()).withEbsConfiguration(new EbsConfiguration().withEbsBlockDeviceConfigs(new EbsBlockDeviceConfig().withVolumeSpecification(new VolumeSpecification().withVolumeType(VOLUME_TYPE).withIops(IOPS).withSizeInGB(SIZE_IN_GB)).withVolumesPerInstance(VOLUMES_PER_INSTANCE)).withEbsOptimized(EBS_OPTIMIZED)), result);
}
use of org.finra.herd.model.api.xml.EmrClusterDefinitionEbsConfiguration in project herd by FINRAOS.
the class EmrDaoImplTest method testGetEbsConfiguration.
@Test
public void testGetEbsConfiguration() {
// Call the method under test.
EbsConfiguration result = emrDaoImpl.getEbsConfiguration(new EmrClusterDefinitionEbsConfiguration(Collections.singletonList(new EmrClusterDefinitionEbsBlockDeviceConfig(new EmrClusterDefinitionVolumeSpecification(VOLUME_TYPE, IOPS, SIZE_IN_GB), VOLUMES_PER_INSTANCE)), EBS_OPTIMIZED));
// Verify the external calls.
verifyNoMoreInteractionsHelper();
// Validate the results.
assertEquals(new EbsConfiguration().withEbsBlockDeviceConfigs(new EbsBlockDeviceConfig().withVolumeSpecification(new VolumeSpecification().withVolumeType(VOLUME_TYPE).withIops(IOPS).withSizeInGB(SIZE_IN_GB)).withVolumesPerInstance(VOLUMES_PER_INSTANCE)).withEbsOptimized(EBS_OPTIMIZED), result);
}
Aggregations