Add test for sa

indeedeng · Jun 2, 2023 · ba99106 · ba99106
1 parent 2a7bc8e
commit ba99106
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/src/test/java/io/iworkflow/integ/RpcTest.java b/src/test/java/io/iworkflow/integ/RpcTest.java
@@ -53,6 +53,13 @@ public void testRPCWorkflowFunc1() throws InterruptedException {
         value = client.invokeRPC(rpcStub::testRpcGetDataAttribute);
         Assertions.assertNull(value);
 
+        client.invokeRPC(rpcStub::testRpcSetKeyword, "test-value");
+        value = client.invokeRPC(rpcStub::testRpcGetKeyword);
+        Assertions.assertEquals("test-value", value);
+        client.invokeRPC(rpcStub::testRpcSetKeyword, null);
+        value = client.invokeRPC(rpcStub::testRpcGetKeyword);
+        Assertions.assertNull(value);
+
         final Long rpcOutput = client.invokeRPC(rpcStub::testRpcFunc1, RPC_INPUT);
 
         Assertions.assertEquals(RPC_OUTPUT, rpcOutput);

diff --git a/src/test/java/io/iworkflow/integ/rpc/RpcWorkflow.java b/src/test/java/io/iworkflow/integ/rpc/RpcWorkflow.java
@@ -128,4 +128,20 @@ public String testRpcGetDataAttribute(Context context, Persistence persistence,
         }
         return persistence.getDataAttribute(TEST_DATA_OBJECT_KEY, String.class);
     }
+
+    @RPC
+    public void testRpcSetKeyword(Context context, String input, Persistence persistence, Communication communication) {
+        if (context.getWorkflowId().isEmpty() || context.getWorkflowRunId().isEmpty()) {
+            throw new RuntimeException("invalid context");
+        }
+        persistence.setSearchAttributeKeyword(TEST_SEARCH_ATTRIBUTE_KEYWORD, input);
+    }
+
+    @RPC
+    public String testRpcGetKeyword(Context context, Persistence persistence, Communication communication) {
+        if (context.getWorkflowId().isEmpty() || context.getWorkflowRunId().isEmpty()) {
+            throw new RuntimeException("invalid context");
+        }
+        return persistence.getSearchAttributeKeyword(TEST_SEARCH_ATTRIBUTE_KEYWORD);
+    }
 }