Mockito издевается над базой звонков - PullRequest
Новая
       16

Mockito издевается над базой звонков

0 голосов
/ 28 сентября 2019

У меня есть класс, который принимает URL базы данных и имя пользователя, подключается к базе данных и выполняет запрос, присваивая результаты хэш-карте. 

import java.sql.*;
import java.util.Map;

public class MyClass {
    private final String dbUrl;
    private final String username;

    MyClass(final String dbUrl, final String username) {
        this.dbUrl = dbUrl;
        this.username = username;
    }

    public void queryDatabase(final Map<Integer, String> userIdToName) {
        final String query = "select user_id, name from users";

        try (final Connection connection = DriverManager.getConnection(dbUrl, username, null);
             final PreparedStatement ps = connection.prepareStatement(query);
             final ResultSet rs = ps.executeQuery()) {
            while (rs.next()) {
                final int userId = rs.getInt(1);
                final String name = rs.getString(2);
                userIdToName.put(userId, name);
            }
        } catch (final SQLException e) {
            System.out.println(e.toString());
        }
    }
}

Я пытаюсь использовать Mockito длямакет queryDatabase().Я добавил PowerMockito для макета DriverManager.

Мой тестовый класс: 

import org.junit.Before;
import org.junit.jupiter.api.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
import java.sql.*;
import java.util.HashMap;
import java.util.Map;
import static org.mockito.ArgumentMatchers.*;
import static org.mockito.Mockito.mock;
import static org.powermock.api.mockito.PowerMockito.mockStatic;
import static org.powermock.api.mockito.PowerMockito.when;

@RunWith(PowerMockRunner.class)
@PrepareForTest(DriverManager.class)
class MyClassTest {
    @Mock private Connection mockConnection = mock(Connection.class);
    @Mock private PreparedStatement mockStatement = mock(PreparedStatement.class);
    @Mock private ResultSet resultSet = mock(ResultSet.class);
    @InjectMocks private static MyClass myClass = new MyClass("test_url", "test_username");

    @Before
    public void setUp() throws Exception {
        mockStatic(DriverManager.class);
        when(DriverManager.getConnection(anyString(), anyString(), null)).thenReturn(mockConnection);
    }

    @Test
    void testQueryDatabase() throws Exception {
        when(mockConnection.prepareStatement(anyString())).thenReturn(mockStatement);
        when(mockStatement.executeQuery(anyString())).thenReturn(resultSet);
        when(resultSet.getInt(Mockito.anyInt())).thenReturn(1);
        when(resultSet.getString(anyInt())).thenReturn("john smith");

        final Map<Integer, String> useridToUser = new HashMap<>();
        myClass.queryDatabase(useridToUser);
        System.out.println(useridToUser);
    }
}

Результат выполнения моего теста: 

java.sql.SQLException: No suitable driver found for test_url
{}

Я ожидаю: 

{1=john smith}

Как я могу правильно издеваться над этим?

Я прочитал, чтобы избежать PowerMock, за исключением случаев, когда это абсолютно необходимо, поэтому я также очень открыт для предложений по рефакторингу, чтобы избежать, если это возможно.

1 Ответ

0 голосов
/ 28 сентября 2019

Вы можете попробовать сделать следующее: 

class MyClassTest extends PowerMockTestCase {

Добавить зависимости: 

    <dependency>
        <groupId>org.powermock.tests</groupId>
        <artifactId>powermock-tests-utils</artifactId>
        <version>1.6.6</version>
        <scope>test</scope>
    </dependency>
    <dependency>
        <groupId>org.powermock</groupId>
        <artifactId>powermock-module-testng</artifactId>
        <version>1.4.9</version>
        <scope>test</scope>
    </dependency>

Также, вероятно, вам нужен initMock в настройке: 

@Before
public void setUp() throws Exception {
    mockStatic(DriverManager.class);
    initMocks(this);
...