non-dts test fix

Author: andystaples

tests/durabletask/test_batch_actions.py

@@ -42,7 +42,7 @@ def test_get_all_entities():
         c = client.TaskHubGrpcClient()
         with pytest.raises(_InactiveRpcError) as exec_info:
             c.get_all_entities()
-        assert "unimplemented" in str(exec_info.value)
+        assert "method QueryEntities not implemented" in str(exec_info.value)
 
 
 def test_clean_entity_storage():
@@ -54,7 +54,7 @@ def test_clean_entity_storage():
         c = client.TaskHubGrpcClient()
         with pytest.raises(_InactiveRpcError) as exec_info:
             c.clean_entity_storage()
-        assert "unimplemented" in str(exec_info.value)
+        assert "method CleanEntityStorage not implemented" in str(exec_info.value)
 
 
 def test_purge_orchestrations_by_status():
@@ -63,64 +63,10 @@ def test_purge_orchestrations_by_status():
         w.start()
 
         c = client.TaskHubGrpcClient()
-
-        # Schedule and let it fail
-        failed_id = c.schedule_new_orchestration(failing_orchestrator)
-        try:
-            c.wait_for_orchestration_completion(failed_id, timeout=30)
-        except client.OrchestrationFailedError:
-            pass  # Expected failure
-
-        # Verify it exists and is failed
-        state_before = c.get_orchestration_state(failed_id)
-        assert state_before is not None
-        assert state_before.runtime_status == client.OrchestrationStatus.FAILED
-
-        # Purge failed orchestrations
-        result = c.purge_orchestrations_by(
-            runtime_status=[client.OrchestrationStatus.FAILED],
-            recursive=True
-        )
-
-        # Verify purge result
-        assert result.deleted_instance_count >= 1
-
-        # Verify the failed orchestration no longer exists
-        state_after = c.get_orchestration_state(failed_id)
-        assert state_after is None
-
-
-def test_purge_orchestrations_by_time_range():
-    with worker.TaskHubGrpcWorker() as w:
-        w.add_orchestrator(empty_orchestrator)
-        w.start()
-
-        c = client.TaskHubGrpcClient()
-
-        # Get current time
-        before_creation = datetime.now(timezone.utc) - timedelta(seconds=5)
-
-        # Schedule orchestration
-        id = c.schedule_new_orchestration(empty_orchestrator, input="ToPurgeByTime")
-        c.wait_for_orchestration_completion(id, timeout=30)
-
-        after_creation = datetime.now(timezone.utc) + timedelta(seconds=5)
-
-        # Verify it exists
-        state_before = c.get_orchestration_state(id)
-        assert state_before is not None
-
-        # Purge by time range
-        result = c.purge_orchestrations_by(
-            created_time_from=before_creation,
-            created_time_to=after_creation,
-            runtime_status=[client.OrchestrationStatus.COMPLETED],
-            recursive=True
-        )
-
-        # Verify purge result
-        assert result.deleted_instance_count >= 1
-
-        # Verify it no longer exists
-        state_after = c.get_orchestration_state(id)
-        assert state_after is None
+        with pytest.raises(_InactiveRpcError) as exec_info:
+            c.purge_orchestrations_by(
+                runtime_status=[client.OrchestrationStatus.FAILED],
+                recursive=True
+            )
+        # sic - error returned from sidecar
+        assert "multi-instance purge is not unimplemented" in str(exec_info.value)