internal Mock <INetworkPeer> GetNetworkPeerWithConnection() { var networkPeer = new Mock <INetworkPeer>(); var connection = new NetworkPeerConnection(this.Network, networkPeer.Object, new TcpClient(), 0, (message, token) => Task.CompletedTask, this.dateTimeProvider, this.loggerFactory, new PayloadProvider().DiscoverPayloads()); networkPeer.Setup(n => n.Connection) .Returns(connection); networkPeer.Setup(n => n.PeerEndPoint) .Returns(new System.Net.IPEndPoint(IPAddress.Loopback, 9999)); networkPeer.Setup(n => n.RemoteSocketAddress) .Returns(IPAddress.Loopback.EnsureIPv6()); networkPeer.Setup(n => n.RemoteSocketPort) .Returns(9999); networkPeer.Setup(n => n.RemoteSocketEndpoint) .Returns(new System.Net.IPEndPoint(IPAddress.Loopback.EnsureIPv6(), 9999)); networkPeer.Setup(n => n.State) .Returns(NetworkPeerState.Connected); var behavior = new Mock <IConnectionManagerBehavior>(); networkPeer.Setup(n => n.Behavior <IConnectionManagerBehavior>()) .Returns(behavior.Object); this.peerAddressManager.AddPeer(networkPeer.Object.PeerEndPoint, networkPeer.Object.PeerEndPoint.Address); this.connectionManager.AddConnectedPeer(networkPeer.Object); return(networkPeer); }
/// <summary>Creates a peer with extended puller behavior.</summary> public INetworkPeer CreatePeer(out ExtendedBlockPullerBehavior mockedBehavior, bool notSupportedVersion = false) { var peer = new Mock <INetworkPeer>(); var connection = new NetworkPeerConnection(Network.StratisMain, peer.Object, new TcpClient(), this.currentPeerId, (message, token) => Task.CompletedTask, new DateTimeProvider(), this.loggerFactory, new PayloadProvider()); this.currentPeerId++; peer.SetupGet(networkPeer => networkPeer.Connection).Returns(connection); var connectionParameters = new NetworkPeerConnectionParameters(); VersionPayload version = connectionParameters.CreateVersion(new IPEndPoint(1, 1), Network.StratisMain, new DateTimeProvider().GetTimeOffset()); if (notSupportedVersion) { version.Version = ProtocolVersion.NOBLKS_VERSION_START; } else { version.Services = NetworkPeerServices.Network; } peer.SetupGet(x => x.PeerVersion).Returns(version); peer.SetupGet(x => x.State).Returns(NetworkPeerState.HandShaked); peer.SetupGet(x => x.MessageReceived).Returns(new AsyncExecutionEvent <INetworkPeer, IncomingMessage>()); ExtendedBlockPullerBehavior behavior = this.CreateBlockPullerBehavior(); behavior.Attach(peer.Object); peer.Setup(x => x.Behavior <IBlockPullerBehavior>()).Returns(() => behavior); mockedBehavior = behavior; return(peer.Object); }
private Mock <INetworkPeer> CreatePeerMock() { var peer = new Mock <INetworkPeer>(); var connection = new NetworkPeerConnection(KnownNetworks.StratisMain, peer.Object, new TcpClient(), 0, (message, token) => Task.CompletedTask, new DateTimeProvider(), this.loggerFactory, new PayloadProvider()); peer.SetupGet(networkPeer => networkPeer.Connection).Returns(connection); var connectionParameters = new NetworkPeerConnectionParameters(); VersionPayload version = connectionParameters.CreateVersion(new IPEndPoint(1, 1), new IPEndPoint(1, 1), KnownNetworks.StratisMain, new DateTimeProvider().GetTimeOffset()); version.Services = NetworkPeerServices.Network; peer.SetupGet(x => x.PeerVersion).Returns(version); peer.SetupGet(x => x.State).Returns(NetworkPeerState.HandShaked); this.StateChanged = new AsyncExecutionEvent <INetworkPeer, NetworkPeerState>(); this.MessageReceived = new AsyncExecutionEvent <INetworkPeer, IncomingMessage>(); peer.Setup(x => x.StateChanged).Returns(() => this.StateChanged); peer.Setup(x => x.MessageReceived).Returns(() => this.MessageReceived); var connectionManagerBehaviorMock = new Mock <IConnectionManagerBehavior>(); connectionManagerBehaviorMock.Setup(x => x.Whitelisted).Returns(this.IsPeerWhitelisted); peer.Setup(x => x.Behavior <IConnectionManagerBehavior>()).Returns(() => connectionManagerBehaviorMock.Object); peer.SetupGet(x => x.PeerEndPoint).Returns(new IPEndPoint(1, 1)); return(peer); }