@Before public void setUp() throws Exception { // Build our test packet IDebugCounterService debugCounter = new DebugCounterServiceImpl(); switchManager = createMock(IOFSwitchManager.class); SwitchManagerCounters counters = new SwitchManagerCounters(debugCounter); expect(switchManager.getCounters()).andReturn(counters).anyTimes(); replay(switchManager); factory = OFFactories.getFactory(OFVersion.OF_13); testMessage = factory.buildRoleReply() .setXid(1) .setRole(OFControllerRole.ROLE_MASTER) .build(); IOFConnectionBackend conn = EasyMock.createNiceMock(IOFConnectionBackend.class); capturedMessage = new Capture<OFMessage>(); conn.write(EasyMock.capture(capturedMessage)); expectLastCall().anyTimes(); expect(conn.getOFFactory()).andReturn(factory).anyTimes(); expect(conn.getAuxId()).andReturn(OFAuxId.MAIN).anyTimes(); EasyMock.replay(conn); IOFConnectionBackend auxConn = EasyMock.createNiceMock(IOFConnectionBackend.class); expect(auxConn.getOFFactory()).andReturn(factory).anyTimes(); expect(auxConn.getAuxId()).andReturn(OFAuxId.of(1)).anyTimes(); EasyMock.replay(auxConn); sw = new OFSwitchTest(conn, switchManager); sw.registerConnection(auxConn); sw.setControllerRole(OFControllerRole.ROLE_MASTER); /* must supply role now, otherwise write() will be blocked if not master/equal/other */ switches = new ConcurrentHashMap<DatapathId, IOFSwitchBackend>(); switches.put(sw.getId(), sw); reset(switchManager); //expect(switchManager.getSwitch(sw.getId())).andReturn(sw).anyTimes(); }
@Before public void setUp() throws Exception { // Build our test packet IDebugCounterService debugCounter = new DebugCounterServiceImpl(); switchManager = createMock(IOFSwitchManager.class); SwitchManagerCounters counters = new SwitchManagerCounters(debugCounter); expect(switchManager.getCounters()).andReturn(counters).anyTimes(); replay(switchManager); factory = OFFactories.getFactory(OFVersion.OF_13); testMessage = factory.buildRoleReply() .setXid(1) .setRole(OFControllerRole.ROLE_MASTER) .build(); IOFConnectionBackend conn = EasyMock.createNiceMock(IOFConnectionBackend.class); capturedMessage = new Capture<OFMessage>(); conn.write(EasyMock.capture(capturedMessage)); expectLastCall().anyTimes(); expect(conn.getOFFactory()).andReturn(factory).anyTimes(); expect(conn.getAuxId()).andReturn(OFAuxId.MAIN).anyTimes(); EasyMock.replay(conn); IOFConnectionBackend auxConn = EasyMock.createNiceMock(IOFConnectionBackend.class); expect(auxConn.getOFFactory()).andReturn(factory).anyTimes(); expect(auxConn.getAuxId()).andReturn(OFAuxId.of(1)).anyTimes(); EasyMock.replay(auxConn); sw = new OFSwitchTest(conn, switchManager); sw.registerConnection(auxConn); switches = new ConcurrentHashMap<DatapathId, IOFSwitchBackend>(); switches.put(sw.getId(), sw); reset(switchManager); //expect(switchManager.getSwitch(sw.getId())).andReturn(sw).anyTimes(); }
public MockSwitchManager(){ switchHandlers = new ConcurrentHashMap<DatapathId, OFSwitchHandshakeHandler>(); switches = new ConcurrentHashMap<DatapathId, IOFSwitch>(); counters = new SwitchManagerCounters(new DebugCounterServiceImpl()); //switchListeners = new CopyOnWriteArrayList<IOFSwitchListener>(); }
@Override public SwitchManagerCounters getCounters() { return this.counters; }
@Before public void setUp() throws Exception { // Build our test packet IDebugCounterService debugCounter = new DebugCounterServiceImpl(); switchManager = createMock(IOFSwitchManager.class); SwitchManagerCounters counters = new SwitchManagerCounters(debugCounter); expect(switchManager.getCounters()).andReturn(counters).anyTimes(); replay(switchManager); factory = OFFactories.getFactory(OFVersion.OF_13); testMessage = factory.buildRoleReply() .setXid(1) .setRole(OFControllerRole.ROLE_MASTER) .build(); IOFConnectionBackend conn = EasyMock.createNiceMock(IOFConnectionBackend.class); capturedMessage = new Capture<Iterable<OFMessage>>(); expect(conn.write(EasyMock.capture(capturedMessage))).andReturn(Collections.<OFMessage>emptyList()).atLeastOnce(); expect(conn.getOFFactory()).andReturn(factory).anyTimes(); expect(conn.getAuxId()).andReturn(OFAuxId.MAIN).anyTimes(); EasyMock.replay(conn); IOFConnectionBackend auxConn = EasyMock.createNiceMock(IOFConnectionBackend.class); expect(auxConn.getOFFactory()).andReturn(factory).anyTimes(); expect(auxConn.getAuxId()).andReturn(OFAuxId.of(1)).anyTimes(); expect(auxConn.write(EasyMock.capture(capturedMessage))).andReturn(Collections.<OFMessage>emptyList()).once(); EasyMock.replay(auxConn); sw = new OFSwitchTest(conn, switchManager); sw.registerConnection(auxConn); sw.setControllerRole(OFControllerRole.ROLE_MASTER); /* must supply role now, otherwise write() will be blocked if not master/equal/other */ switches = new ConcurrentHashMap<DatapathId, IOFSwitchBackend>(); switches.put(sw.getId(), sw); reset(switchManager); //expect(switchManager.getSwitch(sw.getId())).andReturn(sw).anyTimes(); setUpPorts(); }
@Before public void setUp() throws Exception { // Build our test packet IDebugCounterService debugCounter = new DebugCounterServiceImpl(); switchManager = createMock(IOFSwitchManager.class); SwitchManagerCounters counters = new SwitchManagerCounters(debugCounter); expect(switchManager.getCounters()).andReturn(counters).anyTimes(); replay(switchManager); factory = OFFactories.getFactory(OFVersion.OF_13); testMessage = factory.buildRoleReply() .setXid(1) .setRole(OFControllerRole.ROLE_MASTER) .build(); IOFConnectionBackend conn = EasyMock.createNiceMock(IOFConnectionBackend.class); capturedMessage = new Capture<OFMessage>(); conn.write(EasyMock.capture(capturedMessage)); expectLastCall().anyTimes(); expect(conn.getOFFactory()).andReturn(factory).anyTimes(); expect(conn.getAuxId()).andReturn(OFAuxId.MAIN).anyTimes(); EasyMock.replay(conn); IOFConnectionBackend auxConn = EasyMock.createNiceMock(IOFConnectionBackend.class); expect(auxConn.getOFFactory()).andReturn(factory).anyTimes(); expect(auxConn.getAuxId()).andReturn(OFAuxId.of(1)).anyTimes(); EasyMock.replay(auxConn); sw = new OFSwitchTest(conn, switchManager); sw.registerConnection(auxConn); sw.setControllerRole(OFControllerRole.ROLE_MASTER); /* must supply role now, otherwise write() will be blocked if not master/equal/other */ switches = new ConcurrentHashMap<DatapathId, IOFSwitchBackend>(); switches.put(sw.getId(), sw); reset(switchManager); //expect(switchManager.getSwitch(sw.getId())).andReturn(sw).anyTimes(); setUpPorts(); }