Index tests
This commit is contained in:
Sean Garrett
@@ -0,0 +1,45 @@
|
||||
using System.Collections.Generic;
|
||||
using System.Threading;
|
||||
using CoreUnitTests.Infrastructure;
|
||||
using MongoDB.Bson;
|
||||
using MongoDB.Bson.Serialization;
|
||||
using MongoDB.Driver;
|
||||
using MongoDbGenericRepository.DataAccess.Index;
|
||||
using Moq;
|
||||
|
||||
namespace CoreUnitTests.DataAccessTests.MongoDbIndexHandlerTests;
|
||||
|
||||
public class BaseIndexTests : GenericTestContext<MongoDbIndexHandler>
|
||||
{
|
||||
protected (Mock<IAsyncCursor<BsonDocument>>, Mock<IMongoIndexManager<TDocument>>) SetupIndexes<TDocument>(
|
||||
List<BsonDocument> indexes,
|
||||
Mock<IMongoCollection<TDocument>> collection)
|
||||
{
|
||||
var asyncCursor = MockOf<IAsyncCursor<BsonDocument>>();
|
||||
var moveNextSequence = asyncCursor
|
||||
.SetupSequence(x => x.MoveNextAsync(It.IsAny<CancellationToken>()));
|
||||
|
||||
var currentSequence = asyncCursor
|
||||
.SetupSequence(x => x.Current);
|
||||
|
||||
foreach (var bsonDocument in indexes)
|
||||
{
|
||||
moveNextSequence.ReturnsAsync(true);
|
||||
currentSequence.Returns(new[] {bsonDocument});
|
||||
}
|
||||
|
||||
moveNextSequence.ReturnsAsync(false);
|
||||
|
||||
var indexManager = MockOf<IMongoIndexManager<TDocument>>();
|
||||
indexManager
|
||||
.Setup(x => x.ListAsync(It.IsAny<CancellationToken>()))
|
||||
.ReturnsAsync(asyncCursor.Object);
|
||||
|
||||
collection
|
||||
.SetupGet(x => x.Indexes)
|
||||
.Returns(indexManager.Object);
|
||||
|
||||
return (asyncCursor, indexManager);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,180 @@
|
||||
using System;
|
||||
using System.Linq.Expressions;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using AutoFixture;
|
||||
using CoreUnitTests.Infrastructure;
|
||||
using CoreUnitTests.Infrastructure.Model;
|
||||
using FluentAssertions;
|
||||
using MongoDB.Driver;
|
||||
using MongoDbGenericRepository;
|
||||
using MongoDbGenericRepository.Models;
|
||||
using Moq;
|
||||
using Xunit;
|
||||
|
||||
namespace CoreUnitTests.DataAccessTests.MongoDbIndexHandlerTests;
|
||||
|
||||
public class CreateTextIndexAsyncTests : BaseIndexTests
|
||||
{
|
||||
private readonly Expression<Func<TestDocument, object>> fieldExpression = t => t.SomeContent2;
|
||||
|
||||
[Fact]
|
||||
public async Task WhenFieldExpression_ThenCreatesIndex()
|
||||
{
|
||||
// Arrange
|
||||
var expectedIndexName = Fixture.Create<string>();
|
||||
var collection = MockOf<IMongoCollection<TestDocument>>();
|
||||
SetupContext(collection);
|
||||
|
||||
var indexManger = SetupIndexManager(collection, expectedIndexName);
|
||||
|
||||
// Act
|
||||
var result = await Sut.CreateTextIndexAsync<TestDocument, Guid>(fieldExpression);
|
||||
|
||||
// Assert
|
||||
result.Should().Be(expectedIndexName);
|
||||
indexManger.Verify(
|
||||
x => x.CreateOneAsync(
|
||||
It.Is<CreateIndexModel<TestDocument>>(t => t.Keys.EqualToJson("{\"SomeContent2\":\"text\"}")),
|
||||
null,
|
||||
CancellationToken.None),
|
||||
Times.Once);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task WhenFieldExpressionAndOptions_ThenCreatesIndex()
|
||||
{
|
||||
// Arrange
|
||||
var expectedIndexName = Fixture.Create<string>();
|
||||
var collection = MockOf<IMongoCollection<TestDocument>>();
|
||||
var options = Fixture.Create<IndexCreationOptions>();
|
||||
|
||||
SetupContext(collection);
|
||||
var indexManger = SetupIndexManager(collection, expectedIndexName);
|
||||
|
||||
// Act
|
||||
var result = await Sut.CreateTextIndexAsync<TestDocument, Guid>(fieldExpression, options);
|
||||
|
||||
// Assert
|
||||
result.Should().Be(expectedIndexName);
|
||||
indexManger.Verify(
|
||||
x => x.CreateOneAsync(
|
||||
It.Is<CreateIndexModel<TestDocument>>(
|
||||
t => t.Keys.EqualToJson("{\"SomeContent2\":\"text\"}") &&
|
||||
t.Options.EqualTo(options)),
|
||||
null,
|
||||
CancellationToken.None),
|
||||
Times.Once);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task WhenFieldExpressionAndPartitionKey_ThenCreatesIndex()
|
||||
{
|
||||
// Arrange
|
||||
var expectedIndexName = Fixture.Create<string>();
|
||||
var collection = MockOf<IMongoCollection<TestDocument>>();
|
||||
var partitionKey = Fixture.Create<string>();
|
||||
|
||||
var context = SetupContext(collection);
|
||||
|
||||
var indexManger = SetupIndexManager(collection, expectedIndexName);
|
||||
|
||||
// Act
|
||||
var result = await Sut.CreateTextIndexAsync<TestDocument, Guid>(fieldExpression, partitionKey: partitionKey);
|
||||
|
||||
// Assert
|
||||
result.Should().Be(expectedIndexName);
|
||||
indexManger.Verify(
|
||||
x => x.CreateOneAsync(
|
||||
It.Is<CreateIndexModel<TestDocument>>(
|
||||
t => t.Keys.EqualToJson("{\"SomeContent2\":\"text\"}") ),
|
||||
null,
|
||||
CancellationToken.None),
|
||||
Times.Once);
|
||||
context.Verify(x => x.GetCollection<TestDocument>(partitionKey), Times.Once);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task WhenFieldExpressionAndCancellationToken_ThenCreatesIndex()
|
||||
{
|
||||
// Arrange
|
||||
var expectedIndexName = Fixture.Create<string>();
|
||||
var collection = MockOf<IMongoCollection<TestDocument>>();
|
||||
var token = new CancellationToken(true);
|
||||
|
||||
SetupContext(collection);
|
||||
|
||||
var indexManger = SetupIndexManager(collection, expectedIndexName);
|
||||
|
||||
// Act
|
||||
var result = await Sut.CreateTextIndexAsync<TestDocument, Guid>(fieldExpression, cancellationToken: token);
|
||||
|
||||
// Assert
|
||||
result.Should().Be(expectedIndexName);
|
||||
indexManger.Verify(
|
||||
x => x.CreateOneAsync(
|
||||
It.Is<CreateIndexModel<TestDocument>>(
|
||||
t => t.Keys.EqualToJson("{\"SomeContent2\":\"text\"}") ),
|
||||
null,
|
||||
token),
|
||||
Times.Once);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task WhenFieldExpressionAndOptionsAndPartitionKeyAndCancellationToken_ThenCreatesIndex()
|
||||
{
|
||||
// Arrange
|
||||
var expectedIndexName = Fixture.Create<string>();
|
||||
var collection = MockOf<IMongoCollection<TestDocument>>();
|
||||
var token = new CancellationToken(true);
|
||||
var partitionKey = Fixture.Create<string>();
|
||||
var options = Fixture.Create<IndexCreationOptions>();
|
||||
|
||||
var context = SetupContext(collection);
|
||||
|
||||
var indexManger = SetupIndexManager(collection, expectedIndexName);
|
||||
|
||||
// Act
|
||||
var result = await Sut.CreateTextIndexAsync<TestDocument, Guid>(fieldExpression, options, partitionKey, token);
|
||||
|
||||
// Assert
|
||||
result.Should().Be(expectedIndexName);
|
||||
indexManger.Verify(
|
||||
x => x.CreateOneAsync(
|
||||
It.Is<CreateIndexModel<TestDocument>>(
|
||||
t => t.Keys.EqualToJson("{\"SomeContent2\":\"text\"}") &&
|
||||
t.Options.EqualTo(options)),
|
||||
null,
|
||||
token),
|
||||
Times.Once);
|
||||
context.Verify(x => x.GetCollection<TestDocument>(partitionKey), Times.Once);
|
||||
}
|
||||
|
||||
private Mock<IMongoDbContext> SetupContext<TDocument>(Mock<IMongoCollection<TDocument>> collection)
|
||||
{
|
||||
var context = MockOf<IMongoDbContext>();
|
||||
context
|
||||
.Setup(x => x.GetCollection<TDocument>(It.IsAny<string>()))
|
||||
.Returns(collection.Object);
|
||||
|
||||
return context;
|
||||
}
|
||||
|
||||
private Mock<IMongoIndexManager<TDocument>> SetupIndexManager<TDocument>(Mock<IMongoCollection<TDocument>> collection, string indexName)
|
||||
{
|
||||
var indexManager = MockOf<IMongoIndexManager<TDocument>>();
|
||||
indexManager
|
||||
.Setup(
|
||||
x => x.CreateOneAsync(
|
||||
It.IsAny<CreateIndexModel<TDocument>>(),
|
||||
It.IsAny<CreateOneIndexOptions>(),
|
||||
It.IsAny<CancellationToken>()))
|
||||
.ReturnsAsync(indexName);
|
||||
|
||||
collection
|
||||
.SetupGet(x => x.Indexes)
|
||||
.Returns(indexManager.Object);
|
||||
|
||||
return indexManager;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,129 @@
|
||||
using System;
|
||||
using System.Linq;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using AutoFixture;
|
||||
using CoreUnitTests.Infrastructure.Model;
|
||||
using FluentAssertions;
|
||||
using MongoDB.Bson;
|
||||
using MongoDB.Driver;
|
||||
using MongoDbGenericRepository;
|
||||
using Moq;
|
||||
using Xunit;
|
||||
|
||||
namespace CoreUnitTests.DataAccessTests.MongoDbIndexHandlerTests;
|
||||
|
||||
public class GetIndexNamesAsyncTests : BaseIndexTests
|
||||
{
|
||||
[Fact]
|
||||
public async Task WithNoParameters_ReturnsAllIndexNames()
|
||||
{
|
||||
// Arrange
|
||||
var indexNames = Fixture.CreateMany<string>().ToList();
|
||||
var indexes = indexNames.Select(x => new BsonDocument {{"name", x}}).ToList();
|
||||
|
||||
var collection = MockOf<IMongoCollection<TestDocument>>();
|
||||
|
||||
var context = MockOf<IMongoDbContext>();
|
||||
context
|
||||
.Setup(x => x.GetCollection<TestDocument>(It.IsAny<string>()))
|
||||
.Returns(collection.Object);
|
||||
|
||||
var (cursor, manager) = SetupIndexes(indexes, collection);
|
||||
|
||||
// Act
|
||||
var result = await Sut.GetIndexesNamesAsync<TestDocument, Guid>();
|
||||
|
||||
// Assert
|
||||
result.Should().NotBeNull();
|
||||
result.Should().BeEquivalentTo(indexNames);
|
||||
context.Verify(x => x.GetCollection<TestDocument>(null));
|
||||
manager.Verify(x => x.ListAsync(CancellationToken.None));
|
||||
cursor.Verify(x => x.MoveNextAsync(CancellationToken.None));
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task WithPartitionKey_ReturnsAllIndexNames()
|
||||
{
|
||||
// Arrange
|
||||
var partitionKey = Fixture.Create<string>();
|
||||
var indexNames = Fixture.CreateMany<string>().ToList();
|
||||
var indexes = indexNames.Select(x => new BsonDocument {{"name", x}}).ToList();
|
||||
|
||||
var collection = MockOf<IMongoCollection<TestDocument>>();
|
||||
|
||||
var context = MockOf<IMongoDbContext>();
|
||||
context
|
||||
.Setup(x => x.GetCollection<TestDocument>(It.IsAny<string>()))
|
||||
.Returns(collection.Object);
|
||||
|
||||
var (cursor, manager) = SetupIndexes(indexes, collection);
|
||||
|
||||
// Act
|
||||
var result = await Sut.GetIndexesNamesAsync<TestDocument, Guid>(partitionKey);
|
||||
|
||||
// Assert
|
||||
result.Should().NotBeNull();
|
||||
result.Should().BeEquivalentTo(indexNames);
|
||||
context.Verify(x => x.GetCollection<TestDocument>(partitionKey));
|
||||
manager.Verify(x => x.ListAsync(CancellationToken.None));
|
||||
cursor.Verify(x => x.MoveNextAsync(CancellationToken.None));
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task WithCancellationToken_ReturnsAllIndexNames()
|
||||
{
|
||||
// Arrange
|
||||
var token = new CancellationToken();
|
||||
var indexNames = Fixture.CreateMany<string>().ToList();
|
||||
var indexes = indexNames.Select(x => new BsonDocument {{"name", x}}).ToList();
|
||||
|
||||
var collection = MockOf<IMongoCollection<TestDocument>>();
|
||||
|
||||
var context = MockOf<IMongoDbContext>();
|
||||
context
|
||||
.Setup(x => x.GetCollection<TestDocument>(It.IsAny<string>()))
|
||||
.Returns(collection.Object);
|
||||
|
||||
var (cursor, manager) = SetupIndexes(indexes, collection);
|
||||
|
||||
// Act
|
||||
var result = await Sut.GetIndexesNamesAsync<TestDocument, Guid>(cancellationToken:token);
|
||||
|
||||
// Assert
|
||||
result.Should().NotBeNull();
|
||||
result.Should().BeEquivalentTo(indexNames);
|
||||
context.Verify(x => x.GetCollection<TestDocument>(null));
|
||||
manager.Verify(x => x.ListAsync(token));
|
||||
cursor.Verify(x => x.MoveNextAsync(token));
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task WithPartitionKeyCancellationToken_ReturnsAllIndexNames()
|
||||
{
|
||||
// Arrange
|
||||
var partitionKey = Fixture.Create<string>();
|
||||
var token = new CancellationToken();
|
||||
var indexNames = Fixture.CreateMany<string>().ToList();
|
||||
var indexes = indexNames.Select(x => new BsonDocument {{"name", x}}).ToList();
|
||||
|
||||
var collection = MockOf<IMongoCollection<TestDocument>>();
|
||||
|
||||
var context = MockOf<IMongoDbContext>();
|
||||
context
|
||||
.Setup(x => x.GetCollection<TestDocument>(It.IsAny<string>()))
|
||||
.Returns(collection.Object);
|
||||
|
||||
var (cursor, manager) = SetupIndexes(indexes, collection);
|
||||
|
||||
// Act
|
||||
var result = await Sut.GetIndexesNamesAsync<TestDocument, Guid>(partitionKey, token);
|
||||
|
||||
// Assert
|
||||
result.Should().NotBeNull();
|
||||
result.Should().BeEquivalentTo(indexNames);
|
||||
context.Verify(x => x.GetCollection<TestDocument>(partitionKey));
|
||||
manager.Verify(x => x.ListAsync(token));
|
||||
cursor.Verify(x => x.MoveNextAsync(token));
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user