aaq urgency check endpoint V2

aaq/serializers.py

@@ -53,3 +53,7 @@ class ContentFeedbackSerializer(serializers.Serializer):
     feedback_text = serializers.CharField(required=False)
     query_id = serializers.IntegerField(required=True)
     content_id = serializers.IntegerField(required=True)
+
+
+class UrgencyCheckV2Serializer(serializers.Serializer):
+    message_text = serializers.CharField(required=True)

aaq/tests/helpers.py

@@ -118,3 +118,32 @@ def post_search_return_empty(self, request):
         }
 
         return (200, {}, json.dumps(resp_body))
+
+
+class FakeAaqUdV2Api:
+    def post_urgency_detect_return_true(self, request):
+        resp_body = {
+            "details": {
+                "0": {"distance": 0.1, "urgency_rule": "Blurry vision and dizziness"},
+                "1": {"distance": 0.2, "urgency_rule": "Nausea that lasts for 3 days"},
+            },
+            "is_urgent": True,
+            "matched_rules": [
+                "Blurry vision and dizziness",
+                "Nausea that lasts for 3 days",
+            ],
+        }
+
+        return (200, {}, json.dumps(resp_body))
+
+    def post_urgency_detect_return_false(self, request):
+        resp_body = {
+            "details": {
+                "0": {"distance": 0.1, "urgency_rule": "Baby okay"},
+                "1": {"distance": 0.2, "urgency_rule": "Baby healthy"},
+            },
+            "is_urgent": False,
+            "matched_rules": ["Baby okay", "Baby healthy"],
+        }
+
+        return (200, {}, json.dumps(resp_body))

aaq/tests/test_views.py

@@ -6,7 +6,7 @@
 from rest_framework import status
 from rest_framework.test import APITestCase
 
-from .helpers import FakeAaqApi, FakeAaqCoreApi, FakeAaqUdApi, FakeTask
+from .helpers import FakeAaqApi, FakeAaqCoreApi, FakeAaqUdApi, FakeAaqUdV2Api, FakeTask
 
 
 class GetFirstPageViewTests(APITestCase):
@@ -527,3 +527,88 @@ def test_response_feedback_invalid_feedback_text_view(self):
         assert response.json() == {
             "feedback_sentiment": ['"test" is not a valid choice.']
         }
+
+
+class CheckUrgencyV2ViewTests(APITestCase):
+    url = reverse("aaq-check-urgency-v2")
+
+    @responses.activate
+    def test_urgency_check_urgent(self):
+        """
+        Test that we can get an urgency score of 1.0
+        """
+        user = get_user_model().objects.create_user("test")
+        self.client.force_authenticate(user)
+        fakeAaqUdV2Api = FakeAaqUdV2Api()
+        responses.add_callback(
+            responses.POST,
+            "http://aaq_v2/urgency-check-v2",
+            callback=fakeAaqUdV2Api.post_urgency_detect_return_true,
+            content_type="application/json",
+        )
+
+        payload = json.dumps({"message_text": "I am pregnant and out of breath"})
+
+        response = self.client.post(
+            self.url, data=payload, content_type="application/json"
+        )
+
+        assert response.status_code == 200
+        assert response.json() == {
+            "details": {
+                "0": {"distance": 0.1, "urgency_rule": "Blurry vision and dizziness"},
+                "1": {"distance": 0.2, "urgency_rule": "Nausea that lasts for 3 days"},
+            },
+            "is_urgent": True,
+            "matched_rules": [
+                "Blurry vision and dizziness",
+                "Nausea that lasts for 3 days",
+            ],
+        }
+
+    @responses.activate
+    def test_urgency_check_not_urgent(self):
+        """
+        Test that we can get an urgency score of 0.0
+        """
+        user = get_user_model().objects.create_user("test")
+        self.client.force_authenticate(user)
+        fakeAaqUdV2Api = FakeAaqUdV2Api()
+        responses.add_callback(
+            responses.POST,
+            "http://aaq_v2/urgency-check-v2",
+            callback=fakeAaqUdV2Api.post_urgency_detect_return_false,
+            content_type="application/json",
+        )
+
+        payload = json.dumps({"message_text": "I am fine"})
+
+        response = self.client.post(
+            self.url, data=payload, content_type="application/json"
+        )
+
+        assert response.status_code == 200
+        assert response.json() == {
+            "details": {
+                "0": {"distance": 0.1, "urgency_rule": "Baby okay"},
+                "1": {"distance": 0.2, "urgency_rule": "Baby healthy"},
+            },
+            "is_urgent": False,
+            "matched_rules": ["Baby okay", "Baby healthy"],
+        }
+
+    @responses.activate
+    def test_urgency_check_invalid(self):
+        """
+        Test that we can get a field required message
+        """
+        user = get_user_model().objects.create_user("test")
+        self.client.force_authenticate(user)
+        payload = json.dumps({})
+
+        response = self.client.post(
+            self.url, data=payload, content_type="application/json"
+        )
+
+        assert response.status_code == 400
+        assert response.json() == {"message_text": ["This field is required."]}

aaq/urls.py

@@ -38,4 +38,9 @@
         views.search,
         name="aaq-search",
     ),
+    re_path(
+        r"^api/v2/check-urgency",
+        views.check_urgency_v2,
+        name="aaq-check-urgency-v2",
+    ),
 ]

aaq/views.py

@@ -15,6 +15,7 @@
     ResponseFeedbackSerializer,
     SearchSerializer,
     UrgencyCheckSerializer,
+    UrgencyCheckV2Serializer,
 )
 
 from .tasks import send_feedback_task, send_feedback_task_v2
@@ -222,3 +223,20 @@ def search(request, *args, **kwargs):
     }
 
     return Response(json_msg, status=status.HTTP_200_OK)
+
+
+@api_view(("POST",))
+@renderer_classes((JSONRenderer,))
+def check_urgency_v2(request, *args, **kwargs):
+    serializer = UrgencyCheckV2Serializer(data=request.data)
+    serializer.is_valid(raise_exception=True)
+    message_text = serializer.validated_data["message_text"]
+    url = urllib.parse.urljoin(settings.AAQ_V2_API_URL, "/urgency-check-v2")
+    headers = {
+        "Authorization": settings.AAQ_V2_API_URL,
+        "Content-Type": "application/json",
+    }
+
+    response = requests.request("POST", url, json=message_text, headers=headers)
+
+    return Response(response.json(), status=status.HTTP_200_OK)