Add TokenVerifier.force

nens · Aug 16, 2023 · e9c611f · e9c611f
1 parent 3d4aa45
commit e9c611f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/CHANGES.md b/CHANGES.md
@@ -4,7 +4,7 @@
 0.3.1 (unreleased)
 ------------------
 
-- Nothing changed yet.
+- Added `TokenVerifier.force()` for testing purposes.
 
 
 0.3.0 (2023-08-16)

diff --git a/clean_python/oauth2/token_verifier.py b/clean_python/oauth2/token_verifier.py
@@ -45,13 +45,24 @@ class OAuth2SPAClientSettings(BaseModel):
 
 
 class BaseTokenVerifier:
+    def force(self, token: Token) -> None:
+        raise NotImplementedError()
+
     def __call__(self, authorization: Optional[str]) -> Token:
         raise NotImplementedError()
 
 
 class NoAuthTokenVerifier(BaseTokenVerifier):
+    def __init__(self):
+        self.token = Token(
+            claims={"sub": "DEV", "username": "dev", "scope": "superuser"}
+        )
+
+    def force(self, token: Token) -> None:
+        self.token = token
+
     def __call__(self, authorization: Optional[str]) -> Token:
-        return Token(claims={"sub": "DEV", "username": "dev", "scope": "superuser"})
+        return self.token
 
 
 class TokenVerifier(BaseTokenVerifier):