@@ -17,28 +17,36 @@ async def return_401_handler(with_exception):
1717 """Return an HTTP401 error."""
1818 if with_exception :
1919 raise HTTPUnauthorized ()
20- return Response (status = 401 )
20+ return FileResponse (
21+ status = 401 , path = f"{ os .getcwd ()} /swift_browser_ui_frontend/dist/401.html"
22+ )
2123
2224
2325async def return_403_handler (with_exception ):
2426 """Return an HTTP403 error."""
2527 if with_exception :
2628 raise HTTPForbidden ()
27- return Response (status = 403 )
29+ return FileResponse (
30+ status = 403 , path = f"{ os .getcwd ()} /swift_browser_ui_frontend/dist/403.html"
31+ )
2832
2933
3034async def return_404_handler (with_exception ):
3135 """Return or raise an HTTP404 error."""
3236 if with_exception :
3337 raise HTTPNotFound ()
34- return Response (status = 404 )
38+ return FileResponse (
39+ status = 404 , path = f"{ os .getcwd ()} /swift_browser_ui_frontend/dist/404.html"
40+ )
3541
3642
3743async def return_400_handler (with_exception ):
3844 """Return or raise an HTTP400 error."""
3945 if with_exception :
4046 raise HTTPClientError ()
41- return Response (status = 400 )
47+ return FileResponse (
48+ status = 400 , path = f"{ os .getcwd ()} /swift_browser_ui_frontend/dist/400.html"
49+ )
4250
4351
4452async def return_200_handler (_ ):
@@ -89,81 +97,51 @@ async def test_check_session_fail_no_raise(self):
8997
9098 async def test_401_return (self ):
9199 """Test 401 middleware when the 401 status is returned."""
92- patch_setd = unittest .mock .patch (
93- "swift_browser_ui.ui.middlewares.setd" ,
94- new = {"static_directory" : os .getcwd () + "/swift_browser_ui_frontend/dist" },
100+ resp = await swift_browser_ui .ui .middlewares .error_middleware (
101+ None , return_401_handler
95102 )
96- with patch_setd :
97- resp = await swift_browser_ui .ui .middlewares .error_middleware (
98- None , return_401_handler
99- )
100- self .assertEqual (resp .status , 401 )
101- self .assertIsInstance (resp , Response )
103+ self .assertEqual (resp .status , 401 )
104+ self .assertIsInstance (resp , FileResponse )
102105
103106 async def test_401_exception (self ):
104107 """Test 401 middleware when the 401 status is risen."""
105- patch_setd = unittest .mock .patch (
106- "swift_browser_ui.ui.middlewares.setd" ,
107- new = {"static_directory" : os .getcwd () + "/swift_browser_ui_frontend/dist" },
108+ resp = await swift_browser_ui .ui .middlewares .error_middleware (
109+ True , return_401_handler
108110 )
109- with patch_setd :
110- resp = await swift_browser_ui .ui .middlewares .error_middleware (
111- True , return_401_handler
112- )
113- self .assertEqual (resp .status , 401 )
114- self .assertIsInstance (resp , Response )
111+ self .assertEqual (resp .status , 401 )
112+ self .assertIsInstance (resp , FileResponse )
115113
116114 async def test_403_return (self ):
117115 """Test 403 middleware when the 403 status is returned."""
118- patch_setd = unittest .mock .patch (
119- "swift_browser_ui.ui.middlewares.setd" ,
120- new = {"static_directory" : os .getcwd () + "/swift_browser_ui_frontend/dist" },
116+ resp = await swift_browser_ui .ui .middlewares .error_middleware (
117+ None , return_403_handler
121118 )
122- with patch_setd :
123- resp = await swift_browser_ui .ui .middlewares .error_middleware (
124- None , return_403_handler
125- )
126- self .assertEqual (resp .status , 403 )
127- self .assertIsInstance (resp , Response )
119+ self .assertEqual (resp .status , 403 )
120+ self .assertIsInstance (resp , FileResponse )
128121
129122 async def test_403_exception (self ):
130123 """Test 403 middleware when the 403 status is risen."""
131- patch_setd = unittest .mock .patch (
132- "swift_browser_ui.ui.middlewares.setd" ,
133- new = {"static_directory" : os .getcwd () + "/swift_browser_ui_frontend/dist" },
124+ resp = await swift_browser_ui .ui .middlewares .error_middleware (
125+ True , return_403_handler
134126 )
135- with patch_setd :
136- resp = await swift_browser_ui .ui .middlewares .error_middleware (
137- True , return_403_handler
138- )
139- self .assertEqual (resp .status , 403 )
140- self .assertIsInstance (resp , Response )
127+ self .assertEqual (resp .status , 403 )
128+ self .assertIsInstance (resp , FileResponse )
141129
142130 async def test_404_return (self ):
143131 """Test 404 middleware when the 404 status is returned."""
144- patch_setd = unittest .mock .patch (
145- "swift_browser_ui.ui.middlewares.setd" ,
146- new = {"static_directory" : os .getcwd () + "/swift_browser_ui_frontend/dist" },
132+ resp = await swift_browser_ui .ui .middlewares .error_middleware (
133+ None , return_404_handler
147134 )
148- with patch_setd :
149- resp = await swift_browser_ui .ui .middlewares .error_middleware (
150- None , return_404_handler
151- )
152- self .assertEqual (resp .status , 404 )
153- self .assertIsInstance (resp , Response )
135+ self .assertEqual (resp .status , 404 )
136+ self .assertIsInstance (resp , FileResponse )
154137
155138 async def test_404_exception (self ):
156139 """Test 404 middlewrae when the 404 status is risen."""
157- patch_setd = unittest .mock .patch (
158- "swift_browser_ui.ui.middlewares.setd" ,
159- new = {"static_directory" : os .getcwd () + "/swift_browser_ui_frontend/dist" },
140+ resp = await swift_browser_ui .ui .middlewares .error_middleware (
141+ True , return_404_handler
160142 )
161- with patch_setd :
162- resp = await swift_browser_ui .ui .middlewares .error_middleware (
163- True , return_404_handler
164- )
165- self .assertEqual (resp .status , 404 )
166- self .assertIsInstance (resp , Response )
143+ self .assertEqual (resp .status , 404 )
144+ self .assertIsInstance (resp , FileResponse )
167145
168146 async def test_error_middleware_no_error (self ):
169147 """Test the general error middleware with correct status."""
@@ -178,11 +156,7 @@ async def test_error_middleware_no_error(self):
178156
179157 async def test_error_middleware_non_handled_raise (self ):
180158 """Test the general error middleware with other status code."""
181- patch_setd = unittest .mock .patch (
182- "swift_browser_ui.ui.middlewares.setd" ,
183- new = {"static_directory" : os .getcwd () + "/swift_browser_ui_frontend/dist" },
184- )
185- with self .assertRaises (HTTPClientError ), patch_setd :
159+ with self .assertRaises (HTTPClientError ):
186160 await swift_browser_ui .ui .middlewares .error_middleware (
187161 True , return_400_handler
188162 )
0 commit comments