src/test/java/net/onrc/onos/core/newintent/IntentCompilerTest.java

package net.onrc.onos.core.newintent;

import net.onrc.onos.api.flowmanager.FlowId;
import net.onrc.onos.api.flowmanager.FlowIdGenerator;
import net.onrc.onos.api.newintent.IntentId;
import net.onrc.onos.api.newintent.IntentIdGenerator;
import net.onrc.onos.core.topology.BaseTopology;
import net.onrc.onos.core.topology.ITopologyService;
import net.onrc.onos.core.topology.MockTopology;
import net.onrc.onos.core.topology.MutableTopology;
import net.onrc.onos.core.util.Dpid;
import net.onrc.onos.core.util.PortNumber;
import org.junit.Before;

import static org.easymock.EasyMock.createMock;
import static org.easymock.EasyMock.expect;
import static org.easymock.EasyMock.replay;

/**
 * A base test class for intent compiler aggregating
 * common set up logic and variables.
 */
public class IntentCompilerTest {
    protected final IntentId intentId1 = new IntentId(1L);
    protected final IntentId intentId2 = new IntentId(2L);
    protected final FlowId flowId = new FlowId(1L);
    protected final Dpid dpid1 = new Dpid(1L);
    protected final Dpid dpid2 = new Dpid(2L);
    protected final Dpid dpid3 = new Dpid(3L);
    protected final Dpid dpid4 = new Dpid(4L);
    protected final PortNumber port1h = PortNumber.uint32(15);
    protected final PortNumber port12 = PortNumber.uint32(12);
    protected final PortNumber port14 = PortNumber.uint32(14);
    protected final PortNumber port3h = PortNumber.uint32(35);
    protected final PortNumber port32 = PortNumber.uint32(32);
    protected final PortNumber port21 = PortNumber.uint32(21);
    protected final PortNumber port23 = PortNumber.uint32(23);
    protected final PortNumber port41 = PortNumber.uint32(41);
    protected final PortNumber port43 = PortNumber.uint32(43);
    protected IntentIdGenerator intentIdGenerator;
    protected FlowIdGenerator flowIdGenerator;
    protected ITopologyService topologyService;

    @Before
    public void commonSetUp() {
        intentIdGenerator = createMock(IntentIdGenerator.class);
        flowIdGenerator = createMock(FlowIdGenerator.class);
        topologyService = createMock(ITopologyService.class);

        // configure mocks
        expect(intentIdGenerator.getNewId())
                .andReturn(intentId1)
                .andReturn(intentId2);
        expect(flowIdGenerator.getNewId()).andReturn(flowId);
        expect(topologyService.getTopology()).andReturn((MutableTopology) createFakeTopology());
        replay(intentIdGenerator, flowIdGenerator, topologyService);
    }

    protected BaseTopology createFakeTopology() {
        MockTopology mock = new MockTopology();
        mock.createSampleTopology2();

        return mock;
    }
}