diff options
| -rw-r--r-- | pydis_site/apps/api/tests/test_dblogger.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/pydis_site/apps/api/tests/test_dblogger.py b/pydis_site/apps/api/tests/test_dblogger.py index ec2ac017..4f0692c2 100644 --- a/pydis_site/apps/api/tests/test_dblogger.py +++ b/pydis_site/apps/api/tests/test_dblogger.py @@ -9,7 +9,8 @@ from ..models import LogEntry class DatabaseLogHandlerTests(TestCase): def test_logs_to_database(self): - logger = logging.getLogger("test_logger") + module_basename = __name__.split('.')[-1] + logger = logging.getLogger(module_basename) logger.handlers = [DatabaseLogHandler()] logger.warning("I am a test case!") @@ -18,9 +19,9 @@ class DatabaseLogHandlerTests(TestCase): [entry] = LogEntry.objects.all() self.assertEqual(entry.application, 'site') - self.assertEqual(entry.logger_name, "test_logger") + self.assertEqual(entry.logger_name, module_basename) self.assertIsInstance(entry.timestamp, datetime) self.assertEqual(entry.level, 'warning') - self.assertEqual(entry.module, __name__) + self.assertEqual(entry.module, module_basename) self.assertIsInstance(entry.line, int) self.assertEqual(entry.message, "I am a test case!") |