From 14c7355c054a97598a8243f38f9eac0a9a77fda4 Mon Sep 17 00:00:00 2001
From: chuck-dbos <134347445+chuck-dbos@users.noreply.github.com>
Date: Wed, 22 Jan 2025 16:10:47 -0500
Subject: [PATCH] Add missed dbos_class decorators (#183)

---
 tests/queuedworkflow.py       | 1 +
 tests/test_classdecorators.py | 1 +
 2 files changed, 2 insertions(+)

diff --git a/tests/queuedworkflow.py b/tests/queuedworkflow.py
index 94a1635f..53f85ea7 100644
--- a/tests/queuedworkflow.py
+++ b/tests/queuedworkflow.py
@@ -26,6 +26,7 @@ def default_config() -> ConfigFile:
 q = Queue("testq", concurrency=1, limiter={"limit": 1, "period": 1})
 
 
+@DBOS.dbos_class()
 class WF:
     @staticmethod
     @DBOS.workflow()
diff --git a/tests/test_classdecorators.py b/tests/test_classdecorators.py
index 0959fa58..21acd571 100644
--- a/tests/test_classdecorators.py
+++ b/tests/test_classdecorators.py
@@ -187,6 +187,7 @@ def test_func_admin_r_s(var: str) -> str:
 
 # We can put classes in functions to test decorators for now...
 def test_simple_workflow_static(dbos: DBOS) -> None:
+    @DBOS.dbos_class()
     class DBOSTestClassStatic:
         txn_counter: int = 0
         wf_counter: int = 0