- Node nodeMock = mock(Node.class);
- List<Node> nodeList = new ArrayList<>();
- nodeList.add(nodeMock);
- doReturn(nodeList).when(topologyMock).getNode();
-
- NodeId nodeId = new NodeId("nodeIdValue1");
- doReturn(nodeId).when(nodeMock).getNodeId();
+ final NodeKey nodeKey = new NodeKey(new NodeId("nodeIdValue1"));
+ final Node node = new NodeBuilder().withKey(nodeKey).build();
+ doReturn(Map.of(nodeKey, node)).when(topologyMock).getNode();