2023-09-13 10:24:46 +00:00
|
|
|
|
using CodexPlugin;
|
|
|
|
|
using Logging;
|
|
|
|
|
using NUnit.Framework;
|
|
|
|
|
|
2023-11-12 09:36:48 +00:00
|
|
|
|
namespace CodexTests.Helpers
|
2023-09-13 10:24:46 +00:00
|
|
|
|
{
|
|
|
|
|
public interface IFullConnectivityImplementation
|
|
|
|
|
{
|
|
|
|
|
string Description();
|
|
|
|
|
string ValidateEntry(FullConnectivityHelper.Entry entry, FullConnectivityHelper.Entry[] allEntries);
|
|
|
|
|
FullConnectivityHelper.PeerConnectionState Check(FullConnectivityHelper.Entry from, FullConnectivityHelper.Entry to);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public class FullConnectivityHelper
|
|
|
|
|
{
|
|
|
|
|
private static string Nl = Environment.NewLine;
|
|
|
|
|
private readonly ILog log;
|
|
|
|
|
private readonly IFullConnectivityImplementation implementation;
|
|
|
|
|
|
|
|
|
|
public FullConnectivityHelper(ILog log, IFullConnectivityImplementation implementation)
|
|
|
|
|
{
|
|
|
|
|
this.log = log;
|
|
|
|
|
this.implementation = implementation;
|
|
|
|
|
}
|
|
|
|
|
|
2023-09-19 09:51:59 +00:00
|
|
|
|
public void AssertFullyConnected(IEnumerable<ICodexNode> nodes)
|
2023-09-13 10:24:46 +00:00
|
|
|
|
{
|
|
|
|
|
AssertFullyConnected(nodes.ToArray());
|
|
|
|
|
}
|
|
|
|
|
|
2023-09-19 09:51:59 +00:00
|
|
|
|
private void AssertFullyConnected(ICodexNode[] nodes)
|
2023-09-13 10:24:46 +00:00
|
|
|
|
{
|
|
|
|
|
Log($"Asserting '{implementation.Description()}' for nodes: '{string.Join(",", nodes.Select(n => n.GetName()))}'...");
|
2024-03-26 09:31:49 +00:00
|
|
|
|
Assert.That(nodes.Length, Is.GreaterThan(1));
|
|
|
|
|
|
2023-09-13 10:24:46 +00:00
|
|
|
|
var entries = CreateEntries(nodes);
|
|
|
|
|
var pairs = CreatePairs(entries);
|
|
|
|
|
|
|
|
|
|
// Each pair gets two chances.
|
|
|
|
|
CheckAndRemoveSuccessful(pairs);
|
|
|
|
|
CheckAndRemoveSuccessful(pairs);
|
|
|
|
|
|
|
|
|
|
if (pairs.Any())
|
|
|
|
|
{
|
|
|
|
|
var pairDetails = string.Join(Nl, pairs.SelectMany(p => p.GetResultMessages()));
|
|
|
|
|
|
|
|
|
|
Log($"Connections failed:{Nl}{pairDetails}");
|
|
|
|
|
|
|
|
|
|
Assert.Fail(string.Join(Nl, pairs.SelectMany(p => p.GetResultMessages())));
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
Log($"'{implementation.Description()}' = Success! for nodes: {string.Join(",", nodes.Select(n => n.GetName()))}");
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private void CheckAndRemoveSuccessful(List<Pair> pairs)
|
|
|
|
|
{
|
|
|
|
|
var results = new List<string>();
|
|
|
|
|
foreach (var pair in pairs.ToArray())
|
|
|
|
|
{
|
|
|
|
|
pair.Check();
|
|
|
|
|
if (pair.Success)
|
|
|
|
|
{
|
|
|
|
|
results.AddRange(pair.GetResultMessages());
|
|
|
|
|
pairs.Remove(pair);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
Log($"Connections successful:{Nl}{string.Join(Nl, results)}");
|
|
|
|
|
}
|
|
|
|
|
|
2023-09-19 09:51:59 +00:00
|
|
|
|
private Entry[] CreateEntries(ICodexNode[] nodes)
|
2023-09-13 10:24:46 +00:00
|
|
|
|
{
|
|
|
|
|
var entries = nodes.Select(n => new Entry(n)).ToArray();
|
|
|
|
|
|
|
|
|
|
var errors = entries
|
|
|
|
|
.Select(e => implementation.ValidateEntry(e, entries))
|
|
|
|
|
.Where(s => !string.IsNullOrEmpty(s))
|
|
|
|
|
.ToArray();
|
|
|
|
|
|
|
|
|
|
if (errors.Any())
|
|
|
|
|
{
|
|
|
|
|
Assert.Fail("Some node entries failed to validate: " + string.Join(Nl, errors));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return entries;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private List<Pair> CreatePairs(Entry[] entries)
|
|
|
|
|
{
|
|
|
|
|
return CreatePairsIterator(entries).ToList();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private IEnumerable<Pair> CreatePairsIterator(Entry[] entries)
|
|
|
|
|
{
|
|
|
|
|
for (var x = 0; x < entries.Length; x++)
|
|
|
|
|
{
|
|
|
|
|
for (var y = x + 1; y < entries.Length; y++)
|
|
|
|
|
{
|
|
|
|
|
yield return new Pair(implementation, entries[x], entries[y]);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private void Log(string msg)
|
|
|
|
|
{
|
|
|
|
|
log.Log(msg);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public class Entry
|
|
|
|
|
{
|
2023-09-19 09:51:59 +00:00
|
|
|
|
public Entry(ICodexNode node)
|
2023-09-13 10:24:46 +00:00
|
|
|
|
{
|
|
|
|
|
Node = node;
|
|
|
|
|
Response = node.GetDebugInfo();
|
|
|
|
|
}
|
|
|
|
|
|
2023-09-19 09:51:59 +00:00
|
|
|
|
public ICodexNode Node { get; }
|
2024-03-26 07:58:16 +00:00
|
|
|
|
public DebugInfo Response { get; }
|
2023-09-13 10:24:46 +00:00
|
|
|
|
|
|
|
|
|
public override string ToString()
|
|
|
|
|
{
|
2024-03-26 07:58:16 +00:00
|
|
|
|
if (Response == null || string.IsNullOrEmpty(Response.Id)) return "UNKNOWN";
|
|
|
|
|
return Response.Id;
|
2023-09-13 10:24:46 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public enum PeerConnectionState
|
|
|
|
|
{
|
|
|
|
|
Unknown,
|
|
|
|
|
Connection,
|
|
|
|
|
NoConnection,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public class Pair
|
|
|
|
|
{
|
|
|
|
|
private TimeSpan aToBTime = TimeSpan.FromSeconds(0);
|
|
|
|
|
private TimeSpan bToATime = TimeSpan.FromSeconds(0);
|
|
|
|
|
private readonly IFullConnectivityImplementation implementation;
|
|
|
|
|
|
|
|
|
|
public Pair(IFullConnectivityImplementation implementation, Entry a, Entry b)
|
|
|
|
|
{
|
|
|
|
|
this.implementation = implementation;
|
|
|
|
|
A = a;
|
|
|
|
|
B = b;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public Entry A { get; }
|
|
|
|
|
public Entry B { get; }
|
|
|
|
|
public PeerConnectionState AKnowsB { get; private set; }
|
|
|
|
|
public PeerConnectionState BKnowsA { get; private set; }
|
|
|
|
|
public bool Success { get { return AKnowsB == PeerConnectionState.Connection && BKnowsA == PeerConnectionState.Connection; } }
|
|
|
|
|
public bool Inconclusive { get { return AKnowsB == PeerConnectionState.Unknown || BKnowsA == PeerConnectionState.Unknown; } }
|
|
|
|
|
|
|
|
|
|
public void Check()
|
|
|
|
|
{
|
|
|
|
|
aToBTime = Measure(() => AKnowsB = Check(A, B));
|
|
|
|
|
bToATime = Measure(() => BKnowsA = Check(B, A));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public override string ToString()
|
|
|
|
|
{
|
|
|
|
|
return $"[{string.Join(",", GetResultMessages())}]";
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public string[] GetResultMessages()
|
|
|
|
|
{
|
|
|
|
|
var aName = A.ToString();
|
|
|
|
|
var bName = B.ToString();
|
|
|
|
|
|
|
|
|
|
return new[]
|
|
|
|
|
{
|
|
|
|
|
$"[{aName} --> {bName}] = {AKnowsB} ({aToBTime.TotalSeconds} seconds)",
|
|
|
|
|
$"[{aName} <-- {bName}] = {BKnowsA} ({bToATime.TotalSeconds} seconds)"
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static TimeSpan Measure(Action action)
|
|
|
|
|
{
|
|
|
|
|
var start = DateTime.UtcNow;
|
|
|
|
|
action();
|
|
|
|
|
return DateTime.UtcNow - start;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private PeerConnectionState Check(Entry from, Entry to)
|
|
|
|
|
{
|
|
|
|
|
Thread.Sleep(10);
|
|
|
|
|
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
return implementation.Check(from, to);
|
|
|
|
|
}
|
|
|
|
|
catch
|
|
|
|
|
{
|
|
|
|
|
// Didn't get a conclusive answer. Try again later.
|
|
|
|
|
return PeerConnectionState.Unknown;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|