@@ -39,20 +39,27 @@ def test_create_env():
3939 }
4040 mock_create_env .assert_called_once_with (Env (latitude = 0 , longitude = 0 ))
4141
42+
4243def 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+
4650def 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
5764def 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+
6876def 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+
7790def 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
86102def 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+
104121def 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+
108128def 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+
123146def 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
136162def 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+
150177def 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+
164192def 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
175206def 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+
192224def 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+
203238def 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
214252def 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+
224263def 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+
235277def 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