public async Task RetriesWorksTestAsync()
{
byte[] defaultData = new byte[] { 1, 2, 3 };
var data = new ConcurrentDictionary <BobKey, byte[]>();
var behaviour1 = new BobNodeClientMockHelper.MockClientBehaviour();
var stat1 = new BobNodeClientMockHelper.MockClientStat();
var behaviour2 = new BobNodeClientMockHelper.MockClientBehaviour();
var stat2 = new BobNodeClientMockHelper.MockClientStat();
BobNodeClient[] clients = new BobNodeClient[]
{
BobNodeClientMockHelper.CreateMockedClientWithData(data, behaviour: behaviour1, stat: stat1),
BobNodeClientMockHelper.CreateMockedClientWithData(data, behaviour: behaviour2, stat: stat2)
};
behaviour1.ErrorStatus = new Grpc.Core.Status(Grpc.Core.StatusCode.Internal, "Internal error");
using (var client = new BobClusterClient(clients, SequentialNodeSelectionPolicy.Factory, operationRetryCount: 1))
{
await client.PutAsync(BobKey.FromUInt64(1), defaultData);
Assert.Equal(defaultData, await client.GetAsync(BobKey.FromUInt64(1)));
Assert.True((await client.ExistsAsync(new BobKey[] { BobKey.FromUInt64(1) }))[0]);
for (int i = 10; i < 100; i++)
{
await client.PutAsync(BobKey.FromUInt64((ulong)i), defaultData);
}
for (int i = 10; i < 100; i++)
{
Assert.Equal(defaultData, await client.GetAsync(BobKey.FromUInt64((ulong)i)));
}
for (int i = 10; i < 100; i++)
{
Assert.True((await client.ExistsAsync(new BobKey[] { BobKey.FromUInt64((ulong)i) }))[0]);
}
}
}
public async Task PutGetExistOperationTestAsync()
{
byte[] defaultData = new byte[] { 1, 2, 3 };
var data = new ConcurrentDictionary <BobKey, byte[]>();
var stat1 = new BobNodeClientMockHelper.MockClientStat();
var stat2 = new BobNodeClientMockHelper.MockClientStat();
BobNodeClient[] clients = new BobNodeClient[]
{
BobNodeClientMockHelper.CreateMockedClientWithData(data, behaviour: null, stat: stat1),
BobNodeClientMockHelper.CreateMockedClientWithData(data, behaviour: null, stat: stat2)
};
using (var client = new BobClusterClient(clients, SequentialNodeSelectionPolicy.Factory, 0))
{
await client.PutAsync(BobKey.FromUInt64(1), defaultData);
await client.PutAsync(BobKey.FromUInt64(ulong.MaxValue), defaultData);
Assert.Equal(2, stat1.PutRequestCount + stat2.PutRequestCount);
Assert.Equal(defaultData, await client.GetAsync(BobKey.FromUInt64(1)));
Assert.Equal(defaultData, await client.GetAsync(BobKey.FromUInt64(ulong.MaxValue)));
await Assert.ThrowsAsync <BobKeyNotFoundException>(() => client.GetAsync(BobKey.FromUInt64(2)));
Assert.Equal(3, stat1.GetRequestCount + stat2.GetRequestCount);
Assert.Equal(new bool[] { true, false }, await client.ExistsAsync(new BobKey[] { BobKey.FromUInt64(1), BobKey.FromUInt64(2) }));
Assert.Equal(1, stat1.ExistsRequestCount + stat2.ExistsRequestCount);
for (ulong i = 100; i < 10000; i++)
{
await client.PutAsync(BobKey.FromUInt64(i), defaultData);
}
for (ulong i = 100; i < 10000; i++)
{
Assert.Equal(defaultData, await client.GetAsync(BobKey.FromUInt64(i)));
}
for (ulong i = 100; i < 10000; i++)
{
Assert.Equal(defaultData, await client.GetAsync(BobKey.FromUInt64(i), fullGet: true, new CancellationToken()));
}
Assert.Equal(10000 - 100, stat1.RequestsWithFullGet + stat2.RequestsWithFullGet);
Assert.All(await client.ExistsAsync(Enumerable.Range(100, 10000 - 100).Select(o => BobKey.FromUInt64((ulong)o)).ToArray()), res => Assert.True(res));
Assert.All(await client.ExistsAsync(Enumerable.Range(20000, 1000).Select(o => BobKey.FromUInt64((ulong)o)).ToArray(), fullGet: true, new CancellationToken()), res => Assert.False(res));
Assert.Equal(10000 - 100 + 1, stat1.RequestsWithFullGet + stat2.RequestsWithFullGet);
for (ulong i = uint.MaxValue; i < (ulong)uint.MaxValue + 1000; i++)
{
await client.PutAsync(BobKey.FromUInt64(i), defaultData);
}
for (ulong i = uint.MaxValue; i < (ulong)uint.MaxValue + 1000; i++)
{
Assert.Equal(defaultData, await client.GetAsync(BobKey.FromUInt64(i)));
}
Assert.True(stat1.RequestsWithFullGet + stat2.RequestsWithFullGet > 0);
Assert.True(stat1.TotalRequestCount > 0);
Assert.True(stat2.TotalRequestCount > 0);
Assert.True(Math.Abs(stat1.TotalRequestCount - stat2.TotalRequestCount) <= 1);
}
}
