@Test
public void testClose() {
ConnectionAdapter mockedConnectionAdapter = mock(ConnectionAdapter.class);
InetSocketAddress mockRemoteAddress = mock(InetSocketAddress.class);
when(mockedConnectionAdapter.getRemoteAddress()).thenReturn(mockRemoteAddress);
when(connectionContext.getConnectionAdapter()).thenReturn(mockedConnectionAdapter);
NodeId dummyNodeId = new NodeId("dummyNodeId");
when(deviceState.getNodeId()).thenReturn(dummyNodeId);
ConnectionContext mockedAuxiliaryConnectionContext = prepareConnectionContext();
deviceContext.addAuxiliaryConenctionContext(mockedAuxiliaryConnectionContext);
DeviceContextClosedHandler mockedDeviceContextClosedHandler =
mock(DeviceContextClosedHandler.class);
deviceContext.addDeviceContextClosedHandler(mockedDeviceContextClosedHandler);
deviceContext.close();
verify(connectionContext).closeConnection(eq(false));
verify(deviceState).setValid(eq(false));
verify(txChainManager).close();
verify(mockedAuxiliaryConnectionContext).closeConnection(eq(false));
}
