dex-python-sdk 0.0.1__py3-none-any.whl

dex/tests/test_persistence_data_attributes.py

@@ -0,0 +1,67 @@
+# Copyright (c) 2022-2026 Super Durable, Inc.
+#
+# 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
+#
+#     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.
+
+import inspect
+import time
+import unittest
+
+from dex.client import Client
+from dex.tests.worker_server import registry
+from dex.tests.workflows.persistence_data_attributes_workflow import (
+    PersistenceDataAttributesWorkflow,
+    initial_da_1,
+    initial_da_2,
+    initial_da_value_2,
+    initial_da_value_1,
+    test_da_set_key,
+    test_da_set_value,
+    expected_final_das,
+)
+from dex.workflow_options import WorkflowOptions
+
+
+class TestPersistenceDataAttributes(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.client = Client(registry)
+
+    def test_persistence_data_attributes_workflow(self):
+        wf_id = f"{inspect.currentframe().f_code.co_name}-{time.time_ns()}"
+
+        start_options = WorkflowOptions(
+            initial_data_attributes={
+                initial_da_1: initial_da_value_1,
+                initial_da_2: initial_da_value_2,
+            },
+        )
+
+        wf_run_id = self.client.start_workflow(
+            PersistenceDataAttributesWorkflow, wf_id, 100, None, start_options
+        )
+        self.client.set_workflow_data_attributes(
+            PersistenceDataAttributesWorkflow,
+            wf_id,
+            wf_run_id,
+            {
+                test_da_set_key: test_da_set_value,
+            },
+        )
+        self.client.wait_for_workflow_completion(wf_id, None)
+
+        all_data_attributes = self.client.get_all_workflow_data_attributes(
+            PersistenceDataAttributesWorkflow, wf_id, wf_run_id
+        )
+        for k, v in expected_final_das.items():
+            assert k in all_data_attributes
+            assert all_data_attributes[k] == v

dex/tests/test_persistence_search_attributes.py

@@ -0,0 +1,132 @@
+# Copyright (c) 2022-2026 Super Durable, Inc.
+#
+# 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
+#
+#     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.
+
+import inspect
+import time
+import unittest
+
+from dex.client import Client
+from dex.tests.worker_server import registry
+from dex.tests.workflows.persistence_search_attributes_workflow import (
+    PersistenceSearchAttributesWorkflow,
+    sa_keyword_key,
+    sa_double_key,
+    sa_int_key,
+    sa_bool_key,
+    sa_datetime_key,
+    sa_keyword_array_key,
+    initial_sa_keyword,
+    initial_sa_double,
+    initial_sa_int,
+    initial_sa_bool,
+    initial_sa_datetime,
+    initial_sa_keyword_array,
+    SearchAttributeState1,
+    sa_keyword,
+    sa_double,
+    sa_bool,
+    sa_keyword_array,
+    sa_int,
+    final_sa_keyword,
+    final_sa_bool,
+    final_sa_int,
+    final_sa_keyword_array,
+)
+from dex.workflow_options import WorkflowOptions
+
+
+class TestPersistenceSearchAttributes(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.client = Client(registry)
+
+    def test_persistence_search_attributes_workflow(self):
+        wf_id = f"{inspect.currentframe().f_code.co_name}-{time.time_ns()}"
+
+        wf_opts = WorkflowOptions(
+            initial_search_attributes={
+                sa_keyword_key: initial_sa_keyword,
+                sa_double_key: initial_sa_double,
+                sa_int_key: initial_sa_int,
+                sa_bool_key: initial_sa_bool,
+                sa_datetime_key: initial_sa_datetime,
+                sa_keyword_array_key: initial_sa_keyword_array,
+            }
+        )
+        wf_opts.add_wait_for_completion_state_ids(SearchAttributeState1)
+        self.client.start_workflow(
+            PersistenceSearchAttributesWorkflow, wf_id, 100, None, wf_opts
+        )
+
+        initial_returned_search_attributes = self.client.get_all_search_attributes(
+            PersistenceSearchAttributesWorkflow,
+            wf_id,
+        )
+
+        initial_expected_search_attributes = dict()
+        initial_expected_search_attributes[sa_keyword_key] = initial_sa_keyword
+        initial_expected_search_attributes[sa_double_key] = initial_sa_double
+        initial_expected_search_attributes[sa_bool_key] = initial_sa_bool
+        initial_expected_search_attributes[sa_keyword_array_key] = (
+            initial_sa_keyword_array
+        )
+        initial_expected_search_attributes[sa_int_key] = initial_sa_int
+        initial_expected_search_attributes[sa_datetime_key] = (
+            "2024-11-10T00:00:01.731455544Z"  # This is a bug. The dex-server always returns utc time. See https://github.com/superdurable/dex/issues/261
+            # "2024-11-09T18:00:01.731455544-06:00"
+        )
+
+        assert initial_expected_search_attributes == initial_returned_search_attributes
+
+        self.client.wait_for_state_execution_completion_with_state_execution_id(
+            SearchAttributeState1, wf_id
+        )
+
+        returned_search_attributes = self.client.get_all_search_attributes(
+            PersistenceSearchAttributesWorkflow,
+            wf_id,
+        )
+
+        expected_search_attributes = dict()
+        expected_search_attributes[sa_keyword_key] = sa_keyword
+        expected_search_attributes[sa_double_key] = sa_double
+        expected_search_attributes[sa_bool_key] = sa_bool
+        expected_search_attributes[sa_keyword_array_key] = sa_keyword_array
+        expected_search_attributes[sa_int_key] = sa_int
+        expected_search_attributes[sa_datetime_key] = (
+            "2024-11-13T00:00:01.731455544Z"  # This is a bug. The dex-server always returns utc time. See https://github.com/superdurable/dex/issues/261
+            # "2024-11-12T18:00:01.731455544-06:00"
+        )
+
+        assert expected_search_attributes == returned_search_attributes
+
+        self.client.wait_for_workflow_completion(wf_id)
+
+        final_expected_search_attributes = dict()
+        final_expected_search_attributes[sa_keyword_key] = final_sa_keyword
+        final_expected_search_attributes[sa_double_key] = sa_double
+        final_expected_search_attributes[sa_int_key] = final_sa_int
+        final_expected_search_attributes[sa_bool_key] = final_sa_bool
+        final_expected_search_attributes[sa_keyword_array_key] = final_sa_keyword_array
+        final_expected_search_attributes[sa_datetime_key] = (
+            "2024-12-14T00:00:01.731455544Z"  # This is a bug. The dex-server always returns utc time. See https://github.com/superdurable/dex/issues/261
+            # "2024-12-13T18:00:01.731455544-06:00"
+        )
+
+        final_returned_search_attributes = self.client.get_all_search_attributes(
+            PersistenceSearchAttributesWorkflow,
+            wf_id,
+        )
+
+        assert final_expected_search_attributes == final_returned_search_attributes

dex/tests/test_persistence_state_execution_locals.py

@@ -0,0 +1,43 @@
+# Copyright (c) 2022-2026 Super Durable, Inc.
+#
+# 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
+#
+#     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.
+
+import inspect
+import time
+import unittest
+
+from dex.client import Client
+from dex.tests.worker_server import registry
+from dex.tests.workflows.persistence_state_execution_local_workflow import (
+    PersistenceStateExecutionLocalWorkflow,
+    PERSISTENCE_LOCAL_VALUE,
+)
+from dex.workflow_options import WorkflowOptions
+
+
+class TestPersistenceExecutionLocalRead(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.client = Client(registry)
+
+    def test_persistence_execution_local_workflow(self):
+        wf_id = f"{inspect.currentframe().f_code.co_name}-{time.time_ns()}"
+        start_options = WorkflowOptions()
+        self.client.start_workflow(
+            PersistenceStateExecutionLocalWorkflow, wf_id, 200, None, start_options
+        )
+        self.client.wait_for_workflow_completion(wf_id, None)
+        res = self.client.invoke_rpc(
+            wf_id, PersistenceStateExecutionLocalWorkflow.test_persistence_read
+        )
+        assert res == PERSISTENCE_LOCAL_VALUE

dex/tests/test_rpc.py

@@ -0,0 +1,69 @@
+# Copyright (c) 2022-2026 Super Durable, Inc.
+#
+# 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
+#
+#     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.
+
+import inspect
+import time
+import unittest
+
+from dex.client import Client
+from dex.tests.worker_server import registry
+from dex.tests.workflows.rpc_workflow import RPCWorkflow, Mydata
+
+
+class TestRPCs(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.client = Client(registry)
+
+    def test_simple_rpc(self):
+        wf_id = f"{inspect.currentframe().f_code.co_name}-{time.time_ns()}"
+        self.client.start_workflow(RPCWorkflow, wf_id, 10)
+
+        input = Mydata("test", 100)
+        output = self.client.invoke_rpc(
+            wf_id, RPCWorkflow.test_rpc_input_type, input, Mydata
+        )
+        assert output == input
+
+        output = self.client.invoke_rpc(wf_id, RPCWorkflow.test_simple_rpc)
+        assert output == 123
+        wf = RPCWorkflow()
+        output = self.client.invoke_rpc(wf_id, wf.test_simple_rpc)
+        assert output == 123
+
+    def test_complicated_rpc(self):
+        wf_id = f"{inspect.currentframe().f_code.co_name}-{time.time_ns()}"
+        self.client.start_workflow(RPCWorkflow, wf_id, 10)
+        self.client.invoke_rpc(wf_id, RPCWorkflow.test_rpc_persistence_write, 100)
+        res = self.client.invoke_rpc(wf_id, RPCWorkflow.test_rpc_persistence_read)
+        assert res == 100
+        self.client.invoke_rpc(wf_id, RPCWorkflow.test_rpc_trigger_state, 200)
+        res = self.client.invoke_rpc(wf_id, RPCWorkflow.test_rpc_persistence_read)
+        assert res == 200
+
+        res1 = self.client.invoke_rpc(
+            wf_id, RPCWorkflow.test_rpc_publish_to_idle_channel, "input-1"
+        )
+        # Channel Size should be 1
+        assert res1 == 1
+
+        res2 = self.client.invoke_rpc(
+            wf_id, RPCWorkflow.test_rpc_publish_to_idle_channel, "input-2"
+        )
+        # Channel Size should be 2
+        assert res2 == 2
+
+        self.client.invoke_rpc(wf_id, RPCWorkflow.test_rpc_publish_channel)
+        result = self.client.wait_for_workflow_completion(wf_id, str)
+        assert result == "done"

dex/tests/test_rpc_with_memo.py

@@ -0,0 +1,200 @@
+# Copyright (c) 2022-2026 Super Durable, Inc.
+#
+# 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
+#
+#     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.
+
+import inspect
+import time
+import unittest
+
+from dex.client import Client
+from dex.dex_api.models.workflow_stop_type import WorkflowStopType
+from dex.stop_workflow_options import StopWorkflowOptions
+from dex.tests import registry
+from dex.tests.workflows.rpc_memo_workflow import (
+    RpcMemoWorkflow,
+    TEST_DATA_OBJECT_KEY,
+    TEST_SEARCH_ATTRIBUTE_INT,
+    TEST_SEARCH_ATTRIBUTE_KEY,
+    RPC_INPUT,
+    RPC_OUTPUT,
+    TEST_STR,
+    TEST_DELAY,
+)
+
+
+class TestRpcWithMemo(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.client = Client(registry)
+
+    def test_rpc_memo_workflow_func1(self):
+        wf_id = f"{inspect.currentframe().f_code.co_name}-{time.time_ns()}"
+        run_id = self.client.start_workflow(RpcMemoWorkflow, wf_id, 30, 999)
+
+        self.client.invoke_rpc(
+            wf_id, RpcMemoWorkflow.test_rpc_set_data_attribute, TEST_STR
+        )
+        value = self.client.invoke_rpc(
+            wf_id, RpcMemoWorkflow.test_rpc_get_data_attribute_strong_consistency
+        )
+        assert value == TEST_STR
+        time.sleep(TEST_DELAY)
+        value = self.client.invoke_rpc(
+            wf_id, RpcMemoWorkflow.test_rpc_get_data_attribute, None, str
+        )
+
+        self.client.invoke_rpc(wf_id, RpcMemoWorkflow.test_rpc_set_data_attribute, None)
+        value = self.client.invoke_rpc(
+            wf_id, RpcMemoWorkflow.test_rpc_get_data_attribute_strong_consistency
+        )
+        assert value is None
+        time.sleep(TEST_DELAY)
+        value = self.client.invoke_rpc(
+            wf_id, RpcMemoWorkflow.test_rpc_get_data_attribute
+        )
+        assert value is None
+
+        self.client.invoke_rpc(wf_id, RpcMemoWorkflow.test_rpc_set_keyword, TEST_STR)
+        value = self.client.invoke_rpc(
+            wf_id, RpcMemoWorkflow.test_rpc_get_keyword_strong_consistency
+        )
+        assert value == TEST_STR
+        time.sleep(TEST_DELAY)
+        value = self.client.invoke_rpc(wf_id, RpcMemoWorkflow.test_rpc_get_keyword)
+        assert value == TEST_STR
+
+        self.client.invoke_rpc(wf_id, RpcMemoWorkflow.test_rpc_set_keyword, None)
+        value = self.client.invoke_rpc(
+            wf_id, RpcMemoWorkflow.test_rpc_get_keyword_strong_consistency
+        )
+        assert value is None
+        time.sleep(TEST_DELAY)
+        value = self.client.invoke_rpc(wf_id, RpcMemoWorkflow.test_rpc_get_keyword)
+        assert value is None
+
+        rpc_output = self.client.invoke_rpc(
+            wf_id, RpcMemoWorkflow.test_rpc_func1, RPC_INPUT
+        )
+        assert RPC_OUTPUT == rpc_output
+
+        self.client.wait_for_workflow_completion(wf_id)
+
+        data_attributes = self.client.get_workflow_data_attributes(
+            RpcMemoWorkflow, wf_id, run_id, [TEST_DATA_OBJECT_KEY]
+        )
+        assert TEST_DATA_OBJECT_KEY in data_attributes
+        assert data_attributes[TEST_DATA_OBJECT_KEY] == RPC_INPUT
+
+        search_attributes = self.client.get_workflow_search_attributes(
+            RpcMemoWorkflow,
+            wf_id,
+            [TEST_SEARCH_ATTRIBUTE_KEY, TEST_SEARCH_ATTRIBUTE_INT],
+            run_id,
+        )
+        assert TEST_SEARCH_ATTRIBUTE_INT in search_attributes
+        assert TEST_SEARCH_ATTRIBUTE_KEY in search_attributes
+        assert search_attributes[TEST_SEARCH_ATTRIBUTE_INT] == RPC_OUTPUT
+        assert search_attributes[TEST_SEARCH_ATTRIBUTE_KEY] == RPC_INPUT
+
+    def test_rpc_memo_workflow_func0(self):
+        wf_id = f"{inspect.currentframe().f_code.co_name}-{time.time_ns()}"
+        run_id = self.client.start_workflow(RpcMemoWorkflow, wf_id, 30, 999)
+
+        rpc_output = self.client.invoke_rpc(
+            wf_id, RpcMemoWorkflow.test_rpc_func0, TEST_STR
+        )
+        assert RPC_OUTPUT == rpc_output
+
+        self.client.wait_for_workflow_completion(wf_id)
+
+        data_attributes = self.client.get_workflow_data_attributes(
+            RpcMemoWorkflow, wf_id, run_id, [TEST_DATA_OBJECT_KEY]
+        )
+        assert TEST_DATA_OBJECT_KEY in data_attributes
+        assert data_attributes[TEST_DATA_OBJECT_KEY] == TEST_STR
+
+        search_attributes = self.client.get_workflow_search_attributes(
+            RpcMemoWorkflow,
+            wf_id,
+            [TEST_SEARCH_ATTRIBUTE_KEY, TEST_SEARCH_ATTRIBUTE_INT],
+            run_id,
+        )
+        assert TEST_SEARCH_ATTRIBUTE_INT in search_attributes
+        assert TEST_SEARCH_ATTRIBUTE_KEY in search_attributes
+        assert search_attributes[TEST_SEARCH_ATTRIBUTE_INT] == RPC_OUTPUT
+        assert search_attributes[TEST_SEARCH_ATTRIBUTE_KEY] == TEST_STR
+
+    def test_rpc_memo_workflow_proc1(self):
+        wf_id = f"{inspect.currentframe().f_code.co_name}-{time.time_ns()}"
+        run_id = self.client.start_workflow(RpcMemoWorkflow, wf_id, 30, 999)
+
+        self.client.invoke_rpc(wf_id, RpcMemoWorkflow.test_rpc_proc1, RPC_INPUT)
+        self.client.wait_for_workflow_completion(wf_id)
+
+        data_attributes = self.client.get_workflow_data_attributes(
+            RpcMemoWorkflow, wf_id, run_id, [TEST_DATA_OBJECT_KEY]
+        )
+        assert TEST_DATA_OBJECT_KEY in data_attributes
+        assert data_attributes[TEST_DATA_OBJECT_KEY] == RPC_INPUT
+
+        search_attributes = self.client.get_workflow_search_attributes(
+            RpcMemoWorkflow,
+            wf_id,
+            [TEST_SEARCH_ATTRIBUTE_KEY, TEST_SEARCH_ATTRIBUTE_INT],
+            run_id,
+        )
+        assert TEST_SEARCH_ATTRIBUTE_INT in search_attributes
+        assert TEST_SEARCH_ATTRIBUTE_KEY in search_attributes
+        assert search_attributes[TEST_SEARCH_ATTRIBUTE_INT] == RPC_OUTPUT
+        assert search_attributes[TEST_SEARCH_ATTRIBUTE_KEY] == RPC_INPUT
+
+    def test_rpc_memo_workflow_proc0(self):
+        wf_id = f"{inspect.currentframe().f_code.co_name}-{time.time_ns()}"
+        run_id = self.client.start_workflow(RpcMemoWorkflow, wf_id, 30, 999)
+
+        self.client.invoke_rpc(wf_id, RpcMemoWorkflow.test_rpc_proc0)
+        self.client.wait_for_workflow_completion(wf_id)
+
+        data_attributes = self.client.get_workflow_data_attributes(
+            RpcMemoWorkflow, wf_id, run_id, [TEST_DATA_OBJECT_KEY]
+        )
+        assert TEST_DATA_OBJECT_KEY in data_attributes
+        assert data_attributes[TEST_DATA_OBJECT_KEY] == TEST_STR
+
+        search_attributes = self.client.get_workflow_search_attributes(
+            RpcMemoWorkflow,
+            wf_id,
+            [TEST_SEARCH_ATTRIBUTE_KEY, TEST_SEARCH_ATTRIBUTE_INT],
+            run_id,
+        )
+        assert TEST_SEARCH_ATTRIBUTE_INT in search_attributes
+        assert TEST_SEARCH_ATTRIBUTE_KEY in search_attributes
+        assert search_attributes[TEST_SEARCH_ATTRIBUTE_INT] == RPC_OUTPUT
+        assert search_attributes[TEST_SEARCH_ATTRIBUTE_KEY] == TEST_STR
+
+    def test_rpc_memo_workflow_func1_readonly(self):
+        wf_id = f"{inspect.currentframe().f_code.co_name}-{time.time_ns()}"
+        self.client.start_workflow(RpcMemoWorkflow, wf_id, 30, 999)
+
+        rpc_output = self.client.invoke_rpc(
+            wf_id, RpcMemoWorkflow.test_rpc_func1_readonly, RPC_INPUT
+        )
+        assert RPC_OUTPUT == rpc_output
+
+        self.client.stop_workflow(
+            wf_id,
+            StopWorkflowOptions(
+                workflow_stop_type=WorkflowStopType.FAIL,
+                reason=TEST_STR,
+            ),
+        )

dex/tests/test_rpc_with_memo_duplicate_java_tests.py

@@ -0,0 +1,122 @@
+# Copyright (c) 2022-2026 Super Durable, Inc.
+#
+# 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
+#
+#     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.
+
+import inspect
+import time
+import unittest
+
+from dex.client import Client
+from dex.tests.worker_server import registry
+from dex.tests.workflows.java_duplicate_rpc_memo_workflow import (
+    JavaDuplicateRpcMemoWorkflow,
+    RpcMemoWorkflowState2,
+    TEST_DATA_OBJECT_KEY,
+    TEST_SEARCH_ATTRIBUTE_INT,
+    TEST_SEARCH_ATTRIBUTE_KEY,
+    RPC_INPUT,
+    RPC_OUTPUT,
+    TEST_STR,
+    TEST_DELAY,
+)
+
+
+class TestRpcWithMemo(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.client = Client(registry)
+
+    def test_rpc_memo_workflow_func1(self):
+        wf_id = f"{inspect.currentframe().f_code.co_name}-{time.time_ns()}"
+        run_id = self.client.start_workflow(
+            JavaDuplicateRpcMemoWorkflow, wf_id, 30, 999
+        )
+
+        self.client.invoke_rpc(
+            wf_id, JavaDuplicateRpcMemoWorkflow.test_rpc_set_data_attribute, TEST_STR
+        )
+        value = self.client.invoke_rpc(
+            wf_id,
+            JavaDuplicateRpcMemoWorkflow.test_rpc_get_data_attribute_strong_consistency,
+        )
+        assert value == TEST_STR
+        time.sleep(TEST_DELAY)
+        value = self.client.invoke_rpc(
+            wf_id, JavaDuplicateRpcMemoWorkflow.test_rpc_get_data_attribute, None, str
+        )
+
+        self.client.invoke_rpc(
+            wf_id, JavaDuplicateRpcMemoWorkflow.test_rpc_set_data_attribute, None
+        )
+        value = self.client.invoke_rpc(
+            wf_id,
+            JavaDuplicateRpcMemoWorkflow.test_rpc_get_data_attribute_strong_consistency,
+        )
+        assert value is None
+        time.sleep(TEST_DELAY)
+        value = self.client.invoke_rpc(
+            wf_id, JavaDuplicateRpcMemoWorkflow.test_rpc_get_data_attribute
+        )
+        assert value is None
+
+        self.client.invoke_rpc(
+            wf_id, JavaDuplicateRpcMemoWorkflow.test_rpc_set_keyword, TEST_STR
+        )
+        value = self.client.invoke_rpc(
+            wf_id, JavaDuplicateRpcMemoWorkflow.test_rpc_get_keyword_strong_consistency
+        )
+        assert value == TEST_STR
+        time.sleep(TEST_DELAY)
+        value = self.client.invoke_rpc(
+            wf_id, JavaDuplicateRpcMemoWorkflow.test_rpc_get_keyword
+        )
+        assert value == TEST_STR
+
+        self.client.invoke_rpc(
+            wf_id, JavaDuplicateRpcMemoWorkflow.test_rpc_set_keyword, None
+        )
+        value = self.client.invoke_rpc(
+            wf_id, JavaDuplicateRpcMemoWorkflow.test_rpc_get_keyword_strong_consistency
+        )
+        assert value is None
+        time.sleep(TEST_DELAY)
+        value = self.client.invoke_rpc(
+            wf_id, JavaDuplicateRpcMemoWorkflow.test_rpc_get_keyword
+        )
+        assert value is None
+
+        rpc_output = self.client.invoke_rpc(
+            wf_id, JavaDuplicateRpcMemoWorkflow.test_rpc_func1, RPC_INPUT
+        )
+        assert RPC_OUTPUT == rpc_output
+
+        output = self.client.wait_for_workflow_completion(wf_id, int)
+        RpcMemoWorkflowState2.reset_counter()
+        assert 2 == output
+
+        data_attributes = self.client.get_workflow_data_attributes(
+            JavaDuplicateRpcMemoWorkflow, wf_id, run_id, [TEST_DATA_OBJECT_KEY]
+        )
+        assert TEST_DATA_OBJECT_KEY in data_attributes
+        assert data_attributes[TEST_DATA_OBJECT_KEY] == RPC_INPUT
+
+        search_attributes = self.client.get_workflow_search_attributes(
+            JavaDuplicateRpcMemoWorkflow,
+            wf_id,
+            [TEST_SEARCH_ATTRIBUTE_KEY, TEST_SEARCH_ATTRIBUTE_INT],
+            run_id,
+        )
+        assert TEST_SEARCH_ATTRIBUTE_INT in search_attributes
+        assert TEST_SEARCH_ATTRIBUTE_KEY in search_attributes
+        assert search_attributes[TEST_SEARCH_ATTRIBUTE_INT] == RPC_OUTPUT
+        assert search_attributes[TEST_SEARCH_ATTRIBUTE_KEY] == RPC_INPUT

dex/tests/test_signal.py

@@ -0,0 +1,56 @@
+# Copyright (c) 2022-2026 Super Durable, Inc.
+#
+# 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
+#
+#     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.
+
+import inspect
+import time
+import unittest
+
+from dex.client import Client
+from dex.tests.worker_server import registry
+from dex.tests.workflows.wait_signal_workflow import (
+    WaitSignalWorkflow,
+    test_channel_int,
+    test_channel_str,
+    test_channel_none,
+    test_channel1,
+    test_idle_channel_none,
+)
+
+
+class TestSignal(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.client = Client(registry)
+
+    def test_signal(self):
+        wf_id = f"{inspect.currentframe().f_code.co_name}-{time.time_ns()}"
+        self.client.start_workflow(WaitSignalWorkflow, wf_id, 10)
+        self.client.signal_workflow(wf_id, test_channel_int, 123)
+        self.client.signal_workflow(wf_id, test_channel_str, "abc")
+        self.client.signal_workflow(wf_id, test_channel_none)
+
+        self.client.signal_workflow(wf_id, test_channel1)
+        res = self.client.wait_for_workflow_completion(wf_id, str)
+        assert res == "abc"
+
+    def test_signal_channel_size(self):
+        wf_id = f"{inspect.currentframe().f_code.co_name}-{time.time_ns()}"
+        self.client.start_workflow(WaitSignalWorkflow, wf_id, 1)
+        self.client.signal_workflow(wf_id, test_idle_channel_none)
+        self.client.signal_workflow(wf_id, test_idle_channel_none)
+        self.client.signal_workflow(wf_id, test_idle_channel_none)
+        res = self.client.invoke_rpc(
+            wf_id, WaitSignalWorkflow.get_idle_signal_channel_size
+        )
+        assert res == 3