Извлеките поля переменных Consumer, имитируя выражение Lambda

Ваша лямбда не может быть преобразована в Consumer<Connection>, она должна соответствовать типу SAM, ожидаемому Session.doWork, а именно Work

doAnswer((invocation) -> {
    Work jdbcWork = invocation.getArgument(0);
    jdbcWork.execute(connection);
    return null;
}).when(session).doWork(any());

Примечание: session.doWork возвращает void, поэтому я использовал doAnswer — когда, а не когда — thenAnswer.

В общем, вы не можете выполнить кастинг между двумя разными интерфейсами SAM, даже если их структура идентична:

interface StringConsumer1 {
    void execute(String s1);
}

interface StringConsumer2 {
    void execute(String s1);
}

@Test
void testLambdaConversionFails() {
    Throwable thrown = catchThrowable(() -> {
        StringConsumer1 l1 = (aaa) -> System.out.println(aaa);
        StringConsumer2 l2 = (StringConsumer2) l1;
    });

    assertThat(thrown).isInstanceOf(ClassCastException.class);

}

В вашем случае Consumer<Connection> не идентичен Work (название метода и проверенные исключения различаются)

Код немного сложно протестировать, поскольку существует анонимная реализация Work, которая вызывает session.doWork. Аргументом этого метода также является имитируемый connection класс, и все зависит от этого макетного connection класса.

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

Класс, который нужно протестировать;

public class MainClass {

    @Autowired
    private EntityManager entityManager;

    public void executeQuery() {
        Session session = entityManager.unwrap(Session.class);

        try {
            session.doWork(connection -> {
                System.out.println("calling the lambda implementation");
                PreparedStatement preparedStatement = connection.prepareStatement("update some_table set name = ? where id = ? ");
                preparedStatement.setString(1, "new_name");
                preparedStatement.setLong(2, 1L);
                preparedStatement.execute();
            });
        } catch (Exception ex) {
            System.out.println("Error occurred");
        } finally {
            entityManager.flush();
            entityManager.close();
        }
    }
}

Тест;

@ExtendWith(MockitoExtension.class)
class TestHibernateWork {

    @Mock
    EntityManager entityManager;

    @InjectMocks
    MainClass underTest;

    @Test
    void process() {
        Session session = mock(Session.class);
        when(entityManager.unwrap(any())).thenReturn(session);

        // this will help capture the real anonymous implementation of Work; which is the lambda expression
        ArgumentCaptor<Work> workArgumentCaptor = ArgumentCaptor.forClass(Work.class);
        
        // Capture the lambda when doWork is called
        doNothing().when(session).doWork(workArgumentCaptor.capture());

        // call the method to test the lambda
        underTest.executeQuery();

        // get the captured implementation of Work, which is the lambda expression.
        Work work = workArgumentCaptor.getValue();

        try {
            Connection connection = mock(Connection.class);
            PreparedStatement preparedStatement = mock(PreparedStatement.class);
            when(connection.prepareStatement(any())).thenReturn(preparedStatement);

            // call the real implementation of work captured previously using the mock connection
            work.execute(connection);

            // verify the calls made on prepareStatement
            verify(preparedStatement).setString(1, "new_name");
            verify(preparedStatement).setLong(2, 1L);
            verify(preparedStatement).execute();
        } catch (SQLException e) {
            throw new RuntimeException(e);
        }
    }
}