- final Object[] args = invocation.getArguments();
- AbstractPCEPSessionTest.this.receivedMsgs.add((Notification) args[0]);
- return this.channelFuture;
- }).when(this.clientListener).writeAndFlush(any(Notification.class));
- doReturn(null).when(this.channelFuture).addListener(Mockito.any());
- doReturn("TestingChannel").when(this.clientListener).toString();
- doReturn(this.pipeline).when(this.clientListener).pipeline();
- doReturn(this.pipeline).when(this.pipeline).replace(any(ChannelHandler.class), any(String.class),
- any(ChannelHandler.class));
- doReturn(this.eventLoop).when(this.clientListener).eventLoop();
- doNothing().when(this.statsRegistry).bind(any(), any());
- doNothing().when(this.statsRegistry).unbind(any());
- doReturn(null).when(this.eventLoop).schedule(any(Runnable.class), any(long.class),
- any(TimeUnit.class));
- doReturn(true).when(this.clientListener).isActive();
- final InetSocketAddress ra = new InetSocketAddress(this.testAddress, 4189);
- doReturn(ra).when(this.clientListener).remoteAddress();
- final InetSocketAddress la = new InetSocketAddress(this.testAddress, InetSocketAddressUtil.getRandomPort());
- doReturn(la).when(this.clientListener).localAddress();
-
- doReturn(mock(ChannelFuture.class)).when(this.clientListener).close();
-
- doReturn(getDataBroker()).when(this.topologyDependencies).getDataBroker();
- doReturn(this.statsRegistry).when(this.topologyDependencies).getStateRegistry();
-
- @SuppressWarnings("unchecked") final T listenerFactory = (T) ((Class) ((ParameterizedType) this.getClass()
- .getGenericSuperclass()).getActualTypeArguments()[0]).newInstance();
- doReturn(new IpAddressNoZone(new Ipv4AddressNoZone(this.testAddress))).when(this.sessionConfig)
- .getListenAddress();
- doReturn(new PortNumber(Uint16.valueOf(4189))).when(this.sessionConfig).getListenPort();
- doReturn(RPC_TIMEOUT).when(this.sessionConfig).getRpcTimeout();
- doReturn(TEST_TOPOLOGY_ID).when(this.topology).getTopologyId();
- doReturn(Collections.emptyList()).when(this.topology).getNode();
- doReturn(null).when(this.topologyDependencies).getPceServerProvider();
-
- final PCEPTopologyConfiguration configDep = new PCEPTopologyConfiguration(this.sessionConfig, this.topology);
- this.manager = new ServerSessionManager(this.topologyDependencies, listenerFactory, configDep);
+ receivedMsgs.add(invocation.getArgument(0, Notification.class));
+ return channelFuture;
+ }).when(clientListener).writeAndFlush(any(Notification.class));
+ doReturn(null).when(channelFuture).addListener(any());
+ doReturn("TestingChannel").when(clientListener).toString();
+ doReturn(pipeline).when(clientListener).pipeline();
+ doReturn(pipeline).when(pipeline).replace(any(ChannelHandler.class), any(String.class),
+ any(ChannelHandler.class));
+ doReturn(eventLoop).when(clientListener).eventLoop();
+ doAnswer(inv -> NoOpObjectRegistration.of(inv.getArgument(1, PcepSessionState.class)))
+ .when(stateRegistry).bind(any(), any());
+ doReturn(null).when(eventLoop).schedule(any(Runnable.class), any(long.class), any(TimeUnit.class));
+ doReturn(true).when(clientListener).isActive();
+ final InetSocketAddress ra = new InetSocketAddress(testAddress, 4189);
+ doReturn(ra).when(clientListener).remoteAddress();
+ final InetSocketAddress la = new InetSocketAddress(testAddress, InetSocketAddressUtil.getRandomPort());
+ doReturn(la).when(clientListener).localAddress();
+
+ doReturn(mock(ChannelFuture.class)).when(clientListener).close();
+
+ doReturn(getDataBroker()).when(topologyDependencies).getDataBroker();
+ doReturn(stateRegistry).when(topologyDependencies).getStateRegistry();
+ doReturn(timer).when(topologyDependencies).getTimer();
+ doReturn(null).when(topologyDependencies).getPceServerProvider();
+
+ manager = customizeSessionManager(new ServerSessionManager(TOPO_IID, topologyDependencies, RPC_TIMEOUT,
+ new GraphKey("graph-test")));