From 9a87d9a6b0a50552201a16337aead9f6929f09b2 Mon Sep 17 00:00:00 2001
From: Paul V Craven <paul.craven@optimizley.com>
Date: Fri, 25 Apr 2025 10:31:49 -0500
Subject: [PATCH] Import CSRFProtect from a better spot so prisma picks it up

---
 tests/testapp/application.py   | 3 ++-
 tests/testapp/requirements.txt | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/testapp/application.py b/tests/testapp/application.py
index 116efc66..af5f5b33 100644
--- a/tests/testapp/application.py
+++ b/tests/testapp/application.py
@@ -17,7 +17,8 @@
 from os import environ
 
 import user_profile_service
-from flask import CSRFProtect, Flask, request
+from flask import Flask, request
+from flask_wtf.csrf import CSRFProtect
 
 from optimizely import logger, optimizely
 from optimizely.helpers import enums
diff --git a/tests/testapp/requirements.txt b/tests/testapp/requirements.txt
index 4b70123b..dae26c1f 100644
--- a/tests/testapp/requirements.txt
+++ b/tests/testapp/requirements.txt
@@ -1 +1,2 @@
-Flask==2.2.5
+Flask==3.1.0
+flask-wtf==1.2.2
\ No newline at end of file