diff --git a/datacat_integration/auth.py b/datacat_integration/auth.py new file mode 100644 index 0000000000000000000000000000000000000000..4cf6c31f1b77fbffede93e03fd975b2f2bf0d49d --- /dev/null +++ b/datacat_integration/auth.py @@ -0,0 +1,11 @@ +import requests +import requests.auth + +class BearerAuth(requests.auth.AuthBase): + def __init__(self, token: str) -> None: + super().__init__() + self.token = token + + def __call__(self, r: requests.Request) -> requests.Request: + r.headers['Authorization'] = 'Bearer {}'.format(self.token) + return r \ No newline at end of file diff --git a/tests/__init__.py b/tests/__init__.py new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/tests/dummy_tests/test_dummy.py b/tests/dummy_tests/test_dummy.py deleted file mode 100644 index e1e838b99eb3b34c37116c0d689037ca7cd90696..0000000000000000000000000000000000000000 --- a/tests/dummy_tests/test_dummy.py +++ /dev/null @@ -1,7 +0,0 @@ -from unittest import TestCase - -class DummyTest(TestCase): - - def test_dummy(self): - print("Test Print") - self.assertEqual(4, 2*2) \ No newline at end of file diff --git a/tests/test_auth.py b/tests/test_auth.py new file mode 100644 index 0000000000000000000000000000000000000000..db3f1cae21c5561fd9419b3ceb5a7550fa2f4c5a --- /dev/null +++ b/tests/test_auth.py @@ -0,0 +1,20 @@ +from unittest import TestCase + +from requests import Request + +from datacat_integration.auth import BearerAuth + +class AuthTest(TestCase): + def test_auth_creation(self): + token = "dummy_token_text" + header_key = "Authorization" + header_value = f"Bearer {token}" + auth: BearerAuth = BearerAuth(token) + r: Request = Request(auth=auth) + + auth.__call__(r) + + headers = r.headers + + self.assertTrue(headers[header_key], header_value) +