public static IHastingsOriginator MockOriginator(PeerId peer = default,
INeighbours neighbours = default)
{
var memento = new HastingsMemento(peer, neighbours);
return(new HastingsOriginator(memento));
}
public static IHastingsMemento MockMemento(PeerId identifier = default,
INeighbours neighbours = default)
{
var peerParam = identifier ?? PeerIdHelper.GetPeerId(StringHelper.RandomString());
var neighbourParam = neighbours ?? MockNeighbours();
return(new HastingsMemento(peerParam, neighbourParam));
}
public static IHastingsMemento SubMemento(PeerId identifier = default,
INeighbours neighbours = default)
{
var subbedMemento = Substitute.For <IHastingsMemento>();
subbedMemento.Peer.Returns(identifier ?? PeerIdHelper.GetPeerId());
subbedMemento.Neighbours.Returns(neighbours ?? MockNeighbours());
return(subbedMemento);
}
public static IHastingsOriginator SubOriginator(PeerId peer = default,
INeighbours neighbours = default,
ICorrelationId expectedPnr = default)
{
var subbedOriginator = Substitute.For <IHastingsOriginator>();
subbedOriginator.Neighbours.Count.Returns(5);
subbedOriginator.Peer.Returns(peer ?? new PeerId());
subbedOriginator.Neighbours.Returns(neighbours ?? Substitute.For <INeighbours>());
subbedOriginator.PnrCorrelationId.Returns(expectedPnr ?? CorrelationId.GenerateCorrelationId());
return(subbedOriginator);
}
public DiscoveryTestBuilder WithStepProposal(IHastingsOriginator stateCandidate = default,
bool mock = false,
PeerId peer = default,
INeighbours neighbours = default,
ICorrelationId expectedPnr = default)
{
var pnrCorrelationId = expectedPnr ?? CorrelationId.GenerateCorrelationId();
_currentState = mock
? stateCandidate ?? DiscoveryHelper.MockOriginator(peer, neighbours)
: stateCandidate ?? DiscoveryHelper.SubOriginator(peer, neighbours, pnrCorrelationId);
return(this);
}
public DiscoveryTestBuilder WithCurrentStep(IHastingsMemento currentStep = default,
bool mock = false,
PeerId peer = default,
INeighbours neighbours = default)
{
if (_careTaker == null)
{
WithCareTaker();
}
var memento = mock
? currentStep ?? DiscoveryHelper.MockMemento(peer, neighbours)
: currentStep ?? DiscoveryHelper.SubMemento(peer, neighbours);
_careTaker.Add(memento);
return(this);
}
public HastingsMemento(PeerId peer, INeighbours neighbours)
{
Peer = peer;
Neighbours = neighbours;
}
