Avoid hardcoding module name.

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!")