Skip to content

Commit

Permalink
add test
Browse files Browse the repository at this point in the history
  • Loading branch information
Quinn-With-Two-Ns committed Sep 25, 2024
1 parent 860f6d1 commit 5c8ff25
Show file tree
Hide file tree
Showing 2 changed files with 144 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,6 @@ private static String getMessage(WorkflowExecution execution, String workflowTyp
+ execution.getWorkflowId()
+ "', runId='"
+ execution.getRunId()
+ (workflowType == null ? "" : "', workflowType='" + workflowType + '\'')
+ '}';
+ (workflowType == null ? "" : "', workflowType='" + workflowType + '\'');
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,143 @@
/*
* Copyright (C) 2022 Temporal Technologies, Inc. All Rights Reserved.
*
* Copyright (C) 2012-2016 Amazon.com, Inc. or its affiliates. All Rights Reserved.
*
* Modifications copyright (C) 2017 Uber Technologies, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this material except in compliance with the License.
* You may obtain a copy of the License at
*
* http://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 io.temporal.workflow.nexus;

import static org.junit.Assume.assumeTrue;

import io.nexusrpc.handler.OperationHandler;
import io.nexusrpc.handler.OperationImpl;
import io.nexusrpc.handler.ServiceImpl;
import io.temporal.api.common.v1.WorkflowExecution;
import io.temporal.client.WorkflowExecutionAlreadyStarted;
import io.temporal.client.WorkflowFailedException;
import io.temporal.failure.ApplicationFailure;
import io.temporal.failure.NexusOperationFailure;
import io.temporal.failure.TimeoutFailure;
import io.temporal.testing.internal.SDKTestWorkflowRule;
import io.temporal.workflow.*;
import io.temporal.workflow.shared.TestNexusServices;
import io.temporal.workflow.shared.TestWorkflows.TestWorkflow1;
import java.time.Duration;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;

public class OperationFailureConversionTest {

@Rule
public SDKTestWorkflowRule testWorkflowRule =
SDKTestWorkflowRule.newBuilder()
.setWorkflowTypes(TestNexus.class)
.setNexusServiceImplementation(new TestNexusServiceImpl())
.build();

@Before
public void checkRealServer() {
assumeTrue("Test Server always retries these errors", SDKTestWorkflowRule.useExternalService);
}

@Test
public void nexusOperationApplicationFailureNonRetryableFailureConversion() {
TestWorkflow1 workflowStub =
testWorkflowRule.newWorkflowStubTimeoutOptions(TestWorkflow1.class);
WorkflowFailedException exception =
Assert.assertThrows(
WorkflowFailedException.class,
() -> workflowStub.execute("ApplicationFailureNonRetryable"));
Assert.assertTrue(exception.getCause() instanceof NexusOperationFailure);
NexusOperationFailure nexusFailure = (NexusOperationFailure) exception.getCause();
Assert.assertTrue(nexusFailure.getCause() instanceof ApplicationFailure);
ApplicationFailure applicationFailure = (ApplicationFailure) nexusFailure.getCause();
Assert.assertEquals(
"unexpected response status: \"400 Bad Request\": message='failed to call operation', type='TestFailure', nonRetryable=true",
applicationFailure.getOriginalMessage());
}

@Test
public void nexusOperationWorkflowExecutionAlreadyStartedFailureConversion() {
TestWorkflow1 workflowStub =
testWorkflowRule.newWorkflowStubTimeoutOptions(TestWorkflow1.class);
WorkflowFailedException exception =
Assert.assertThrows(
WorkflowFailedException.class,
() -> workflowStub.execute("WorkflowExecutionAlreadyStarted"));
Assert.assertTrue(exception.getCause() instanceof NexusOperationFailure);
NexusOperationFailure nexusFailure = (NexusOperationFailure) exception.getCause();
Assert.assertTrue(nexusFailure.getCause() instanceof ApplicationFailure);
ApplicationFailure applicationFailure = (ApplicationFailure) nexusFailure.getCause();
Assert.assertEquals(
"unexpected response status: \"400 Bad Request\": workflowId='id', runId='runId', workflowType='TestWorkflow'",
applicationFailure.getOriginalMessage());
}

@Test
public void nexusOperationApplicationFailureFailureConversion() {
TestWorkflow1 workflowStub =
testWorkflowRule.newWorkflowStubTimeoutOptions(TestWorkflow1.class);
WorkflowFailedException exception =
Assert.assertThrows(
WorkflowFailedException.class, () -> workflowStub.execute("ApplicationFailure"));
Assert.assertTrue(exception.getCause() instanceof NexusOperationFailure);
NexusOperationFailure nexusFailure = (NexusOperationFailure) exception.getCause();
Assert.assertTrue(nexusFailure.getCause() instanceof TimeoutFailure);
}

public static class TestNexus implements TestWorkflow1 {
@Override
public String execute(String testcase) {
NexusOperationOptions options =
NexusOperationOptions.newBuilder()
.setScheduleToCloseTimeout(Duration.ofSeconds(5))
.build();

NexusServiceOptions serviceOptions =
NexusServiceOptions.newBuilder().setOperationOptions(options).build();
TestNexusServices.TestNexusService1 testNexusService =
Workflow.newNexusServiceStub(TestNexusServices.TestNexusService1.class, serviceOptions);
testNexusService.operation(testcase);
return "fail";
}
}

@ServiceImpl(service = TestNexusServices.TestNexusService1.class)
public class TestNexusServiceImpl {
@OperationImpl
public OperationHandler<String, String> operation() {
return OperationHandler.sync(
(ctx, details, name) -> {
if (name.equals("ApplicationFailure")) {
throw ApplicationFailure.newFailure("failed to call operation", "TestFailure");
} else if (name.equals("ApplicationFailureNonRetryable")) {
throw ApplicationFailure.newNonRetryableFailure(
"failed to call operation", "TestFailure");
} else if (name.equals("WorkflowExecutionAlreadyStarted")) {
throw new WorkflowExecutionAlreadyStarted(
WorkflowExecution.newBuilder().setWorkflowId("id").setRunId("runId").build(),
"TestWorkflow",
new RuntimeException("already started"));
}
Assert.fail();
return "fail";
});
}
}
}

0 comments on commit 5c8ff25

Please sign in to comment.