测试休眠环境

Question

我想编写有关修订的测试。在控制台中，我看到了 Hibernate 的更新调用，但没有插入 AUD-Table。

测试方法：

@DataJpaTest
class JPAHistoryTest {

    @Test
    public void test() {
        def entity = // create Entity
        def entity2 = // create same Entity
        entity2.setPrice(entity2.getPrice() + 10)
        entity2.setLastUpdate(entity2.lastUpdate.plusSeconds(10))

        service.save(entity)
        service.save(entity2)
        repository.flush() // Hibernate updates changes

        assert repository.findRevisions(entity.id).content.empty == false // FAIL!
    }
}

我的实体看起来像：

@Entity
@Audited
class Entity {
    @Id @GeneratedValue Long id
    @Column(nullable = false) BigDecimal price
}

非常感谢。

Answer 1

我发现我保留@DataJpaTest 并添加@Transactional(propagation = NOT_SUPPORTED) 以确保测试方法不会启动事务。因为如果它们将在事务中运行，那么当测试关闭事务时，将写入 envers 历史条目。

@RunWith(SpringRunner)
@DataJpaTest
@Transactional(propagation = NOT_SUPPORTED)
class JPAHistoryTest {
    @After
    void after() {
        repository.deleteAll()
    }

    @Test
    public void testTwoInsert() {
        def entity1 = // ...
        def entity2 = // ...

        sut.save(entity1 )
        sut.save(entity2 )

        assert repository.findRevisions(entity1.id).content.size() == 1
        assert repository.findRevisions(entity2.id).content.size() == 1
    }
}

Answer 2

我发现了和你一样的问题并尝试了@michael-hegner 的答案，但是我使用了TestEntityManager 类，所以当没有交易时我无法获得EntityManager（propagation 设置为NOT_SUPPORTED）。

在我的情况下，解决方案是从 TestEntityManager 手动提交事务，首先保存实体和更改，然后再查询修订。

这是一个测试类：

@DataJpaTest
class UserRepositoryTest {

    @Autowired
    private TestEntityManager testEntityManager;

    @Test
    public void shouldBeAudited() {
        User user = getTestUser();
        testEntityManager.persistAndFlush(user);

        user.setPassword("tset");
        testEntityManager.merge(user);

        // This line is critical here to commit transaction and trigger audit logs
        testEntityManager.getEntityManager().getTransaction().commit();

        // With propagation = NOT_SUPPORTED this doesn't work: testEntityManager.getEntityManager()
        AuditReader auditReader = AuditReaderFactory.get(testEntityManager.getEntityManager());
        List revisions = auditReader.createQuery()
                .forRevisionsOfEntity(User.class, true)
                .add(AuditEntity.id().eq(user.getId()))
                .getResultList();

        assertEquals(1, revisions.size());
    }

    private User getTestUser() {
        User user = new User();
        user.setUsername("test");
        user.setEmail("test");
        user.setPassword("test");
        return user;
    }

}

在测试之后手动删除用户可能是强制性的，因为事务已提交，而在其他测试中它可能会导致一些问题。

Answer 3

老实说，我在这里不遵循您的命名约定。通常，service 方法调用由生产代码中的一些事务上下文包装。

@Service
public class Service1 {
  @Transactional
  public void save(Object object) {
  }
}

@Service
public class Service2 {
  @Transactional
  public void save(Object object) {
  }
}

对于需要在单个事务上下文中绑定两个服务方法调用的情况，您可以通过另一个服务调用来包装这两个单独的服务。

@Service
public class ServiceWrapper {
  @Autowired
  private Service1 service1;
  @Autowired
  private Service2 service2;

  @Transactional
  public void save(Object object1, Object object2) {
    service1.save( object1 );
    service2.save( object2 );
  }
}

重要的是要指出这个ServiceWrapper 不一定必须存在于您的生产代码中。该 bean 可能是您专门为测试而创建的。关键是这个 bean 以非常类似于 Spring 的方式构建，以满足您需要的事务边界。

现在在您的测试中，您可以使用 repository 来查找实体，因为包装器内的事务提交数据，从而可以按预期在 Envers 审计表中进行查找。

@Test
public void testSomething() {
  Object entity1 = //
  Object entity2 = //
  serviceWrapper.save( entity1, entity2 );

  // now use the repository to find the entity revision
}

Answer 4

如果您不想像 Michael Hegner 的回答那样更改整个测试类的事务行为，您可以执行以下操作：

@RunWith(SpringRunner)
@DataJpaTest
class JPAHistoryTest {

    @Autowired
    private MyRepository repository;
    @Autowired
    private PlatformTransactionManager platformTransactionManager;

    @After
    void after() {
        repository.deleteAll()
    }

    @Test
    public void testTwoInsert() {
        TransactionTemplate tx = new TransactionTemplate(platformTransactionManager);
        tx.setPropagationBehavior(TransactionDefinition.PROPAGATION_NOT_SUPPORTED);

        def entity1 = // ...
        def entity2 = // ...

        tx.execute(status -> repository.save(entity1))
        tx.execute(status -> repository.save(entity2))

        assert repository.findRevisions(entity1.id).content.size() == 1
        assert repository.findRevisions(entity2.id).content.size() == 1
    }
}

因此，使用PROPAGATION_NOT_SUPPORTED 创建一个TransactionTemplate 并使用该交易模板进行保存。这也会触发正在保存的审计信息。