Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ inline fun ActivityOptions.copy(
* @see ActivityOptions.Builder.setRetryOptions
* @see ActivityOptions.getRetryOptions
*/
inline fun ActivityOptions.Builder.setRetryOptions(
inline fun @TemporalDsl ActivityOptions.Builder.setRetryOptions(
retryOptions: @TemporalDsl RetryOptions.Builder.() -> Unit
) {
setRetryOptions(RetryOptions(retryOptions))
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ inline fun WorkflowOptions.copy(
* @see WorkflowOptions.Builder.setRetryOptions
* @see WorkflowOptions.getRetryOptions
*/
inline fun WorkflowOptions.Builder.setRetryOptions(
inline fun @TemporalDsl WorkflowOptions.Builder.setRetryOptions(
retryOptions: @TemporalDsl RetryOptions.Builder.() -> Unit
) {
setRetryOptions(RetryOptions(retryOptions))
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ inline fun ChildWorkflowOptions.copy(
* @see ChildWorkflowOptions.Builder.setRetryOptions
* @see ChildWorkflowOptions.getRetryOptions
*/
inline fun ChildWorkflowOptions.Builder.setRetryOptions(
inline fun @TemporalDsl ChildWorkflowOptions.Builder.setRetryOptions(
retryOptions: @TemporalDsl RetryOptions.Builder.() -> Unit
) {
setRetryOptions(RetryOptions(retryOptions))
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,27 @@ class ActivityOptionsExtTest {
assertEquals(builderActivityOptions, dslActivityOptions)
}

@Test
fun `setRetryOptions DSL extension should work on ActivityOptions builder directly`() {
val builder = ActivityOptions.newBuilder().setTaskQueue("TestQueue")
builder.setRetryOptions {
setInitialInterval(Duration.ofMillis(50))
setMaximumAttempts(5)
}

val expected = ActivityOptions.newBuilder()
.setTaskQueue("TestQueue")
.setRetryOptions(
RetryOptions.newBuilder()
.setInitialInterval(Duration.ofMillis(50))
.setMaximumAttempts(5)
.build()
)
.build()

assertEquals(expected, builder.build())
}

@Test
fun `ActivityOptions copy() DSL should merge override options`() {
val sourceOptions = ActivityOptions {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,27 @@ class WorkflowOptionsExtTest {
assertEquals(builderOptions, dslOptions)
}

@Test
fun `setRetryOptions DSL extension should work on WorkflowOptions builder directly`() {
val builder = WorkflowOptions.newBuilder().setTaskQueue("TestQueue")
builder.setRetryOptions {
setInitialInterval(Duration.ofMillis(50))
setMaximumAttempts(5)
}

val expected = WorkflowOptions.newBuilder()
.setTaskQueue("TestQueue")
.setRetryOptions(
RetryOptions.newBuilder()
.setInitialInterval(Duration.ofMillis(50))
.setMaximumAttempts(5)
.build()
)
.build()

assertEquals(expected, builder.build())
}

@Test
fun `WorkflowOptions copy() DSL should merge override options`() {
val sourceOptions = WorkflowOptions {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,27 @@ class ChildWorkflowOptionsExtTest {
assertEquals(builderOptions, dslOptions)
}

@Test
fun `setRetryOptions DSL extension should work on ChildWorkflowOptions builder directly`() {
val builder = ChildWorkflowOptions.newBuilder().setTaskQueue("TestQueue")
builder.setRetryOptions {
setInitialInterval(Duration.ofMillis(50))
setMaximumAttempts(3)
}

val expected = ChildWorkflowOptions.newBuilder()
.setTaskQueue("TestQueue")
.setRetryOptions(
RetryOptions.newBuilder()
.setInitialInterval(Duration.ofMillis(50))
.setMaximumAttempts(3)
.build()
)
.build()

assertEquals(expected, builder.build())
}

@Test
fun `ChildWorkflowOptions copy() DSL should merge override options`() {
val sourceOptions = ChildWorkflowOptions {
Expand Down