public async Task Create_RestoreFromBacklog()
{
var backlog = new JsonFileRpcBacklog(new DirectoryInfo(backlogDir));
string targetPeerID = "TestClient";
int callsCount = 10;
// Create queue and enqueue
var queue = await RpcQueue.Create(targetPeerID, backlog);
for (int i = 0; i < callsCount; i++)
{
await queue.Enqueue(CreateCall("TestMethod", targetPeerID));
}
// Recreate queue and check contents
queue = await RpcQueue.Create(targetPeerID, backlog);
Assert.AreEqual(callsCount, queue.Count);
}
public async Task Enqueue_And_Peek_And_Dequeue_CorrectOrder()
{
int callsCount = 100;
string targetPeerID = "TestClient";
var backlog = new JsonFileRpcBacklog(new DirectoryInfo(backlogDir));
var queue = await RpcQueue.Create(targetPeerID, backlog);
var allTasks = new List <Task>();
// Enqueue
allTasks.Add(Task.Run(async() => {
await Task.Delay(1000); // Wait a moment, so that the test starts with an empty backlog
for (int iCall = 0; iCall < callsCount; iCall++)
{
await queue.Enqueue(CreateCall("TestMethod", targetPeerID));
await Task.Delay(50);
}
}));
// Peek and dequeue, with checking the order
allTasks.Add(Task.Run(async() => {
ulong lastID = 0;
int receivedCallsCount = 0;
while (receivedCallsCount < callsCount)
{
if (await queue.Peek() is RpcCall peekedCall)
{
// Check that the ID is higher than the last one
Assert.IsTrue(peekedCall.Method.ID > lastID);
// Dequeue
var dequeuedCall = await queue.Dequeue();
// Dequeued call must be the same call (same ID) as the previously peeked call
Assert.AreEqual(dequeuedCall !.Method.ID, peekedCall.Method.ID);
lastID = dequeuedCall.Method.ID;
receivedCallsCount++;
}
await Task.Delay(50);
}
}));
// Wait until tasks are finished
await Task.WhenAll(allTasks);
// Backlog must also be empty now
Assert.AreEqual(0, (await backlog.ReadAll(targetPeerID)).Count);
}
public async Task Enqueue_RemoveObsolete()
{
int callsCount = 100;
string targetPeerID = "TestClient";
var retryLatestMethods = new HashSet <string> {
"Method5", "Method8"
};
var backlog = new JsonFileRpcBacklog(new DirectoryInfo(backlogDir));
var queue = await RpcQueue.Create(targetPeerID, backlog);
var allCalls = new List <RpcCall>();
// Enqueue
for (int iCall = 0; iCall < callsCount; iCall++)
{
string methodName = "Method" + random.Next(10);
var retry = retryLatestMethods.Contains(methodName)
? RpcRetryStrategy.RetryLatest : RpcRetryStrategy.Retry;
var call = CreateCall(methodName, targetPeerID, retry);
allCalls.Add(call);
await queue.Enqueue(call);
await Task.Delay(50);
}
// Test backlog: Only last call of the RetryLatest methods must be there
var backlogCalls = await backlog.ReadAll(targetPeerID);
var callsWithoutObsolete = allCalls.Where((call, index) => {
if (call.RetryStrategy == RpcRetryStrategy.RetryLatest)
{
return(allCalls.FindLastIndex(it => it.Method.Name == call.Method.Name) == index);
}
return(true);
}).ToList();
CollectionAssert.AreEqual(callsWithoutObsolete, backlogCalls);
// Test queue: All methods must still be there
Assert.AreEqual(allCalls.Count, queue.Count);
for (int i = 0; i < allCalls.Count; i++)
{
Assert.AreEqual(allCalls[i], await queue.Dequeue());
}
}
