use of org.apache.heron.eco.definition.EcoTopologyDefinition in project heron by twitter.
the class EcoTest method testSubmit_StormTopologyType_BehavesAsExpected.
@Test
public void testSubmit_StormTopologyType_BehavesAsExpected() throws Exception {
FileInputStream mockStream = PowerMockito.mock(FileInputStream.class);
FileInputStream mockPropsStream = PowerMockito.mock(FileInputStream.class);
final String topologyName = "the name";
EcoTopologyDefinition topologyDefinition = new EcoTopologyDefinition();
topologyDefinition.setName(topologyName);
when(mockEcoParser.parseFromInputStream(eq(mockStream), eq(mockPropsStream), eq(false))).thenReturn(topologyDefinition);
subject.submit(mockStream, mockPropsStream, false);
verify(mockEcoParser).parseFromInputStream(same(mockStream), same(mockPropsStream), eq(false));
verify(mockEcoSubmitter).submitStormTopology(any(String.class), any(Config.class), any(StormTopology.class));
}
use of org.apache.heron.eco.definition.EcoTopologyDefinition in project heron by twitter.
the class EcoTest method testSubmit_HeronTopologyType_BehavesAsExpected.
@Test
public void testSubmit_HeronTopologyType_BehavesAsExpected() throws Exception {
FileInputStream mockStream = PowerMockito.mock(FileInputStream.class);
FileInputStream mockPropsStream = PowerMockito.mock(FileInputStream.class);
final String topologyName = "the name";
EcoTopologyDefinition topologyDefinition = new EcoTopologyDefinition();
topologyDefinition.setName(topologyName);
topologyDefinition.setType("heron");
when(mockEcoParser.parseFromInputStream(eq(mockStream), eq(mockPropsStream), eq(false))).thenReturn(topologyDefinition);
subject.submit(mockStream, mockPropsStream, false);
verify(mockEcoParser).parseFromInputStream(same(mockStream), same(mockPropsStream), eq(false));
verify(mockEcoSubmitter).submitHeronTopology(any(String.class), any(Config.class), any(HeronTopology.class));
}
use of org.apache.heron.eco.definition.EcoTopologyDefinition in project heron by twitter.
the class ConfigBuilderTest method testBuildConfig_MBAllocationTooSmall_ExceptionThrown.
@Test(expected = IllegalArgumentException.class)
public void testBuildConfig_MBAllocationTooSmall_ExceptionThrown() throws Exception {
Config config = null;
try {
EcoParser ecoParser = new EcoParser();
InputStream inputStream = new ByteArrayInputStream(INCORRECT_RAM_MEGABYTES.getBytes());
FileInputStream mockPropsStream = PowerMockito.mock(FileInputStream.class);
EcoTopologyDefinition ecoTopologyDefinition = ecoParser.parseFromInputStream(inputStream, mockPropsStream, false);
config = subject.buildConfig(ecoTopologyDefinition);
} finally {
assertNull(config);
}
}
use of org.apache.heron.eco.definition.EcoTopologyDefinition in project heron by twitter.
the class ConfigBuilderTest method testBuildConfig_ByteAllocationTooSmall_ExceptionThrown.
@Test(expected = IllegalArgumentException.class)
public void testBuildConfig_ByteAllocationTooSmall_ExceptionThrown() throws Exception {
Config config = null;
try {
EcoParser ecoParser = new EcoParser();
InputStream inputStream = new ByteArrayInputStream(INCORRECT_RAM_BYTES.getBytes());
FileInputStream mockPropsStream = PowerMockito.mock(FileInputStream.class);
EcoTopologyDefinition ecoTopologyDefinition = ecoParser.parseFromInputStream(inputStream, mockPropsStream, false);
config = subject.buildConfig(ecoTopologyDefinition);
} finally {
assertNull(config);
}
}
use of org.apache.heron.eco.definition.EcoTopologyDefinition in project heron by twitter.
the class ConfigBuilderTest method testBuildConfig_IncorrectByteResourceFormat_ExceptionThrow.
@Test(expected = IllegalArgumentException.class)
public void testBuildConfig_IncorrectByteResourceFormat_ExceptionThrow() throws Exception {
Config config = null;
try {
EcoParser ecoParser = new EcoParser();
InputStream inputStream = new ByteArrayInputStream(INCORRECT_BYTES_FORMAT_YAML.getBytes());
FileInputStream mockPropsStream = PowerMockito.mock(FileInputStream.class);
EcoTopologyDefinition ecoTopologyDefinition = ecoParser.parseFromInputStream(inputStream, mockPropsStream, false);
config = subject.buildConfig(ecoTopologyDefinition);
} finally {
assertNull(config);
}
}
Aggregations