|
1 | 1 | /*
|
2 |
| - * Copyright 2012-2020 the original author or authors. |
| 2 | + * Copyright 2012-2022 the original author or authors. |
3 | 3 | *
|
4 | 4 | * Licensed under the Apache License, Version 2.0 (the "License");
|
5 | 5 | * you may not use this file except in compliance with the License.
|
|
18 | 18 |
|
19 | 19 | import io.r2dbc.spi.Connection;
|
20 | 20 | import io.r2dbc.spi.ConnectionFactory;
|
| 21 | +import io.r2dbc.spi.TransactionDefinition; |
21 | 22 | import org.junit.jupiter.api.Test;
|
22 | 23 | import reactor.core.publisher.Mono;
|
23 | 24 | import reactor.test.StepVerifier;
|
|
34 | 35 | import org.springframework.transaction.reactive.TransactionalOperator;
|
35 | 36 |
|
36 | 37 | import static org.assertj.core.api.Assertions.assertThat;
|
| 38 | +import static org.mockito.ArgumentMatchers.any; |
37 | 39 | import static org.mockito.BDDMockito.given;
|
38 | 40 | import static org.mockito.Mockito.mock;
|
39 | 41 |
|
@@ -77,7 +79,7 @@ ConnectionFactory connectionFactory() {
|
77 | 79 | ConnectionFactory connectionFactory = mock(ConnectionFactory.class);
|
78 | 80 | Connection connection = mock(Connection.class);
|
79 | 81 | given(connectionFactory.create()).willAnswer((invocation) -> Mono.just(connection));
|
80 |
| - given(connection.beginTransaction()).willReturn(Mono.empty()); |
| 82 | + given(connection.beginTransaction(any(TransactionDefinition.class))).willReturn(Mono.empty()); |
81 | 83 | given(connection.commitTransaction()).willReturn(Mono.empty());
|
82 | 84 | given(connection.close()).willReturn(Mono.empty());
|
83 | 85 | return connectionFactory;
|
|
0 commit comments