|
2 | 2 | from test_plus.test import TestCase |
3 | 3 | from app_code_share.models import CodeShare |
4 | 4 | from django.utils.crypto import get_random_string |
| 5 | +from django.test.client import RequestFactory |
5 | 6 |
|
| 7 | +from code_share.views import ( |
| 8 | + page_not_found_view, |
| 9 | + my_custom_error_view, |
| 10 | + permission_denied_view, |
| 11 | + bad_request_view, |
| 12 | +) |
6 | 13 |
|
7 | 14 |
|
8 | 15 | class TestCodeShare(TestCase): |
@@ -35,3 +42,30 @@ def test_detail_resolve(self): |
35 | 42 | resolve('/app/' + self.hash_value + '/').view_name, |
36 | 43 | 'code_share:view_by_hash' |
37 | 44 | ) |
| 45 | + |
| 46 | + |
| 47 | +class TestMyErrorPages(TestCase): |
| 48 | + |
| 49 | + def test_400_error(self): |
| 50 | + factory = RequestFactory() |
| 51 | + request = factory.get('/400') |
| 52 | + response = bad_request_view(request) |
| 53 | + self.assertEqual(response.status_code, 400) |
| 54 | + |
| 55 | + def test_403_error(self): |
| 56 | + factory = RequestFactory() |
| 57 | + request = factory.get('/403') |
| 58 | + response = permission_denied_view(request) |
| 59 | + self.assertEqual(response.status_code, 403) |
| 60 | + |
| 61 | + def test_404_error(self): |
| 62 | + factory = RequestFactory() |
| 63 | + request = factory.get('/404') |
| 64 | + response = page_not_found_view(request) |
| 65 | + self.assertEqual(response.status_code, 404) |
| 66 | + |
| 67 | + def test_500_error(self): |
| 68 | + factory = RequestFactory() |
| 69 | + request = factory.get('/500') |
| 70 | + response = my_custom_error_view(request) |
| 71 | + self.assertEqual(response.status_code, 500) |
0 commit comments