using CodexContractsPlugin; using CodexPlugin; using Core; using DistTestCore; using DistTestCore.Helpers; using GethPlugin; using NUnit.Framework.Constraints; namespace Tests { public class CodexDistTest : DistTest { private readonly List onlineCodexNodes = new List(); public CodexDistTest() { ProjectPlugin.Load(); ProjectPlugin.Load(); ProjectPlugin.Load(); ProjectPlugin.Load(); } public ICodexNode AddCodex() { return AddCodex(s => { }); } public ICodexNode AddCodex(Action setup) { return AddCodex(1, setup)[0]; } public ICodexNodeGroup AddCodex(int numberOfNodes) { return AddCodex(numberOfNodes, s => { }); } public ICodexNodeGroup AddCodex(int numberOfNodes, Action setup) { var group = Ci.StartCodexNodes(numberOfNodes, s => { setup(s); OnCodexSetup(s); }); onlineCodexNodes.AddRange(group); return group; } public PeerConnectionTestHelpers CreatePeerConnectionTestHelpers() { return new PeerConnectionTestHelpers(GetTestLog()); } public PeerDownloadTestHelpers CreatePeerDownloadTestHelpers() { return new PeerDownloadTestHelpers(GetTestLog(), Get().GetFileManager()); } public IEnumerable GetAllOnlineCodexNodes() { return onlineCodexNodes; } public void AssertBalance(IGethNode gethNode, ICodexContracts contracts, ICodexNode codexNode, Constraint constraint, string msg = "") { AssertHelpers.RetryAssert(constraint, () => contracts.GetTestTokenBalance(gethNode, codexNode), nameof(AssertBalance) + msg); } protected virtual void OnCodexSetup(ICodexSetup setup) { } } }