add tests

collective · May 11, 2024 · 63e7142 · 63e7142
1 parent 2b41569
commit 63e7142
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/pas/plugins/tfa/tests/test_helpers.py b/src/pas/plugins/tfa/tests/test_helpers.py
@@ -103,3 +103,16 @@ def test_qr_code_generation(self):
             secret,
         )
         self.assertEqual(str(base64_image), QR_CODE_REF)
+
+    def test_get_or_create_secret(self):
+        from pas.plugins.tfa.helpers import get_or_create_secret
+
+        secret = get_or_create_secret()
+
+        self.assertIsNotNone(secret)
+
+    def test_get_secret_key(self):
+        from pas.plugins.tfa.helpers import get_secret_key
+
+        key = get_secret_key()
+        self.assertIsNotNone(key)