fix test suite

DeaconDesperado · Apr 16, 2024 · d8c41cd · d8c41cd
1 parent eaf96e1
commit d8c41cd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/flask_mab/__init__.py b/flask_mab/__init__.py
@@ -10,19 +10,13 @@
     :license: BSD, see LICENSE for more details.
 """
 
-from flask import current_app, g, request
+from flask import current_app, request
 import json
 import flask_mab.storage
 from flask_mab.mab import Mab
 import types
-from collections import namedtuple
-from flask import _request_ctx_stack
 from functools import wraps
 
-try:
-    from flask import _app_ctx_stack as stack
-except ImportError:
-    from flask import _request_ctx_stack as stack
 
 __version__ = "2.0.1"
 

diff --git a/tests/test_blueprints.py b/tests/test_blueprints.py
@@ -70,7 +70,7 @@ def test_improper_configuration(self):
             rv = app_client.get("/")
 
         with self.assertRaises(MABConfigException):
-            app_client.set_cookie("localhost", "MAB", '{"some_bandit": "blue"}')
+            app_client.set_cookie("MAB", '{"some_bandit": "blue"}', domain="localhost")
             app_client.get("/reward")
 
     def get_arm(self, headers):