root
/
spring-boot
mirror of https://github.com/spring-projects/spring-boot.git
1
0
Fork 0
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Adapt mock to API change

This commit is contained in:
Stephane Nicoll 2022-03-17 10:34:22 +01:00
parent ca16ae0804
commit 1784fc19bb
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spring-boot-project/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/r2dbc/R2dbcTransactionManagerAutoConfigurationTests.java
View File

@ -1,5 +1,5 @@
/*
* Copyright 2012-2020 the original author or authors.
* Copyright 2012-2022 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -18,6 +18,7 @@ package org.springframework.boot.autoconfigure.r2dbc;
import io.r2dbc.spi.Connection;
import io.r2dbc.spi.ConnectionFactory;
import io.r2dbc.spi.TransactionDefinition;
import org.junit.jupiter.api.Test;
import reactor.core.publisher.Mono;
import reactor.test.StepVerifier;
@ -34,6 +35,7 @@ import org.springframework.transaction.reactive.TransactionSynchronizationManage
import org.springframework.transaction.reactive.TransactionalOperator;
import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.BDDMockito.given;
import static org.mockito.Mockito.mock;
@ -77,7 +79,7 @@ class R2dbcTransactionManagerAutoConfigurationTests {
ConnectionFactory connectionFactory = mock(ConnectionFactory.class);
Connection connection = mock(Connection.class);
given(connectionFactory.create()).willAnswer((invocation) -> Mono.just(connection));
given(connection.beginTransaction()).willReturn(Mono.empty());
given(connection.beginTransaction(any(TransactionDefinition.class))).willReturn(Mono.empty());
given(connection.commitTransaction()).willReturn(Mono.empty());
given(connection.close()).willReturn(Mono.empty());
return connectionFactory;