diff options
| -rw-r--r-- | backend/routes/forms/unittesting.py | 14 | ||||
| -rw-r--r-- | resources/unittest_template.py | 2 |
2 files changed, 9 insertions, 7 deletions
diff --git a/backend/routes/forms/unittesting.py b/backend/routes/forms/unittesting.py index c093718..13fa639 100644 --- a/backend/routes/forms/unittesting.py +++ b/backend/routes/forms/unittesting.py @@ -36,12 +36,14 @@ def _make_unit_code(units: dict[str, str]) -> str: result = "" for unit_name, unit_code in units.items(): - test_prefix = "test_" if unit_name != "setUp" else "" - - result += ( - f"\ndef {test_prefix}{unit_name.removeprefix('#')}(self):" # Function definition - f"\n{indent(unit_code, ' ')}" # Unit code - ) + # Function definition + if unit_name == "setUp": + result += "\ndef setUp(self):" + else: + result += f"\nasync def {unit_name.removeprefix('#')}(self):" + + # Unite code + result += f"\n{indent(unit_code, ' ')}" return indent(result, " ") diff --git a/resources/unittest_template.py b/resources/unittest_template.py index 6f704f3..104b3b4 100644 --- a/resources/unittest_template.py +++ b/resources/unittest_template.py @@ -15,7 +15,7 @@ from unittest import mock ### USER CODE -class RunnerTestCase(unittest.TestCase): +class RunnerTestCase(unittest.IsolatedAsyncioTestCase): ### UNIT CODE |