使用junit和mockito测试jpql查询

import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnitRunner;

import javax.persistence.EntityManager;
import javax.persistence.Query;
import java.util.ArrayList;
import java.util.List;

import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.when;

@RunWith(MockitoJUnitRunner.class)
public class JpqlQueryTest {

    @Mock
    private EntityManager entityManager;

    @InjectMocks
    private JpqlQueryService jpqlQueryService;

    @Test
    public void testJpqlQuery() {
        // 模拟查询结果
        List<String> mockResult = new ArrayList<>();
        mockResult.add("result1");
        mockResult.add("result2");

        // 模拟JPQL查询
        Query mockQuery = Mockito.mock(Query.class);
        when(mockQuery.getResultList()).thenReturn(mockResult);

        // 模拟EntityManager
        when(entityManager.createQuery(Mockito.anyString())).thenReturn(mockQuery);

        // 调用被测试的方法
        List<String> result = jpqlQueryService.executeQuery("SELECT name FROM User");

        // 验证查询结果
        assertEquals(2, result.size());
        assertEquals("result1", result.get(0));
        assertEquals("result2", result.get(1));
    }
}