模拟 dao 在保存 object 时返回 false

Question

我是测试驱动开发的新手。 我正在尝试测试 object 是否已成功保存到我的数据库中。 到目前为止我的代码：

@ExtendWith(MockitoExtension.class)
public class ParkingDataBaseIT {
    @Mock
    private TicketDAO mockTicketDAO;
    
    @Mock
    public static DataBaseTestConfig mockDataBaseTestConfig = new DataBaseTestConfig();
    
    @Mock
    private static Connection mockConnection;

    @BeforeEach
    public void setup() throws SQLException, ClassNotFoundException {
        // Connection attempt.
        when(mockDataBaseTestConfig.getConnection()).thenReturn(mockConnection);
    }

    @AfterAll
    public static void tearDown() {
        // ...
    }

    @DisplayName("Tests if saving a ticket is successful.")
    @Test
    public void testParkingACarAlpha() throws SQLException, ClassNotFoundException {
        // Given.
        Date inTime = new Date();
        Ticket ticket = new Ticket();
        
        ticket.setVehicleRegNumber("ABCDEF");
        ticket.setPrice(0);
        ticket.setInTime(inTime);
        ticket.setOutTime(null);
        
        // When.
        // MockDao always returning false when saving.
        final boolean saved = mockTicketDAO.saveTicket(ticket, mockDataBaseTestConfig.getConnection());
        
        // Then.
        assertEquals(true, saved);
        assertThat(saved).isEqualTo(true);
    }
}

这是与 MySQL 建立连接的方法：

public Connection getConnection() throws ClassNotFoundException, SQLException {
    Class.forName("com.mysql.cj.jdbc.Driver");
    return DriverManager.getConnection("jdbc:mysql://localhost:3306/testing", "root", "");
}

最后，这就是我保存机票的方式：

public boolean saveTicket(Ticket ticket, Connection con) {
    try {
        PreparedStatement ps = con.prepareStatement("insert into ticket(vehicleRegNumber, price, inTime, outTime) values(?, ?, ?, ?, ?)");
        
        ps.setInt(1, ticket.getParkingSpot().getId());
        ps.setString(2, ticket.getVehicleRegNumber());
        ps.setDouble(3, ticket.getPrice());
        ps.setTimestamp(4, new Timestamp(ticket.getInTime().getTime()));
        ps.setTimestamp(5, (ticket.getOutTime() == null) ? null: (new Timestamp(ticket.getOutTime().getTime())));
        
        return ps.execute();
    } catch (Exception e) {
        logger.error("Cannot fetch available slot.", e);
    } finally {
        dataBaseConfig.closeConnection(con);
    }
    
    return false;
}

使用 MySQL 的请求有效，但在保存 object 时，我从 ps.execute() 得到一个错误； 而不是真的，但我肯定做错了什么。

我注意到saveTicket(Ticket ticket, Connection con)中的ps是 null（无法调用“java.sql.PreparedStatement.setInt(int, int)”，因为“ps”为空），但是当我 sysout 值时con具有 mockConnection。

Answer 1

由于mockDataBaseTestConfig.getConnection()总是返回 null，显然是行不通的。

审查后，我只通过 mocking dao 使其工作（注意测试已重命名为 testSavingTicket_shouldReturnTrue）。

private Ticket t;

@Test
public void testSavingTicket_shouldReturnTrue() throws SQLException, ClassNotFoundException {
    // Given.
    TicketDAO mockTicketDao = mock(TicketDAO.class);
    
    t = new Ticket();
    
    t.setVehicleRegNumber("ABCDEF");
    t.setPrice(0);
    t.setInTime(new Date());
    t.setOutTime(null);
    
    when(mockTicketDao.saveTicket(t)).thenReturn(true);
    
    // When.
    final boolean saved = mockTicketDao.saveTicket(t);
    
    // Then.
    assertEquals(true, saved);
    assertThat(saved).isEqualTo(true);
}