diff --git a/cl/alerts/tests.py b/cl/alerts/tests.py index c3aed63227..65867461f2 100644 --- a/cl/alerts/tests.py +++ b/cl/alerts/tests.py @@ -1840,6 +1840,11 @@ def setUpTestData(cls): jurisdiction="FB", citation_string="Bankr. C.D. Cal.", ) + cls.docket = DocketFactory( + court=cls.court_1, + date_argued=now().date(), + docket_number="20-5736", + ) cls.user_profile = UserProfileWithParentsFactory() NeonMembership.objects.create( level=NeonMembership.LEGACY, user=cls.user_profile.user @@ -1890,6 +1895,12 @@ def setUpTestData(cls): name="Test Alert OA Monthly", query="q=Test+OA&type=oa", ) + cls.search_alert_7 = AlertFactory( + user=cls.user_profile.user, + rate=Alert.REAL_TIME, + name="Test Alert OA RT Docket ID", + query=f"q=docket_id:{cls.docket.pk}&type=oa", + ) @classmethod def tearDownClass(cls): @@ -2046,6 +2057,18 @@ def test_send_oa_search_alert_webhooks(self, mock_abort_audio): content["payload"]["results"][0]["caseName"], rt_oral_argument.case_name, ) + self.assertEqual( + content["payload"]["results"][0]["docketNumber"], + rt_oral_argument.docket.docket_number, + ) + self.assertEqual( + content["payload"]["results"][0]["snippet"], + rt_oral_argument.transcript, + ) + self.assertEqual( + content["payload"]["results"][0]["judge"], + rt_oral_argument.judges, + ) self.assertEqual( content["payload"]["results"][0]["court"], rt_oral_argument.docket.court.full_name, @@ -2054,8 +2077,74 @@ def test_send_oa_search_alert_webhooks(self, mock_abort_audio): content["payload"]["results"][0]["source"], rt_oral_argument.source, ) + + # Confirm no HL fields are properly displayed. + with mock.patch( + "cl.api.webhooks.requests.post", + side_effect=lambda *args, **kwargs: MockResponse( + 200, mock_raw=True + ), + ): + mock_date = now().replace(day=1, hour=5) + with time_machine.travel( + mock_date, tick=False + ), self.captureOnCommitCallbacks(execute=True): + # When the Audio object is created it should trigger an alert. + transcript_response = { + "response": { + "results": [ + { + "alternatives": [ + { + "transcript": "This a different transcript.", + "confidence": 0.85, + }, + ] + }, + ] + } + } + json_transcript = json.dumps(transcript_response) + rt_oral_argument_2 = AudioWithParentsFactory.create( + case_name="No HL OA Alert", + docket=self.docket, + stt_status=Audio.STT_COMPLETE, + judges="George Smith", + stt_google_response=json_transcript, + ) + + self.assertEqual(len(mail.outbox), 3, msg="Wrong number of emails.") + text_content = mail.outbox[2].body + + # Confirm all fields are displayed in the plain text version. + self.assertIn(rt_oral_argument_2.case_name, text_content) + self.assertIn(rt_oral_argument_2.judges, text_content) + self.assertIn(rt_oral_argument_2.docket.docket_number, text_content) + self.assertIn(rt_oral_argument_2.transcript, text_content) + self.assertIn( + rt_oral_argument_2.docket.court.citation_string, text_content + ) + + # Extract HTML version. + html_content = None + for content, content_type in mail.outbox[2].alternatives: + if content_type == "text/html": + html_content = content + break + + # Confirm all fields are displayed in the HTML version. + self.assertIn(rt_oral_argument_2.case_name, html_content) + self.assertIn(rt_oral_argument_2.judges, html_content) + self.assertIn(rt_oral_argument_2.docket.docket_number, html_content) + self.assertIn(rt_oral_argument_2.transcript, html_content) + self.assertIn( + rt_oral_argument_2.docket.court.citation_string, + html_content.replace(" ", " "), + ) + webhook_events.delete() rt_oral_argument.delete() + rt_oral_argument_2.delete() def test_send_alert_on_document_creation(self, mock_abort_audio): """Avoid sending Search Alerts on document updates.""" @@ -2498,7 +2587,7 @@ def test_send_multiple_rt_alerts(self, mock_abort_audio): self.assertEqual(len(memberships), 11) total_rt_alerts = Alert.objects.filter(rate=Alert.REAL_TIME) # 2 created in setUpTestData + 10 - self.assertEqual(total_rt_alerts.count(), 12) + self.assertEqual(total_rt_alerts.count(), 13) # Clear the outbox mail.outbox = [] diff --git a/cl/custom_filters/templatetags/extras.py b/cl/custom_filters/templatetags/extras.py index 9f29ad4b51..40d2813cda 100644 --- a/cl/custom_filters/templatetags/extras.py +++ b/cl/custom_filters/templatetags/extras.py @@ -235,13 +235,11 @@ def get_highlight(result: AttrDict | dict[str, any], field: str) -> any: """ hl_value = None + original_value = getattr(result, field, "") if isinstance(result, AttrDict) and hasattr(result.meta, "highlight"): hl_value = getattr(result.meta.highlight, field, None) elif isinstance(result, dict): hl_value = result.get("meta", {}).get("highlight", {}).get(field) + original_value = result.get(field, "") - return ( - render_string_or_list(hl_value) - if hl_value - else getattr(result, field, "") - ) + return render_string_or_list(hl_value) if hl_value else original_value