fix: adjusted course query on enrollments extraction

It must to be able to query courses with an entire course_id
or just a piece of it.

Author: Tiago-Salles

nau_openedx_extensions/partner_integration/facade.py

@@ -353,7 +353,11 @@ def _execute_enrollments_query(self, enrollments_base_query, start_dt, end_dt, c
             enrollments_query = enrollments_base_query
 
             if courses:
-                enrollments_query = enrollments_query.filter(course__id__in=courses).select_related("user")
+                q = Q()
+                for code in courses:
+                    q |= Q(course__id__icontains=code)
+
+                enrollments_query = enrollments_query.filter(q).select_related("user")
 
             if not start_dt or not end_dt:
                 start = datetime.now()

nau_openedx_extensions/partner_integration/tests/test_api.py

@@ -771,6 +771,32 @@ def test_multiple_courses_returns_combined_results(self):
         returned_courses = [r["course_id"] for r in response.data["results"]]
         self.assertIn(str(courses[0].id), returned_courses)
         self.assertIn(str(courses[1].id), returned_courses)
+        
+    def test_multiple_courses_with_only_codes(self):
+        """
+        Test that multiple course filters return combined results.
+        1. Authenticates a partner client.
+        2. Calls the data extractor endpoint with multiple course codes.
+        3. Validates the response contains enrollments for all specified courses.
+        """
+        partner_client = self.base_data["partner_clients"][0]
+        access_token = self.authenticate_partner_client(partner_client)
+
+        courses = self.base_data["courses"][:2]
+
+        self.http_client.credentials(**{"HTTP_AUTHORIZATION": f"Bearer {access_token}"})
+        response = self.http_client.post(self.endpoint,
+                                         data={"courses": [
+                                            str(course.id).split("+")[1]
+                                            for course in courses]},
+                                         format="json",
+                                         )
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        returned_courses = [r["course_id"] for r in response.data["results"]]
+        self.assertIn(str(courses[0].id), returned_courses)
+        self.assertIn(str(courses[1].id), returned_courses)
 
     def test_date_filter_limits_results(self):
         """