Skip to content

Commit 47ae609

Browse files
committed
Add test class for testing directly with a database connection
1 parent 4da415c commit 47ae609

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

tests/test_utils.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,3 +63,13 @@ def setUpClass(cls) -> None:
6363
from code_submitter.server import database
6464

6565
cls.database = database
66+
67+
68+
class InTransactionTestCase(DatabaseTestCase):
69+
def setUp(self) -> None:
70+
super().setUp()
71+
self.await_(self.database.connect())
72+
73+
def tearDown(self) -> None:
74+
self.await_(self.database.disconnect())
75+
super().tearDown()

0 commit comments

Comments
 (0)