Add more test cases

dowjones · Jan 23, 2023 · 980b1fd · 980b1fd
1 parent feb6767
commit 980b1fd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/unit_test.py b/tests/unit_test.py
@@ -841,10 +841,11 @@ def test_user_mfa_challenge_with_no_mfas(sample_headers, sample_json_response):
 @pytest.mark.parametrize(
     "return_value,side_effect,expected",
     [
-        ({"status": "SUCCESS"}, None, 0),
+        ({"status": "SUCCESS", "sessionToken": "pytest"}, None, 0),
         ({"status": "SUCCESS", "factorResult": "SUCCESS"}, None, 0),
         ({"status": "MFA_CHALLENGE", "factorResult": "REJECTED"}, None, 2),
         ({"status": "MFA_CHALLENGE", "factorResult": "TIMEOUT"}, None, 2),
+        ({"status": "UNKNOWN", "factorResult": "UNKNOWN"}, None, 2),
         (
             {
                 "status": "MFA_CHALLENGE",