public async Task IterateBatchTest()
{
var inMemoryStorePartition = new InMemoryDbStore();
for (int i = 0; i < 100; i++)
{
await inMemoryStorePartition.Put($"key{i}".ToBytes(), $"val{i}".ToBytes());
}
int[] counter = { 11 };
await inMemoryStorePartition.IterateBatch(
$"key{counter[0]}".ToBytes(),
20,
(key, value) =>
{
Assert.Equal($"key{counter[0]}", key.FromBytes <string>());
Assert.Equal($"val{counter[0]}", value.FromBytes <string>());
counter[0]++;
return(Task.CompletedTask);
});
counter[0] = 61;
await inMemoryStorePartition.IterateBatch(
$"key{counter[0]}".ToBytes(),
20,
(key, value) =>
{
Assert.Equal($"key{counter[0]}", key.FromBytes <string>());
Assert.Equal($"val{counter[0]}", value.FromBytes <string>());
counter[0]++;
return(Task.CompletedTask);
});
}
public async Task UpdateDuringIterateTest()
{
int totalCount = 100;
var inMemoryStorePartition = new InMemoryDbStore();
for (int i = 0; i <= totalCount; i++)
{
await inMemoryStorePartition.Put($"key{i}".ToBytes(), $"val{i}".ToBytes());
}
Task iterateBatch = Task.Run(
async() =>
{
await inMemoryStorePartition.IterateBatch(
10,
async(key, value) => { await Task.Delay(TimeSpan.FromMilliseconds(500)); });
});
Task updateTask = Task.Run(
async() =>
{
await inMemoryStorePartition.Remove("key0".ToBytes());
await Task.Delay(TimeSpan.FromMilliseconds(500));
await inMemoryStorePartition.Put("key20".ToBytes(), "newValue".ToBytes());
await Task.Delay(TimeSpan.FromMilliseconds(500));
await inMemoryStorePartition.Remove("key8".ToBytes());
});
await Task.WhenAll(updateTask, iterateBatch);
Assert.True(iterateBatch.IsCompletedSuccessfully);
Assert.True(updateTask.IsCompletedSuccessfully);
}
public async Task IterateEmptyBatchTest()
{
var inMemoryStorePartition = new InMemoryDbStore();
bool callbackCalled = false;
await inMemoryStorePartition.IterateBatch(20, (key, value) =>
{
callbackCalled = true;
return(Task.CompletedTask);
});
Assert.False(callbackCalled);
}
async Task <(IMessageStore, ICheckpointStore, InMemoryDbStore)> GetMessageStore(bool checkEntireQueueOnCleanup, int ttlSecs = 300, int messageCleanupIntervalSecs = 30)
{
var dbStoreProvider = new InMemoryDbStoreProvider();
IStoreProvider storeProvider = new StoreProvider(dbStoreProvider);
InMemoryDbStore inMemoryDbStore = dbStoreProvider.GetDbStore("messages") as InMemoryDbStore;
ICheckpointStore checkpointStore = CheckpointStore.Create(storeProvider);
IMessageStore messageStore = new MessageStore(storeProvider, checkpointStore, TimeSpan.FromSeconds(ttlSecs), checkEntireQueueOnCleanup, messageCleanupIntervalSecs);
await messageStore.AddEndpoint("module1");
await messageStore.AddEndpoint("module2");
return(messageStore, checkpointStore, inMemoryDbStore);
}
public async Task GetMultipleOperationsTest()
{
int totalCount = 10000;
var inMemoryStorePartition = new InMemoryDbStore();
for (int i = 0; i <= totalCount; i++)
{
await inMemoryStorePartition.Put($"key{i}".ToBytes(), $"val{i}".ToBytes());
}
Task getTask = Task.Run(() => inMemoryStorePartition.Get($"key{totalCount}".ToBytes()));
Task removeTask = Task.Run(async() => await inMemoryStorePartition.Remove("key0".ToBytes()));
Task putTask = Task.Run(async() => await inMemoryStorePartition.Put("key1".ToBytes(), "newvalue".ToBytes()));
await Task.WhenAll(removeTask, putTask, getTask);
Assert.True(getTask.IsCompletedSuccessfully);
Assert.True(removeTask.IsCompletedSuccessfully);
Assert.True(putTask.IsCompletedSuccessfully);
}
public async Task GetFirstLastValueTest()
{
var inMemoryStorePartition = new InMemoryDbStore();
await inMemoryStorePartition.Put("key1".ToBytes(), "val1".ToBytes());
await inMemoryStorePartition.Put("key2".ToBytes(), "val2".ToBytes());
await inMemoryStorePartition.Put("key3".ToBytes(), "val3".ToBytes());
Option <(byte[] key, byte[] value)> firstValueOption = await inMemoryStorePartition.GetFirstEntry();
Assert.True(firstValueOption.HasValue);
(byte[] key, byte[] value)firstValue = firstValueOption.OrDefault();
Assert.Equal("key1", firstValue.key.FromBytes <string>());
Assert.Equal("val1", firstValue.value.FromBytes <string>());
Option <(byte[] key, byte[] value)> lastValueOption = await inMemoryStorePartition.GetLastEntry();
Assert.True(lastValueOption.HasValue);
(byte[] key, byte[] value)lastValue = lastValueOption.OrDefault();
Assert.Equal("key3", lastValue.key.FromBytes <string>());
Assert.Equal("val3", lastValue.value.FromBytes <string>());
}
public async Task GetIndexTest()
{
var inMemoryStorePartition = new InMemoryDbStore();
await inMemoryStorePartition.Put("key1".ToBytes(), "val1".ToBytes());
await inMemoryStorePartition.Put("key2".ToBytes(), "val2".ToBytes());
await inMemoryStorePartition.Put("key3".ToBytes(), "val3".ToBytes());
Assert.Equal(0, inMemoryStorePartition.GetIndex("key1".ToBytes()));
Assert.Equal(1, inMemoryStorePartition.GetIndex("key2".ToBytes()));
Assert.Equal(2, inMemoryStorePartition.GetIndex("key3".ToBytes()));
await inMemoryStorePartition.Remove("key2".ToBytes());
Assert.Equal(0, inMemoryStorePartition.GetIndex("key1".ToBytes()));
Assert.Equal(-1, inMemoryStorePartition.GetIndex("key2".ToBytes()));
Assert.Equal(1, inMemoryStorePartition.GetIndex("key3".ToBytes()));
await inMemoryStorePartition.Put("key3".ToBytes(), "newVal3".ToBytes());
Assert.Equal(1, inMemoryStorePartition.GetIndex("key3".ToBytes()));
}
