Models, Project files, some pytests, requirements.. etcHEADmaster

1 files changed, 23 insertions, 0 deletions

diff --git a/tests/managers/amt.py b/tests/managers/amt.py
new file mode 100644
index 0000000..0b2e501
--- /dev/null
+++ b/tests/managers/amt.py
@@ -0,0 +1,23 @@
+from jb.decorators import HTM, HM, HQM
+from jb.managers.amt import AMTManager
+
+
+def test_create_hit_type(hit_type):
+    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(hit_type_with_amt_id, question):
+    question = HQM.get_or_create(question)
+
+    hit_type = hit_type_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)
+    assert hit.amt_hit_id is not None
+    assert hit.id is None
+    HM.create(hit)
+    assert hit.id is not None