如何使用Spring Boot Spring Data JPA单元测试悲观锁定

Question

我想测试FooRepository.lock（）是否工作，有人调用lock（）后，其他人调用应该是等待。

以下是我的最佳尝试，它不起作用。我认为原因是entityManger和fooRepository都参与了同一事务。

如何从另一个事务调用lock（）？或者对悲观锁的单元测试有什么建议？谢谢

Foo存储库测试：

package com.example.demo;

import java.util.UUID;

import javax.persistence.LockModeType;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest;
import org.springframework.boot.test.autoconfigure.orm.jpa.TestEntityManager;
import org.springframework.test.context.junit4.SpringRunner;

@RunWith(SpringRunner.class)
@DataJpaTest
public class FooRepositoryTest {

    @Autowired
    private TestEntityManager entityManager;    

    @Autowired
    private FooRepository fooRepository;

    @Test
    public void lockTest() {
        // given
        Foo foo = new Foo();
        foo.setName("foo-name");

        UUID fooId = fooRepository.save(foo).getFooId();
        entityManager.flush();
        entityManager.clear();

        // when
        Foo savedFoo = fooRepository.findOne(fooId);
        fooRepository.lock(savedFoo);

        // then
        // I want something like this to be lock wait,
        // something to confirm the above fooRepository.lock() work
        entityManager.getEntityManager().lock(savedFoo, LockModeType.PESSIMISTIC_WRITE);
    }

}

傅类：

package com.example.demo;

import java.util.UUID;

import javax.persistence.Entity;
import javax.persistence.GeneratedValue;
import javax.persistence.Id;

@Entity
public class Foo {  
    @Id
    @GeneratedValue
    private UUID fooId;

    private String name;

    public UUID getFooId() {
        return fooId;
    }

    public void setFooId(UUID fooId) {
        this.fooId = fooId;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

}

FooApplication类:

package com.example.demo;

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;

@SpringBootApplication
public class FooApplication {

    public static void main(String[] args) {
        SpringApplication.run(FooApplication.class, args);
    }
}

类 Foo存储库：

package com.example.demo;

import java.util.UUID;

import org.springframework.data.jpa.repository.JpaRepository;

public interface FooRepository extends JpaRepository<Foo, UUID>, FooRepositoryCustom {  
}

FooRepositoryCustom类:

package com.example.demo;

public interface FooRepositoryCustom {  
    public void lock(Foo foo);  
}

类FooRepositoryImpl：

package com.example.demo;

import javax.persistence.EntityManager;
import javax.persistence.LockModeType;

import org.springframework.beans.factory.annotation.Autowired;

public class FooRepositoryImpl implements FooRepositoryCustom {

    @Autowired
    private EntityManager entityManager;

    @Override
    public void lock(Foo foo) {
        entityManager.lock(foo, LockModeType.PESSIMISTIC_WRITE);
    }   
}




Answer 1

你把单元测试弄错了。

您编写单元测试不是为了行使某些第三方框架实现的功能。单元测试是为您的单元准备的！

换句话说：您不需要验证锁定是否按预期工作。您的单位可以：

entityManager.lock(foo, LockModeType.PESSIMISTIC_WRITE);

因此，您唯一可以考虑在这里进行测试的是：确保使用预期的参数调用实体管理器lock（）方法。

意思是：验证你的代码是否使用了你认为应该使用的框架——但是不要去测试其他人的代码！你看——当你的单元测试显示框架是错误的时候，你会怎么做…你不能改变它！（当然，你可以写一个错误报告）

免责声明：在特殊情况下，您可能会假设某个第三方产品存在错误——那么编写一个单元测试来测试这个假设可能非常有用。这样您以后就可以针对该产品的新版本运行单元测试，看看错误是否仍然存在。