diff --git a/src/main/kotlin/org/springframework/data/r2dbc/core/UpdatedRowsFetchSpecExtensions.kt b/src/main/kotlin/org/springframework/data/r2dbc/core/UpdatedRowsFetchSpecExtensions.kt new file mode 100644 index 0000000..b2dba05 --- /dev/null +++ b/src/main/kotlin/org/springframework/data/r2dbc/core/UpdatedRowsFetchSpecExtensions.kt @@ -0,0 +1,26 @@ +/* + * Copyright 2018-2019 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. + * You may obtain a copy of the License at + * + * https://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.springframework.data.r2dbc.core + +import kotlinx.coroutines.reactive.awaitSingle + +/** + * Coroutines variant of [UpdatedRowsFetchSpec.rowsUpdated]. + * + * @author Fred Montariol + */ +suspend fun UpdatedRowsFetchSpec.awaitRowsUpdated(): Int = + rowsUpdated().awaitSingle() diff --git a/src/test/kotlin/org/springframework/data/r2dbc/core/UpdatedRowsFetchSpecExtensionsTests.kt b/src/test/kotlin/org/springframework/data/r2dbc/core/UpdatedRowsFetchSpecExtensionsTests.kt new file mode 100644 index 0000000..3e1dda7 --- /dev/null +++ b/src/test/kotlin/org/springframework/data/r2dbc/core/UpdatedRowsFetchSpecExtensionsTests.kt @@ -0,0 +1,47 @@ +/* + * Copyright 2018-2019 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. + * You may obtain a copy of the License at + * + * https://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.springframework.data.r2dbc.core + +import io.mockk.every +import io.mockk.mockk +import io.mockk.verify +import kotlinx.coroutines.runBlocking +import org.assertj.core.api.Assertions.assertThat +import org.junit.Test +import reactor.core.publisher.Mono + +/** + * Unit tests for [UpdatedRowsFetchSpec] extensions. + * + * @author Fred Montariol + */ +class UpdatedRowsFetchSpecExtensionsTests { + + @Test // gh-212 + fun awaitRowsUpdatedWithValue() { + + val spec = mockk() + every { spec.rowsUpdated() } returns Mono.just(42) + + runBlocking { + assertThat(spec.awaitRowsUpdated()).isEqualTo(42) + } + + verify { + spec.rowsUpdated() + } + } +}