use of com.sequenceiq.cloudbreak.cloud.aws.common.subnetselector.SubnetFilterStrategy in project cloudbreak by hortonworks.
the class AwsNetworkConnectorTest method prepareMock.
public void prepareMock(List<CloudSubnet> cloudSubnets) {
Map<SubnetFilterStrategyType, SubnetFilterStrategy> subnetFilterStrategyMap = new HashMap<>();
subnetFilterStrategyMap.put(SubnetFilterStrategyType.MULTIPLE_PREFER_PRIVATE, subnetFilterStrategy);
subnetFilterStrategyMap.put(SubnetFilterStrategyType.MULTIPLE_PREFER_PUBLIC, subnetFilterStrategy);
when(subnetFilterStrategy.filter(any(), anyInt())).thenReturn(new SubnetSelectionResult(cloudSubnets));
ReflectionTestUtils.setField(underTest, "subnetFilterStrategyMap", subnetFilterStrategyMap);
}
Aggregations