Blackfied code

Author: Henry Chan

setup.py

@@ -4,54 +4,51 @@
 
 base_dir = os.path.dirname(__file__)
 
-with open('README.md', 'r') as f:
+with open("README.md", "r") as f:
     long_description = f.read()
 
 about = {}
-with open(os.path.join(base_dir, 'workos', '__about__.py')) as f:
+with open(os.path.join(base_dir, "workos", "__about__.py")) as f:
     exec(f.read(), about)
 
 dev_requirements = [
-    'flake8',
-    'pytest==4.6.9',
-    'pytest-cov==2.8.1',
-    'six==1.13.0',
+    "flake8",
+    "pytest==4.6.9",
+    "pytest-cov==2.8.1",
+    "six==1.13.0",
 ]
 if sys.version_info.major == 3:
-    dev_requirements.extend([
-        'black==19.10b0',
-        'twine==3.1.1',
-    ])
+    dev_requirements.extend(
+        ["black==19.10b0", "twine==3.1.1",]
+    )
 
 setup(
-    name=about['__package_name__'],
-    version=about['__version__'],
-    author=about['__author__'],
-    author_email=about['__author_email__'],
-    url=about['__package_url__'],
-    description=about['__description__'],
+    name=about["__package_name__"],
+    version=about["__version__"],
+    author=about["__author__"],
+    author_email=about["__author_email__"],
+    url=about["__package_url__"],
+    description=about["__description__"],
     long_description=long_description,
-    long_description_content_type='text/markdown',
-    packages=find_packages(exclude=['tests*', ]),
+    long_description_content_type="text/markdown",
+    packages=find_packages(exclude=["tests*",]),
     zip_safe=False,
-    license=about['__license__'],
-    install_requires=['requests>=2.22.0'],
-    extras_require={
-        'dev': dev_requirements,
-    },
+    license=about["__license__"],
+    install_requires=["requests>=2.22.0"],
+    extras_require={"dev": dev_requirements,},
     classifiers=[
-        'Development Status :: 1 - Planning',
-        'Intended Audience :: Developers',
-        'License :: OSI Approved :: MIT License',
-        'Operating System :: OS Independent',
-        'Programming Language :: Python',
-        'Programming Language :: Python :: 2',
-        'Programming Language :: Python :: 2.7',
-        'Programming Language :: Python :: 3',
-        'Programming Language :: Python :: 3.4',
-        'Programming Language :: Python :: 3.5',
-        'Programming Language :: Python :: 3.6',
-        'Programming Language :: Python :: 3.7',
-        'Programming Language :: Python :: 3.8',
+        "Development Status :: 1 - Planning",
+        "Intended Audience :: Developers",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
+        "Programming Language :: Python",
+        "Programming Language :: Python :: 2",
+        "Programming Language :: Python :: 2.7",
+        "Programming Language :: Python :: 3",
+        "Programming Language :: Python :: 3.4",
+        "Programming Language :: Python :: 3.5",
+        "Programming Language :: Python :: 3.6",
+        "Programming Language :: Python :: 3.7",
+        "Programming Language :: Python :: 3.8",
     ],
-)
+)

tests/conftest.py

@@ -3,6 +3,7 @@
 
 import workos
 
+
 class MockResponse(object):
     def __init__(self, response_dict, status_code, headers=None):
         self.response_dict = response_dict
@@ -12,18 +13,22 @@ def __init__(self, response_dict, status_code, headers=None):
     def json(self):
         return self.response_dict
 
+
 @pytest.fixture
 def set_api_key(monkeypatch):
-    monkeypatch.setattr(workos, 'api_key', 'sk_abdsomecharactersm284')
+    monkeypatch.setattr(workos, "api_key", "sk_abdsomecharactersm284")
+
 
 @pytest.fixture
 def set_project_id(monkeypatch):
-    monkeypatch.setattr(workos, 'project_id', 'project_b27needthisforssotemxo')
+    monkeypatch.setattr(workos, "project_id", "project_b27needthisforssotemxo")
+
 
 @pytest.fixture
 def set_api_key_and_project_id(set_api_key, set_project_id):
     pass
 
