| package org.apache.samoa.topology.impl; |
| |
| /* |
| * #%L |
| * SAMOA |
| * %% |
| * Copyright (C) 2014 - 2015 Apache Software Foundation |
| * %% |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| * #L% |
| */ |
| |
| import static org.junit.Assert.*; |
| import mockit.Mocked; |
| import mockit.Tested; |
| |
| import org.apache.samoa.core.EntranceProcessor; |
| import org.apache.samoa.core.Processor; |
| import org.apache.samoa.topology.EntranceProcessingItem; |
| import org.apache.samoa.topology.ProcessingItem; |
| import org.apache.samoa.topology.Stream; |
| import org.apache.samoa.topology.Topology; |
| import org.apache.samoa.topology.impl.SimpleComponentFactory; |
| import org.apache.samoa.topology.impl.SimpleEntranceProcessingItem; |
| import org.apache.samoa.topology.impl.SimpleProcessingItem; |
| import org.apache.samoa.topology.impl.SimpleStream; |
| import org.apache.samoa.topology.impl.SimpleTopology; |
| import org.junit.Before; |
| import org.junit.Test; |
| |
| /** |
| * @author Anh Thu Vu |
| * |
| */ |
| public class SimpleComponentFactoryTest { |
| |
| @Tested |
| private SimpleComponentFactory factory; |
| @Mocked |
| private Processor processor, processorReplica; |
| @Mocked |
| private EntranceProcessor entranceProcessor; |
| |
| private final int parallelism = 3; |
| private final String topoName = "TestTopology"; |
| |
| @Before |
| public void setUp() throws Exception { |
| factory = new SimpleComponentFactory(); |
| } |
| |
| @Test |
| public void testCreatePiNoParallelism() { |
| ProcessingItem pi = factory.createPi(processor); |
| assertNotNull("ProcessingItem created is null.", pi); |
| assertEquals("ProcessingItem created is not a SimpleProcessingItem.", SimpleProcessingItem.class, pi.getClass()); |
| assertEquals("Parallelism of PI is not 1", 1, pi.getParallelism(), 0); |
| } |
| |
| @Test |
| public void testCreatePiWithParallelism() { |
| ProcessingItem pi = factory.createPi(processor, parallelism); |
| assertNotNull("ProcessingItem created is null.", pi); |
| assertEquals("ProcessingItem created is not a SimpleProcessingItem.", SimpleProcessingItem.class, pi.getClass()); |
| assertEquals("Parallelism of PI is not ", parallelism, pi.getParallelism(), 0); |
| } |
| |
| @Test |
| public void testCreateStream() { |
| ProcessingItem pi = factory.createPi(processor); |
| |
| Stream stream = factory.createStream(pi); |
| assertNotNull("Stream created is null", stream); |
| assertEquals("Stream created is not a SimpleStream.", SimpleStream.class, stream.getClass()); |
| } |
| |
| @Test |
| public void testCreateTopology() { |
| Topology topology = factory.createTopology(topoName); |
| assertNotNull("Topology created is null.", topology); |
| assertEquals("Topology created is not a SimpleTopology.", SimpleTopology.class, topology.getClass()); |
| } |
| |
| @Test |
| public void testCreateEntrancePi() { |
| EntranceProcessingItem entrancePi = factory.createEntrancePi(entranceProcessor); |
| assertNotNull("EntranceProcessingItem created is null.", entrancePi); |
| assertEquals("EntranceProcessingItem created is not a SimpleEntranceProcessingItem.", |
| SimpleEntranceProcessingItem.class, entrancePi.getClass()); |
| assertSame("EntranceProcessor is not set correctly.", entranceProcessor, entrancePi.getProcessor()); |
| } |
| |
| } |