Skip to content

Commit fc4de13

Browse files
apply make format
1 parent 4be60a6 commit fc4de13

File tree

1 file changed

+62
-18
lines changed

1 file changed

+62
-18
lines changed

tests/test_routes/test_environment_route.py

Lines changed: 62 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -39,20 +39,27 @@ def test_create_env():
3939
}
4040
mock_create_env.assert_called_once_with(Env(latitude=0, longitude=0))
4141

42+
4243
def test_create_env_invalid_input():
43-
response = client.post("/environments/", json={"latitude": "foo", "longitude": "bar"})
44+
response = client.post(
45+
"/environments/", json={"latitude": "foo", "longitude": "bar"}
46+
)
4447
assert response.status_code == 422
4548

49+
4650
def test_create_env_server_error():
4751
with patch.object(
4852
EnvController, "create_env", side_effect=Exception("error")
49-
) as mock_create_env:
53+
):
5054
with pytest.raises(Exception):
5155
response = client.post(
5256
"/environments/", json={"latitude": 0, "longitude": 0}
5357
)
5458
assert response.status_code == 500
55-
assert response.json() == {"detail": "Failed to create environment: error"}
59+
assert response.json() == {
60+
"detail": "Failed to create environment: error"
61+
}
62+
5663

5764
def test_read_env(stub_env):
5865
with patch.object(
@@ -65,23 +72,32 @@ def test_read_env(stub_env):
6572
assert response.json() == expected_content
6673
mock_read_env.assert_called_once_with("123")
6774

75+
6876
def test_read_env_not_found():
6977
with patch.object(
70-
EnvController, "get_env_by_id", side_effect=HTTPException(status_code=404, detail="Environment not found")
78+
EnvController,
79+
"get_env_by_id",
80+
side_effect=HTTPException(
81+
status_code=404, detail="Environment not found"
82+
),
7183
) as mock_read_env:
7284
response = client.get("/environments/123")
7385
assert response.status_code == 404
7486
assert response.json() == {"detail": "Environment not found"}
7587
mock_read_env.assert_called_once_with("123")
7688

89+
7790
def test_read_env_server_error():
7891
with patch.object(
7992
EnvController, "get_env_by_id", side_effect=Exception("error")
80-
) as mock_read_env:
93+
):
8194
with pytest.raises(Exception):
8295
response = client.get("/environments/123")
8396
assert response.status_code == 500
84-
assert response.json() == {"detail": "Failed to read environment: error"}
97+
assert response.json() == {
98+
"detail": "Failed to read environment: error"
99+
}
100+
85101

86102
def test_update_env():
87103
with patch.object(
@@ -101,15 +117,21 @@ def test_update_env():
101117
"123", Env(latitude=1, longitude=1)
102118
)
103119

120+
104121
def test_update_env_invalid_input():
105-
response = client.put("/environments/123", json={"latitude": "foo", "longitude": "bar"})
122+
response = client.put(
123+
"/environments/123", json={"latitude": "foo", "longitude": "bar"}
124+
)
106125
assert response.status_code == 422
107126

127+
108128
def test_update_env_not_found():
109129
with patch.object(
110130
EnvController,
111131
"update_env_by_id",
112-
side_effect=HTTPException(status_code=404, detail="Environment not found"),
132+
side_effect=HTTPException(
133+
status_code=404, detail="Environment not found"
134+
),
113135
) as mock_update_env:
114136
response = client.put(
115137
"/environments/123", json={"longitude": 1, "latitude": 1}
@@ -120,18 +142,22 @@ def test_update_env_not_found():
120142
"123", Env(latitude=1, longitude=1)
121143
)
122144

145+
123146
def test_update_env_server_error():
124147
with patch.object(
125148
EnvController,
126149
"update_env_by_id",
127150
side_effect=Exception("error"),
128-
) as mock_update_env:
151+
):
129152
with pytest.raises(Exception):
130153
response = client.put(
131154
"/environments/123", json={"longitude": 1, "latitude": 1}
132155
)
133156
assert response.status_code == 500
134-
assert response.json() == {"detail": "Failed to update environment: error"}
157+
assert response.json() == {
158+
"detail": "Failed to update environment: error"
159+
}
160+
135161

