mirror of https://github.com/abpframework/abp
Merge pull request #5865 from abpframework/liangshiwei/mongodb
Enable MongoDB transaction for unit testpull/5872/head
maliming
5 years ago
committed by
GitHub
commit
d18193c27f
@ -0,0 +1,90 @@
|
||||
using System;
|
||||
using System.Threading.Tasks;
|
||||
using Microsoft.Extensions.DependencyInjection;
|
||||
using Shouldly;
|
||||
using Volo.Abp.Domain.Repositories;
|
||||
using Volo.Abp.TestApp.Domain;
|
||||
using Volo.Abp.TestApp.Testing;
|
||||
using Volo.Abp.Uow;
|
||||
using Xunit;
|
||||
|
||||
namespace Volo.Abp.MongoDB.Transactions
|
||||
{
|
||||
public class Transaction_Tests : TestAppTestBase<AbpMongoDbTestModule>
|
||||
{
|
||||
private readonly IBasicRepository<Person, Guid> _personRepository;
|
||||
private readonly IUnitOfWorkManager _unitOfWorkManager;
|
||||
|
||||
public Transaction_Tests()
|
||||
{
|
||||
_personRepository = GetRequiredService<IBasicRepository<Person, Guid>>();
|
||||
_unitOfWorkManager = GetRequiredService<IUnitOfWorkManager>();
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task Should_Rollback_Transaction_When_An_Exception_Is_Thrown()
|
||||
{
|
||||
var personId = Guid.NewGuid();
|
||||
const string exceptionMessage = "thrown to rollback the transaction!";
|
||||
|
||||
try
|
||||
{
|
||||
await WithUnitOfWorkAsync(new AbpUnitOfWorkOptions { IsTransactional = true }, async () =>
|
||||
{
|
||||
await _personRepository.InsertAsync(new Person(personId, "Adam", 42));
|
||||
throw new Exception(exceptionMessage);
|
||||
});
|
||||
}
|
||||
catch (Exception e) when (e.Message == exceptionMessage)
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
var person = await _personRepository.FindAsync(personId);
|
||||
person.ShouldBeNull();
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task Should_Rollback_Transaction_Manually()
|
||||
{
|
||||
var personId = Guid.NewGuid();
|
||||
|
||||
await WithUnitOfWorkAsync(new AbpUnitOfWorkOptions { IsTransactional = true }, async () =>
|
||||
{
|
||||
_unitOfWorkManager.Current.ShouldNotBeNull();
|
||||
|
||||
await _personRepository.InsertAsync(new Person(personId, "Adam", 42));
|
||||
|
||||
await _unitOfWorkManager.Current.RollbackAsync();
|
||||
});
|
||||
|
||||
var person = await _personRepository.FindAsync(personId);
|
||||
person.ShouldBeNull();
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task Should_Rollback_Transaction_Manually_With_Double_DbContext_Transaction()
|
||||
{
|
||||
var personId = Guid.NewGuid();
|
||||
var bookId = Guid.NewGuid();
|
||||
|
||||
using (var scope = ServiceProvider.CreateScope())
|
||||
{
|
||||
var uowManager = scope.ServiceProvider.GetRequiredService<IUnitOfWorkManager>();
|
||||
|
||||
using (uowManager.Begin(new AbpUnitOfWorkOptions { IsTransactional = true }))
|
||||
{
|
||||
_unitOfWorkManager.Current.ShouldNotBeNull();
|
||||
|
||||
await _personRepository.InsertAsync(new Person(personId, "Adam", 42));
|
||||
|
||||
await _unitOfWorkManager.Current.SaveChangesAsync();
|
||||
|
||||
//Will automatically rollback since not called the Complete!
|
||||
}
|
||||
}
|
||||
|
||||
(await _personRepository.FindAsync(personId)).ShouldBeNull();
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue