Skip to content

Commit f2c7bd9

Browse files
yt-msMidnighter
authored andcommitted
style: black reformats
1 parent 8adfdbb commit f2c7bd9

File tree

2 files changed

+12
-18
lines changed

2 files changed

+12
-18
lines changed

src/structurizr/api/structurizr_client.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -223,9 +223,7 @@ def unlock_workspace(self) -> bool:
223223
request = self._client.build_request(
224224
"DELETE", self._lock_url, params=self._params
225225
)
226-
request.headers.update(
227-
self._add_headers(request)
228-
)
226+
request.headers.update(self._add_headers(request))
229227
response = self._client.send(request)
230228
response.raise_for_status()
231229
response = APIResponse.parse_raw(response.text)
@@ -254,7 +252,7 @@ def _add_headers(
254252
255253
"""
256254
method = request.method
257-
url_path = request.url.raw_path.decode('ascii')
255+
url_path = request.url.raw_path.decode("ascii")
258256
definition_md5 = self._md5(content)
259257
nonce = self._number_once()
260258
message_digest = self._message_digest(

tests/unit/api/test_structurizr_client.py

Lines changed: 10 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -114,24 +114,20 @@ def test_archive_workspace(client, mocker):
114114

115115
def test_add_headers_authentication(client: StructurizrClient, mocker):
116116
"""Validate the headers are added correctly, including authentication."""
117-
mocker.patch.object(
118-
client,
119-
"_number_once",
120-
return_value="1529225966174"
121-
)
117+
mocker.patch.object(client, "_number_once", return_value="1529225966174")
122118
request = client._client.build_request("GET", client._workspace_url)
123119
headers = client._add_headers(request)
124-
assert headers['Nonce'] == "1529225966174"
125-
assert headers['X-Authorization'] == (
126-
'7f4e4edc-f61c-4ff2-97c9-ea4bc2a7c98c:'
127-
'ZmJhNTVkMDM2NGEwN2I5YjRhMDgwZWNhMjA0ODIzZD'
128-
'kyMTg3YzliMzVhMjBlNmM4ZjAxMDAwOGU4OGJlODEwMQ=='
120+
assert headers["Nonce"] == "1529225966174"
121+
assert headers["X-Authorization"] == (
122+
"7f4e4edc-f61c-4ff2-97c9-ea4bc2a7c98c:"
123+
"ZmJhNTVkMDM2NGEwN2I5YjRhMDgwZWNhMjA0ODIzZD"
124+
"kyMTg3YzliMzVhMjBlNmM4ZjAxMDAwOGU4OGJlODEwMQ=="
129125
)
130-
assert 'Content-MD5' not in headers
131-
assert 'Content-Type' not in headers
126+
assert "Content-MD5" not in headers
127+
assert "Content-Type" not in headers
132128

133129
# Check the additional headers needed for PUTs
134130
request = client._client.build_request("PUT", client._workspace_url)
135131
headers = client._add_headers(request, content="Hello", content_type="World")
136-
assert headers['Content-MD5'] == 'OGIxYTk5NTNjNDYxMTI5NmE4MjdhYmY4YzQ3ODA0ZDc='
137-
assert headers['Content-Type'] == 'World'
132+
assert headers["Content-MD5"] == "OGIxYTk5NTNjNDYxMTI5NmE4MjdhYmY4YzQ3ODA0ZDc="
133+
assert headers["Content-Type"] == "World"

0 commit comments

Comments
 (0)