Как написать JUnit тест к методу который лазит в БД?

Question

[SankaSanka]

Есть простой метод добавляющий акк в базу:

@Override
    public boolean addAccount(String currUserEmail, String account, String info) throws SQLException {
        boolean res = false;
        PreparedStatement ps = null;
        ResultSet rs = null;
        Connection con = null;
        try {
            con = ConnectionPool.getInstance().getConnection();
            ps = (PreparedStatement) con.prepareStatement(SQL_ADD_CARD);
            ps.setString(1, currUserEmail);
            ps.setString(2, account);
            ps.setString(3, info);
            ps.executeUpdate();
 
            res = true;
 
        } finally {
            ConnectionPool.close(rs);
            ConnectionPool.close(ps);
            ConnectionPool.close(con);
        }
        return res;
 
    }

Пытаюсь к нему написать тест:

private FactoryDAO factoryDAO = FactoryDAO.getInstance();  
 
@Test
    public void testaddAccount() throws SQLException {
        String currUserEmail = "mail";
        String account = "account";
        String info = "info";
        
        assertEquals(true, factoryDAO.getAccountDao().addAccount(currUserEmail, account, info));
    }

Но он естественно не работает, ибо доступ в БД через пулл, который работает только когда запущен томкат.
Как правильно сделать, чтобы он отработал с базой данных?

Правильно ли то, что я пытаюсь сделать в принципе? (Это у меня вроде курсового, должно быть все по-человечески).

Answer 1

[Дмитрий Шицков]

Приходят в голову такие варианты:

• Если нужен только юнит-тест, то используя PowerMock, замокать методы ConnectionPool. Пример: https://stackoverflow.com/a/18910765
  
• Можно написать интеграционный тест, реализовав тестовый ConnectionPool и запустив реальную базу данных в докере, используя Java Testcontainers https://www.testcontainers.org/
  

Comments

[SankaSanka]

спасибо. я правильно понимаю если первый вариант то в базу оно совсем не полезет.
но я тогда не понимаю что собственно я тестирую.

а по второму- можно пожалуйста какой-нибудь пример в коде для совсем зеленых. Заранее спасибо огромное)

[Дмитрий Шицков]

SankaSanka, в первом варианте в базу не пойдет, и не надо. Вы проверите работу метода addAccount. Что является работой этого метода? Вызов нескольких других методов и передача в них параметров, а так же возврат результата. Если вызванные методы будут моками, то как минимум можно будет в тесте проверить, что они были вызваны с правильными параметрами. Так же, проверить результат работы addAccount. Этого более чем достаточно для того, чтобы убедиться, что метод addAccount работает корректно.

Второй вариант является уже интеграционным тестом, так как проводится не только проверка работы самого метода, но и прочих компонентов, которые вы не разрабатывали(ConnectionPool, Connection, Database и прочие), а результат работы метода вовсе будет оцениваться по факту записи в БД. Данный вариант тестирования сложнее и я не думаю, что в данном случае он вам вообще необходим. И тут кратким примером не обойтись.

[SankaSanka]

Дмитрий Шицков, спасибо, теперь понятнее стало. я просто в тестах совсем впервые и о существовании моков узнала позавчера).
нет ли у вас готового примера для вот этого

в первом варианте в базу не пойдет, и не надо. Вы проверите работу метода addAccount. Что является работой этого метода? Вызов нескольких других методов и передача в них параметров, а так же возврат результата. Если вызванные методы будут моками, то как минимум можно будет в тесте проверить, что они были вызваны с правильными параметрами. Так же, проверить результат работы addAccount. Этого более чем достаточно для того, чтобы убедиться, что метод addAccount работает корректно.

по какому-нибудь похожему методу. стаковерфлоу выше не помог там даже методы нереальные какие-то которых нет в PowerMock
у меня явно не хватает уровня для понимания всего этого. а по работающему примеру я бы справилась. Заранее спасибо)

[Дмитрий Шицков]

SankaSanka, Код написан в блокноте по документации и не факт что заработает сразу. Необходимы библиотеки Mockito и PowerMock. Не уверен, может естьс пособ обойтись без Mockito. В моем представлении тест должен быть таким

@RunWith(PowerMockRunner.class)
@PrepareForTest(ConnectionPool.class)
public class TestClass {
    private FactoryDAO factoryDAO = FactoryDAO.getInstance();  

    @Test
    public void testaddAccount() throws SQLException {
        String currUserEmail = "mail";
        String account = "account";
        String info = "info";

        PreparedStatement statementMock = mock(PreparedStatement.class);
        Connection connectionMock = mock(Connection.class);
        ConnectionPool connectionPoolMock = mock(ConnectionPool.class);
        
        when(connectionPoolMock.getConnection()).thenReturn(statementMock);
        when(connectionMock.prepareStatement(SQL_ADD_CARD)).thenReturn(statementMock);
        
        mockStatic(ConnectionPool.class);
        expect(ConnectionPool.getInstance()).andReturn(connectionPoolMock);
        
        assertEquals(true, factoryDAO.getAccountDao().addAccount(currUserEmail, account, info));

        verify(statementMock).setString(1, currUserEmail);
        verify(statementMock).setString(2, account);
        verify(statementMock).setString(3, info);
    }
}

[SankaSanka]

Дмитрий Шицков, спасибо огромное :) Дай вам бог здоровья!

вот итоговый результат может пригодится кому-то

@Test
    public void testaddAccount() throws SQLException {
        String currUserEmail = "mail";
        String account = "account";
        String info = "info";

        PreparedStatement statementMock = mock(PreparedStatement.class);
        Connection connectionMock = mock(Connection.class);
        ConnectionPool connectionPoolMock = mock(ConnectionPool.class);
        
        when(connectionPoolMock.getConnection()).thenReturn(connectionMock);
        when(connectionMock.prepareStatement(SQL_ADD_CARD)).thenReturn(statementMock);
        
        PowerMockito.mockStatic(ConnectionPool.class);
        when(ConnectionPool.getInstance()).thenReturn(connectionPoolMock);
        
        assertEquals(true, factoryDAO.getAccountDao().addAccount(currUserEmail, account, info));

        verify(statementMock).setString(1, currUserEmail);
        verify(statementMock).setString(2, account);
        verify(statementMock).setString(3, info);
    }