java - 在java中使用mockito测试sql查询时出错

Question

我想使用 EasyMock、Mock 或 Mockito 测试 java 中的下一个函数。 我想测试当我调用 leerCantPueblos() 函数时，会执行查询(“SELECT conarpueblos()”)。

private void leerCantPueblos(){
        cantPueblosBD = leerBD.cantidadPueblos();
        try {
            while(cantPueblosBD.next()){
                CANTIDADPUEBLOS = cantPueblosBD.getInt("contarpueblos");
            }
        } catch (SQLException e) {e.printStackTrace();}
    }

该函数调用cantidadPueblos()，即下一个函数。

public ResultSet cantidadPueblos() {
        try {
            sentencia = conexion.createStatement();
            ResultSet cantidadPueblos = sentencia.executeQuery("SELECT contarpueblos()");
            return cantidadPueblos;
        } catch (SQLException e) {}

        return null;
    }

我尝试进行下一个 JUnit 测试，但抛出 nullpointerException

package tests;

import java.sql.Connection;
import java.sql.SQLException;
import java.sql.Statement;

import org.easymock.Mock;
import org.junit.Before;
import org.junit.Test;
import org.mockito.InjectMocks;
import org.mockito.Mockito;
import org.mockito.MockitoAnnotations;

import server.ConexionBD;

public class TestBD { 

    @InjectMocks
        ConexionBD conexionDB;

        @Mock
        Connection conexion;

        @Mock
        Statement sentencia;

        @Before
        public void setup() throws SQLException {
            MockitoAnnotations.initMocks(this);
            //Mockito.initMocks(this);
            Mockito.when(conexion.createStatement()).thenReturn(sentencia);
        }

        @Test
        public void cantidadPueblos_shouldExecuteQuery() throws SQLException {
            conexionDB.cantidadPueblos();

            Mockito.verify(sentencia).executeQuery("SELECT contarpueblos()");
        }
}   

下一行失败:

Mockito.when(conexion.createStatement()).thenReturn(sentencia);

Answer 1

除了 setup 方法的第一行之外，测试看起来不错。

public void setup() throws SQLException {
    sentencia = conexion.createStatement();
    MockitoAnnotations.initMocks(this);
    //Mockito.initMocks(this);
    Mockito.when(conexion.createStatement()).thenReturn(sentencia);
}

当在初始化模拟之前调用 conexion.createStatement() 时，它会抛出 NPE。只需删除 sentencia = conexion.createStatement(); 行，这里是多余的。

<小时/> 更新
结果发现使用了错误的导入。 import org.easymock.Mock 应替换为 import org.mockito.Mock;