Coverage Summary for Class: NodeChallengeManager (co.rsk.net.discovery)

Class Class, % Method, % Line, %
NodeChallengeManager 0% (0/1) 0% (0/4) 0% (0/9)


1 /* 2  * This file is part of RskJ 3  * Copyright (C) 2017 RSK Labs Ltd. 4  * 5  * This program is free software: you can redistribute it and/or modify 6  * it under the terms of the GNU Lesser General Public License as published by 7  * the Free Software Foundation, either version 3 of the License, or 8  * (at your option) any later version. 9  * 10  * This program is distributed in the hope that it will be useful, 11  * but WITHOUT ANY WARRANTY; without even the implied warranty of 12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 13  * GNU Lesser General Public License for more details. 14  * 15  * You should have received a copy of the GNU Lesser General Public License 16  * along with this program. If not, see <http://www.gnu.org/licenses/>. 17  */ 18  19 package co.rsk.net.discovery; 20  21 import co.rsk.net.discovery.message.PingPeerMessage; 22 import com.google.common.annotations.VisibleForTesting; 23 import org.ethereum.net.rlpx.Node; 24  25 import java.util.Map; 26 import java.util.concurrent.ConcurrentHashMap; 27  28 /** 29  * Created by mario on 22/02/17. 30  */ 31 public class NodeChallengeManager { 32  private Map<String, NodeChallenge> activeChallenges = new ConcurrentHashMap<>(); 33  34  public NodeChallenge startChallenge(Node challengedNode, Node challenger, PeerExplorer explorer) { 35  PingPeerMessage pingMessage = explorer.sendPing(challengedNode.getAddress(), 1, challengedNode); 36  String messageId = pingMessage.getMessageId(); 37  NodeChallenge challenge = new NodeChallenge(challengedNode, challenger, messageId); 38  activeChallenges.put(messageId, challenge); 39  return challenge; 40  } 41  42  public NodeChallenge removeChallenge(String challengeId) { 43  return activeChallenges.remove(challengeId); 44  } 45  46  @VisibleForTesting 47  public int activeChallengesCount() { 48  return activeChallenges.size(); 49  } 50 }