From 2759f4cdfd1450cb891dfa806dc93d1101533d96 Mon Sep 17 00:00:00 2001
From: Max Nanis
Date: Thu, 10 Sep 2026 19:47:53 -0700
Subject: Simple Magic Link tests (just to use redis_config param)

---
 tests/http/test_magic.py | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 tests/http/test_magic.py

(limited to 'tests/http/test_magic.py')

diff --git a/tests/http/test_magic.py b/tests/http/test_magic.py
new file mode 100644
index 0000000..f645677
--- /dev/null
+++ b/tests/http/test_magic.py
@@ -0,0 +1,48 @@
+from uuid import uuid4
+
+from generalresearch.redis_helper import RedisConfig
+
+from jb.api.magic_token import (
+    consume_amt_account_link_token,
+    consume_magic_token,
+    create_amt_account_link_token,
+    create_magic_token,
+)
+from jb.models.auth import AmtAccountLink
+
+
+class TestViewFunctions:
+
+    def test_create_amt_account_link_token(
+        self, redis_config: RedisConfig, amt_worker_id: str
+    ):
+        email = f"{uuid4().hex[:8]}@jamesbillings67.com"
+        res = create_amt_account_link_token(
+            email=email, amt_worker_id=amt_worker_id, redis_config=redis_config
+        )
+        assert isinstance(res, str)
+
+    def test_create_and_retrieve_token(
+        self, redis_config: RedisConfig, amt_worker_id: str
+    ):
+        email = f"{uuid4().hex[:8]}@jamesbillings67.com"
+        token = create_amt_account_link_token(
+            email=email, amt_worker_id=amt_worker_id, redis_config=redis_config
+        )
+
+        res = consume_amt_account_link_token(token=token, redis_config=redis_config)
+        assert isinstance(res, AmtAccountLink)
+        assert res.email == email
+        assert res.amt_worker_id == amt_worker_id
+
+    def test_create_magic_link(self, redis_config: RedisConfig, amt_worker_id: str):
+        email = f"{uuid4().hex[:8]}@jamesbillings67.com"
+        res = create_magic_token(user_email=email, redis_config=redis_config)
+        assert isinstance(res, str)
+
+    def test_consume_magic_link(self, redis_config: RedisConfig, amt_worker_id: str):
+        email = f"{uuid4().hex[:8]}@jamesbillings67.com"
+        token = create_magic_token(user_email=email, redis_config=redis_config)
+
+        res = consume_magic_token(token=token, redis_config=redis_config)
+        assert res == email
-- 
cgit v1.2.3