136162
def test_delete_env():
137163
with patch.object(
@@ -147,6 +173,7 @@ def test_delete_env():
147173
}
148174
mock_delete_env.assert_called_once_with("123")
149175

176+
150177
def test_delete_env_not_found():
151178
with patch.object(
152179
EnvController,
@@ -161,16 +188,20 @@ def test_delete_env_not_found():
161188
}
162189
mock_delete_env.assert_called_once_with("123")
163190

191+
164192
def test_delete_env_server_error():
165193
with patch.object(
166194
EnvController,
167195
"delete_env_by_id",
168196
side_effect=Exception("error"),
169-
) as mock_delete_env:
197+
):
170198
with pytest.raises(Exception):
171199
response = client.delete("/environments/123")
172200
assert response.status_code == 500
173-
assert response.json() == {"detail": "Failed to delete environment: error"}
201+
assert response.json() == {
202+
"detail": "Failed to delete environment: error"
203+
}
204+
174205

175206
def test_simulate_env(stub_env_summary):
176207
with patch.object(
@@ -189,27 +220,34 @@ def test_simulate_env(stub_env_summary):
189220
assert response.json() == expected_content
190221
mock_simulate_env.assert_called_once_with("123")
191222

223+
192224
def test_simulate_env_not_found():
193225
with patch.object(
194226
EnvController,
195227
"simulate_env",
196-
side_effect=HTTPException(status_code=404, detail="Environment not found"),
228+
side_effect=HTTPException(
229+
status_code=404, detail="Environment not found"
230+
),
197231
) as mock_simulate_env:
198232
response = client.get("/environments/123/summary")
199233
assert response.status_code == 404
200234
assert response.json() == {"detail": "Environment not found"}
201235
mock_simulate_env.assert_called_once_with("123")
202236

237+
203238
def test_simulate_env_server_error():
204239
with patch.object(
205240
EnvController,
206241
"simulate_env",
207242
side_effect=Exception("error"),
208-
) as mock_simulate_env:
243+
):
209244
with pytest.raises(Exception):
210245
response = client.get("/environments/123/summary")
211246
assert response.status_code == 500
212-
assert response.json() == {"detail": "Failed to simulate environment: error"}
247+
assert response.json() == {
248+
"detail": "Failed to simulate environment: error"
249+
}
250+
213251

214252
def test_read_rocketpy_env():
215253
with patch.object(
@@ -221,24 +259,30 @@ def test_read_rocketpy_env():
221259
assert response.headers["content-type"] == "application/octet-stream"
222260
mock_read_rocketpy_env.assert_called_once_with("123")
223261

262+
224263
def test_read_rocketpy_env_not_found():
225264
with patch.object(
226265
EnvController,
227266
"get_rocketpy_env_binary",
228-
side_effect=HTTPException(status_code=404, detail="Environment not found"),
267+
side_effect=HTTPException(
268+
status_code=404, detail="Environment not found"
269+
),
229270
) as mock_read_rocketpy_env:
230271
response = client.get("/environments/123/rocketpy")
231272
assert response.status_code == 404
232273
assert response.json() == {"detail": "Environment not found"}
233274
mock_read_rocketpy_env.assert_called_once_with("123")
234275

276+
235277
def test_read_rocketpy_env_server_error():
236278
with patch.object(
237279
EnvController,
238280
"get_rocketpy_env_binary",
239281
side_effect=Exception("error"),
240-
) as mock_read_rocketpy_env:
282+
):
241283
with pytest.raises(Exception):
242284
response = client.get("/environments/123/rocketpy")
243285
assert response.status_code == 500
244-
assert response.json() == {"detail": "Failed to read rocketpy environment: error"}
286+
assert response.json() == {
287+
"detail": "Failed to read rocketpy environment: error"
288+
}

0 commit comments

Comments
 (0)