From 9379e95446869fb01718ef21a1e2f26a9b730b6b Mon Sep 17 00:00:00 2001 From: Daniel Quinn Date: Sun, 28 Feb 2016 00:53:18 +0000 Subject: [PATCH] Added a test for the new by_group() feature --- src/documents/tests/test_logger.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/src/documents/tests/test_logger.py b/src/documents/tests/test_logger.py index d5527d7c6..8d31fe4ec 100644 --- a/src/documents/tests/test_logger.py +++ b/src/documents/tests/test_logger.py @@ -122,3 +122,21 @@ class TestPaperlessLog(TestCase): self.logger.critical("This is a critical message", extra=kw1) self.assertEqual(Log.objects.all().count(), 4) self.assertEqual(Log.objects.filter(component=c1).count(), 2) + + def test_groupped_query(self): + + kw = { + "group": uuid.uuid4(), + "component": Log.COMPONENT_MAIL + } + with mock.patch("logging.StreamHandler.emit") as __: + self.logger.info("Message 0", extra=kw) + self.logger.info("Message 1", extra=kw) + self.logger.info("Message 2", extra=kw) + self.logger.info("Message 3", extra=kw) + + self.assertEqual(Log.objects.all().by_group().count(), 1) + self.assertEqual( + Log.objects.all().by_group()[0]["Messages"], + "Message 0\nMessage 1\nMessage 2\nMessage 3" + )