use of org.talend.core.model.metadata.MetadataTalendTypeFilter in project tdi-studio-se by Talend.
the class NodeUtilTest method createMetadataTalendTypeFilter.
@Test
public void createMetadataTalendTypeFilter() {
DummyComponent comp = Mockito.mock(DummyComponent.class);
Mockito.when(comp.getType()).thenReturn(ComponentCategory.CATEGORY_4_CAMEL.getName());
//$NON-NLS-1$
INode node = new DataNode(comp, "");
MetadataTalendTypeFilter filter = NodeUtil.createMetadataTalendTypeFilter(node);
assertTrue(filter instanceof DummyMetadataTalendTypeFilter);
Mockito.when(comp.getType()).thenReturn(ComponentCategory.CATEGORY_4_DI.getName());
//$NON-NLS-1$
node = new DataNode(comp, "");
filter = NodeUtil.createMetadataTalendTypeFilter(node);
assertTrue(filter instanceof DummyMetadataTalendTypeFilter);
Mockito.when(comp.getType()).thenReturn(ComponentCategory.CATEGORY_4_MAPREDUCE.getName());
//$NON-NLS-1$
node = new DataNode(comp, "");
filter = NodeUtil.createMetadataTalendTypeFilter(node);
assertTrue(filter instanceof MrMetadataTalendTypeFilter);
Mockito.when(comp.getType()).thenReturn(ComponentCategory.CATEGORY_4_SPARK.getName());
//$NON-NLS-1$
node = new DataNode(comp, "");
filter = NodeUtil.createMetadataTalendTypeFilter(node);
assertTrue(filter instanceof SparkMetadataTalendTypeFilter);
Mockito.when(comp.getType()).thenReturn(ComponentCategory.CATEGORY_4_SPARKSTREAMING.getName());
//$NON-NLS-1$
node = new DataNode(comp, "");
filter = NodeUtil.createMetadataTalendTypeFilter(node);
assertTrue(filter instanceof SparkMetadataTalendTypeFilter);
Mockito.when(comp.getType()).thenReturn(ComponentCategory.CATEGORY_4_STORM.getName());
//$NON-NLS-1$
node = new DataNode(comp, "");
filter = NodeUtil.createMetadataTalendTypeFilter(node);
assertTrue(filter instanceof StormMetadataTalendTypeFilter);
}
Aggregations