Controller implements IFloodlightProviderService, 
            IStorageSourceListener { @Override public void setRole(Role role) { if (role == null) throw new NullPointerException("Role can not be null."); if (role == Role.MASTER && this.role == Role.SLAVE) { updateAllInactiveSwitchInfo(); } synchronized(roleChanger) { if (role.equals(this.role)) { log.debug("Ignoring role change: role is already {}", role); return; } Role oldRole = this.role; this.role = role; log.debug("Submitting role change request to role {}", role); roleChanger.submitRequest(connectedSwitches, role); try { this.updates.put(new HARoleUpdate(role, oldRole)); } catch (InterruptedException e) { log.error("Failure adding update to queue", e); } } }  void setStorageSourceService(IStorageSourceService storageSource); void setCounterStore(ICounterStoreService counterStore); void setPktInProcessingService(IPktInProcessingTimeService pits); void setRestApiService(IRestApiService restApi); void setThreadPoolService(IThreadPoolService tp); @Override Role getRole(); @Override void setRole(Role role); @Override synchronized void addOFMessageListener(OFType type, 
                                                  IOFMessageListener listener); @Override synchronized void removeOFMessageListener(OFType type,
                                                     IOFMessageListener listener); void removeOFMessageListeners(OFType type); @Override Map<Long, IOFSwitch> getSwitches(); @Override void addOFSwitchListener(IOFSwitchListener listener); @Override void removeOFSwitchListener(IOFSwitchListener listener); @Override Map<OFType, List<IOFMessageListener>> getListeners(); @Override @LogMessageDocs({ @LogMessageDoc(message="Failed to inject OFMessage {message} onto " + "a null switch", explanation="Failed to process a message because the switch " + " is no longer connected."), @LogMessageDoc(level="ERROR", message="Error reinjecting OFMessage on switch {switch}", explanation="An I/O error occured while attempting to " + "process an OpenFlow message", recommendation=LogMessageDoc.CHECK_SWITCH) }) boolean injectOfMessage(IOFSwitch sw, OFMessage msg,
                                   FloodlightContext bc); @Override @LogMessageDoc(message="Calling System.exit", explanation="The controller is terminating") synchronized void terminate(); @Override boolean injectOfMessage(IOFSwitch sw, OFMessage msg); @Override void handleOutgoingMessage(IOFSwitch sw, OFMessage m,
                                      FloodlightContext bc); @Override BasicFactory getOFMessageFactory(); @Override String getControllerId(); @LogMessageDocs({ @LogMessageDoc(message="Listening for switch connections on {address}", explanation="The controller is ready and listening for new" + " switch connections"), @LogMessageDoc(message="Storage exception in controller " + "updates loop; terminating process", explanation=ERROR_DATABASE, recommendation=LogMessageDoc.CHECK_CONTROLLER), @LogMessageDoc(level="ERROR", message="Exception in controller updates loop", explanation="Failed to dispatch controller event", recommendation=LogMessageDoc.GENERIC_ACTION) }) void run(); void setConfigParams(Map<String, String> configParams); void init(Map<String, String> configParams); @LogMessageDoc(message="Waiting for storage source", explanation="The system database is not yet ready", recommendation="If this message persists, this indicates " + "that the system database has failed to start. " + LogMessageDoc.CHECK_CONTROLLER) void startupComponents(); @Override void addInfoProvider(String type, IInfoProvider provider); @Override void removeInfoProvider(String type, IInfoProvider provider); Map<String, Object> getControllerInfo(String type); @Override void addHAListener(IHAListener listener); @Override void removeHAListener(IHAListener listener); @Override Map<String, String> getControllerNodeIPs(); @Override void rowsModified(String tableName, Set<Object> rowKeys); @Override void rowsDeleted(String tableName, Set<Object> rowKeys); @Override long getSystemStartTime(); @Override void setAlwaysClearFlowsOnSwAdd(boolean value); boolean getAlwaysClearFlowsOnSwAdd();  }
@Test public void testSetRoleNull() { try { controller.setRole(null); fail("Should have thrown an Exception"); } catch (NullPointerException e) { } }
@Test public void testSetRole() { controller.connectedSwitches.add(new OFSwitchImpl()); RoleChanger roleChanger = createMock(RoleChanger.class); roleChanger.submitRequest(controller.connectedSwitches, Role.SLAVE); controller.roleChanger = roleChanger; assertEquals("Check that update queue is empty", 0, controller.updates.size()); replay(roleChanger); controller.setRole(Role.SLAVE); verify(roleChanger); Controller.IUpdate upd = controller.updates.poll(); assertNotNull("Check that update queue has an update", upd); assertTrue("Check that update is HARoleUpdate", upd instanceof Controller.HARoleUpdate); Controller.HARoleUpdate roleUpd = (Controller.HARoleUpdate)upd; assertSame(null, roleUpd.oldRole); assertSame(Role.SLAVE, roleUpd.newRole); }