Continued:

[fba.git] / fba / helpers / cookies.py

diff --git a/fba/helpers/cookies.py b/fba/helpers/cookies.py
index d98df82f0c0ab1033b91e9decfa75e987b4a646a..15f4fc77f1ea08295d773fc0ccc52ac8c37a7a1f 100644 (file)
--- a/fba/helpers/cookies.py
+++ b/fba/helpers/cookies.py
@@ -16,6 +16,7 @@
 
 import logging
 
+from fba.helpers import blacklist
 from fba.helpers import domain as domain_helper
 
 logging.basicConfig(level=logging.INFO)
@@ -27,7 +28,10 @@ _cookies = {}
 def store(domain: str, cookies: dict):
     logger.debug("domain='%s',cookies()=%d - CALLED!", domain, len(cookies))
     domain_helper.raise_on(domain)
-    if not isinstance(cookies, dict):
+
+    if blacklist.is_blacklisted(domain):
+        raise ValueError(f"domain='{domain}' is blacklisted but function was invoked")
+    elif not isinstance(cookies, dict):
         raise ValueError(f"Parameter cookies[]='{type(cookies)}' is not of type 'dict'")
 
     _cookies[domain] = cookies
@@ -38,6 +42,9 @@ def get_all(domain: str) -> dict:
     logger.debug("domain='%s' - CALLED!", domain)
     domain_helper.raise_on(domain)
 
+    if blacklist.is_blacklisted(domain):
+        raise ValueError(f"domain='{domain}' is blacklisted but function was invoked")
+
     if domain not in _cookies:
         return dict()
 
@@ -48,6 +55,9 @@ def has(domain: str) -> bool:
     logger.debug("domain='%s' - CALLED!", domain)
     domain_helper.raise_on(domain)
 
+    if blacklist.is_blacklisted(domain):
+        raise ValueError(f"domain='{domain}' is blacklisted but function was invoked")
+
     has_cookies = domain in _cookies
 
     logger.debug("has_cookies='%s' - EXIT!", has_cookies)
@@ -57,6 +67,9 @@ def clear(domain: str):
     logger.debug("domain='%s' - CALLED!", domain)
     domain_helper.raise_on(domain)
 
+    if blacklist.is_blacklisted(domain):
+        raise ValueError(f"domain='{domain}' is blacklisted but function was invoked")
+
     if has(domain):
         logger.debug("Removing cookies for domain='%s' ...", domain)
         del _cookies[domain]