+
 @pytest.fixture
 def mock_request_method(monkeypatch):
     def inner(method, response_dict, status_code, headers=None):
@@ -34,6 +39,7 @@ def mock(*args, **kwargs):
 
     return inner
 
+
 @pytest.fixture
 def capture_and_mock_requests(monkeypatch):
     def inner():
@@ -43,9 +49,9 @@ def capture(*args, **kwargs):
             captured_requests.append((args, kwargs))
             return MockResponse({}, 200)
 
-        monkeypatch.setattr(requests, 'get', capture)
-        monkeypatch.setattr(requests, 'post', capture)
+        monkeypatch.setattr(requests, "get", capture)
+        monkeypatch.setattr(requests, "post", capture)
 
         return captured_requests
 
-    return inner
+    return inner

tests/test_client.py

@@ -3,6 +3,7 @@
 from workos import client
 from workos.exceptions import ConfigurationException
 
+
 class TestClient(object):
     @pytest.fixture(autouse=True)
     def setup(self):
@@ -17,22 +18,22 @@ def test_initialize_sso_missing_api_key(self, set_project_id):
 
         message = str(ex)
 
-        assert 'api_key' in message
-        assert 'project_id' not in message
+        assert "api_key" in message
+        assert "project_id" not in message
 
     def test_initialize_sso_missing_project_id(self, set_api_key):
         with pytest.raises(ConfigurationException) as ex:
             client.sso
-        
+
         message = str(ex)
 
-        assert 'project_id' in message
-        assert 'api_key' not in message
-    
+        assert "project_id" in message
+        assert "api_key" not in message
+
     def test_initialize_sso_missing_api_key_and_project_id(self):
         with pytest.raises(ConfigurationException) as ex:
             client.sso
 
         message = str(ex)
 
-        assert all(setting in message for setting in ('api_key', 'project_id',))
+        assert all(setting in message for setting in ("api_key", "project_id",))

tests/test_sso.py

@@ -7,62 +7,62 @@
 from workos.sso import SSO
 from workos.utils.request import RESPONSE_TYPE_CODE
 
+
 class TestSSO(object):
     @pytest.fixture(autouse=True)
     def setup(self, set_api_key_and_project_id):
-        self.customer_domain = 'workos.com'
-        self.redirect_uri = 'https://localhost/auth/callback'
-        self.state = { 'things': 'with_stuff', }
+        self.customer_domain = "workos.com"
+        self.redirect_uri = "https://localhost/auth/callback"
+        self.state = {
+            "things": "with_stuff",
+        }
 
         self.sso = SSO()
 
     @pytest.fixture
     def mock_profile(self):
         return {
-            'id': 'prof_01DWAS7ZQWM70PV93BFV1V78QV',
-            'email': '[email protected]',
-            'first_name': 'WorkOS',
-            'last_name': 'Demo',
-            'connection_type': 'OktaSAML',
-            'idp_id': '00u1klkowm8EGah2H357'
+            "id": "prof_01DWAS7ZQWM70PV93BFV1V78QV",
+            "email": "[email protected]",
+            "first_name": "WorkOS",
+            "last_name": "Demo",
+            "connection_type": "OktaSAML",
+            "idp_id": "00u1klkowm8EGah2H357",
         }
 
     def test_authorization_url_has_expected_query_params(self):
         authorization_url = self.sso.get_authorization_url(
-            self.customer_domain,
-            self.redirect_uri,
-            state=self.state
+            self.customer_domain, self.redirect_uri, state=self.state
         )
 
         parsed_url = urlparse(authorization_url)
 
         assert dict(parse_qsl(parsed_url.query)) == {
-            'domain': self.customer_domain,
-            'client_id': workos.project_id,
-            'redirect_uri': self.redirect_uri,
-            'response_type': RESPONSE_TYPE_CODE,
-            'state': json.dumps(self.state),
+            "domain": self.customer_domain,
+            "client_id": workos.project_id,
+            "redirect_uri": self.redirect_uri,
+            "response_type": RESPONSE_TYPE_CODE,
+            "state": json.dumps(self.state),
         }
 
