|
| 1 | +import types |
| 2 | +import tempfile |
| 3 | +from itertools import islice |
| 4 | + |
1 | 5 | from django.urls import reverse |
2 | 6 | from django.test import TestCase, Client |
3 | 7 | from django.contrib.auth import get_user_model |
4 | 8 |
|
| 9 | +from ..utils import get_log_files, readlines_reverse |
| 10 | + |
5 | 11 | User = get_user_model() |
6 | 12 |
|
7 | 13 |
|
8 | 14 | class TestLogsViewer(TestCase): |
9 | | - |
10 | 15 | def setUp(self): |
11 | 16 | self.user = User.objects.create_user( |
12 | | - username='john', |
13 | | - password='Foobar@1234', |
| 17 | + username="john", |
| 18 | + password="Foobar@1234", |
14 | 19 | is_active=True, |
15 | 20 | is_staff=True, |
16 | | - is_superuser=True |
| 21 | + is_superuser=True, |
17 | 22 | ) |
18 | 23 | self.client_login = Client() |
19 | 24 | self.client_login.force_login(self.user) |
20 | 25 | self.client_anon = Client() |
21 | | - self.url = reverse('log_file_view') |
| 26 | + self.url = reverse("log_file_view") |
22 | 27 |
|
23 | 28 | def test_redirection(self): |
24 | 29 | response = self.client_anon.get(self.url) |
25 | 30 | self.assertEqual(response.status_code, 302) |
26 | 31 |
|
27 | 32 | def test_logs_viewer(self): |
28 | | - # response = self.client_login.get(self.url) |
29 | | - # self.assertEqual(response.status_code, 200) |
30 | | - # self.assertContains(response, 'Django Log Viewer') |
31 | | - # self.assertTemplateUsed(response, 'log_viewer/logfile_viewer.html') |
| 33 | + response = self.client_login.get(self.url) |
| 34 | + self.assertEqual(response.status_code, 200) |
| 35 | + self.assertContains(response, "Django Log Viewer") |
| 36 | + self.assertTemplateUsed(response, "log_viewer/logfile_viewer.html") |
| 37 | + |
| 38 | + def test_get_log_files(self): |
| 39 | + with tempfile.TemporaryDirectory() as log_dir: |
| 40 | + open("%s/default.log" % log_dir, "a").close() |
| 41 | + result = get_log_files(log_dir, 1, 1) |
| 42 | + self.assertEqual( |
| 43 | + result, |
| 44 | + { |
| 45 | + "logs": { |
| 46 | + "": ["default.log"], |
| 47 | + }, |
| 48 | + "next_page_files": 2, |
| 49 | + "last_files": True, |
| 50 | + }, |
| 51 | + ) |
| 52 | + |
| 53 | + def test_readlines_reverse(self): |
| 54 | + with tempfile.TemporaryDirectory() as log_dir: |
| 55 | + with open("%s/default.log" % log_dir, "w") as qfile: |
| 56 | + qfile.write( |
| 57 | + "[WARNING] 2022-08-26 13:02:46,070 django.request: Not Found: /404/" |
| 58 | + ) |
| 59 | + qfile.write( |
| 60 | + "[ERROR] 2022-08-26 13:02:50,962 django.request: Internal Server Error: /admin/" |
| 61 | + ) |
| 62 | + qfile.write( |
| 63 | + "[INFO] 2022-08-26 13:02:36,604 django.utils.autoreload: Watching for file changes" |
| 64 | + ) |
| 65 | + qfile.close() |
32 | 66 |
|
33 | | - # FIXME: django.urls.exceptions.NoReverseMatch: 'admin' is not a registered namespace |
34 | | - ... |
| 67 | + with open( |
| 68 | + "%s/default.log" % log_dir, encoding="utf8", errors="ignore" |
| 69 | + ) as qfile: |
| 70 | + lines = readlines_reverse(qfile, exclude=None) |
| 71 | + lines_string = " ".join(map(str, lines)) |
| 72 | + self.assertTrue(isinstance(lines, types.GeneratorType)) |
| 73 | + self.assertIn("[WARNING]", lines_string) |
| 74 | + self.assertIn("[ERROR]", lines_string) |
| 75 | + self.assertIn("[INFO]", lines_string) |
0 commit comments