feat: add missing handling for tests

- resetoverridelogs test endpoint
- handle totp overrides for tests

Author: namsnath

tests/test-server/app.py

@@ -744,6 +744,12 @@ def reset_override_params_api():
     return jsonify({"ok": True})
 
 
+@app.route("/test/resetoverridelogs", methods=["GET"])  # type: ignore
+def reset_override_logs():
+    override_logging.reset_override_logs()
+    return jsonify({"ok": True})
+
+
 @app.route("/test/getoverridelogs", methods=["GET"])  # type: ignore
 def get_override_logs():
     return jsonify({"logs": override_logging.override_logs})

tests/test-server/test_functions_mapper.py

@@ -142,6 +142,10 @@ async def get_mfa_requirements_for_auth(
                 required_secondary_factors_for_tenant: Any,
                 user_context: Dict[str, Any],
             ) -> MFARequirementList:
+                # Test specifies an override, return the required data
+                if 'getMFARequirementsForAuth:async()=>["totp"]' in eval_str:
+                    return ["totp"]
+
                 return ["otp-phone"] if user_context.get("requireFactor") else []
 
             original_implementation.get_mfa_requirements_for_auth = (