Passing Managers into flow tasks for better pytest usage. Conftests broken out into seperate fixture files. Extensive type hinting.

1 files changed, 36 insertions, 0 deletions

diff --git a/tests/managers/test_amt.py b/tests/managers/test_amt.py
new file mode 100644
index 0000000..63d3737
--- /dev/null
+++ b/tests/managers/test_amt.py
@@ -0,0 +1,36 @@
+from jb.managers.amt import AMTManager
+from jb.models.hit import HitType, HitQuestion
+
+from jb.managers.hit import HitQuestionManager, HitTypeManager, HitManager
+
+
+class TestAMTManager:
+
+    def test_create_hit_type(self, hit_type: HitType):
+        assert hit_type.amt_hit_type_id is None
+        AMTManager.create_hit_type(hit_type=hit_type)
+        assert hit_type.amt_hit_type_id is not None
+
+    def test_create_hit_with_hit_type(
+        self,
+        hqm: HitQuestionManager,
+        htm: HitTypeManager,
+        hm: HitManager,
+        hit_type_record_with_amt_id: HitType,
+        question_record: HitQuestion,
+    ):
+
+        hit_type = hit_type_record_with_amt_id
+        hit_type = [
+            x
+            for x in htm.filter_active()
+            if x.amt_hit_type_id == hit_type.amt_hit_type_id
+        ][0]
+
+        hit = AMTManager.create_hit_with_hit_type(
+            hit_type=hit_type, question=question_record
+        )
+        assert hit.amt_hit_id is not None
+        assert hit.id is None
+        hm.create(hit)
+        assert hit.id is not None