-
     def test_get_profile_returns_expected_workosprofile_object(
         self, mock_profile, mock_request_method
     ):
         response_dict = {
-            'profile': {
-                'object': 'profile',
-                'id': mock_profile['id'],
-                'email': mock_profile['email'],
-                'first_name': mock_profile['first_name'],
-                'connection_type': mock_profile['connection_type'],
-                'last_name': mock_profile['last_name'],
-                'idp_id': mock_profile['idp_id'],
+            "profile": {
+                "object": "profile",
+                "id": mock_profile["id"],
+                "email": mock_profile["email"],
+                "first_name": mock_profile["first_name"],
+                "connection_type": mock_profile["connection_type"],
+                "last_name": mock_profile["last_name"],
+                "idp_id": mock_profile["idp_id"],
             },
-            'access_token': '01DY34ACQTM3B1CSX1YSZ8Z00D',
+            "access_token": "01DY34ACQTM3B1CSX1YSZ8Z00D",
         }
 
-        mock_request_method('post', response_dict, 200)
+        mock_request_method("post", response_dict, 200)
 
         profile = self.sso.get_profile(123)
 
-        assert profile.to_dict() == mock_profile
+        assert profile.to_dict() == mock_profile

tests/utils/test_requests.py

@@ -1,7 +1,9 @@
 import pytest
 
 from workos.exceptions import (
-    AuthenticationException, AuthorizationException, BadRequestException,
+    AuthenticationException,
+    AuthorizationException,
+    BadRequestException,
     ServerException,
 )
 from workos.utils.request import RequestHelper, BASE_HEADERS
@@ -13,6 +15,7 @@
     500: ServerException,
 }
 
+
 class TestRequestHelper(object):
     def test_set_base_api_url(self):
         pass
@@ -23,29 +26,29 @@ def test_request_raises_expected_exception_for_status_code(
         request_helper = RequestHelper()
 
         for status_code, exception in STATUS_CODE_TO_EXCEPTION_MAPPING.items():
-            mock_request_method('get', {}, status_code)
+            mock_request_method("get", {}, status_code)
 
             with pytest.raises(exception):
-                request_helper.request('bad_place')
+                request_helper.request("bad_place")
 
     def test_request_exceptions_include_expected_request_data(
         self, mock_request_method
     ):
         request_helper = RequestHelper()
 
-        request_id = 'request-123'
-        response_message = 'stuff happened'
+        request_id = "request-123"
+        response_message = "stuff happened"
 
         for status_code, exception in STATUS_CODE_TO_EXCEPTION_MAPPING.items():
             mock_request_method(
-                'get',
-                {'message': response_message, },
+                "get",
+                {"message": response_message,},
                 status_code,
-                headers={'X-Request-ID': request_id}
+                headers={"X-Request-ID": request_id},
             )
 
             try:
-                request_helper.request('bad_place')
+                request_helper.request("bad_place")
             except exception as ex:
                 assert ex.message == response_message
                 assert ex.request_id == request_id
@@ -56,17 +59,14 @@ def test_request_exceptions_include_expected_request_data(
     def test_request_bad_body_raises_expected_exception_with_request_data(
         self, mock_request_method
     ):
-        request_id = 'request-123'
+        request_id = "request-123"
 
         mock_request_method(
-            'get',
-            'this_isnt_json',
-            200,
-            headers={'X-Request-ID': request_id}
+            "get", "this_isnt_json", 200, headers={"X-Request-ID": request_id}
         )
 
         try:
-            RequestHelper().request('bad_place')
+            RequestHelper().request("bad_place")
         except ServerException as ex:
             assert ex.message == None
             assert ex.request_id == request_id
@@ -77,11 +77,11 @@ def test_request_bad_body_raises_expected_exception_with_request_data(
     def test_request_includes_base_headers(self, capture_and_mock_requests):
         requests = capture_and_mock_requests()
 
-        RequestHelper().request('ok_place')
+        RequestHelper().request("ok_place")
 
         assert len(requests) == 1
 
-        base_headers = set(BASE_HEADERS.items()) 
-        headers = set(requests[0][1]['headers'].items())
+        base_headers = set(BASE_HEADERS.items())
+        headers = set(requests[0][1]["headers"].items())
 
-        assert base_headers.issubset(headers)
+        assert base_headers.issubset(headers)