NDH-676 clean up test assertion initial

Signed-off-by: Isaac Milarsky <imilarsky@gmail.com>

Author: IsaacMilarky

backend/npdfhir/tests/test_endpoint.py

@@ -90,15 +90,15 @@ def test_list_entries_are_fhir_endpoints(self):
         bundle = response.data["results"]
         self.assertGreater(len(bundle["entry"]), 0)
 
-        first_entry = bundle["entry"][0]
-        self.assertIn("resource", first_entry)
+        for entry in bundle["entry"]:
+            self.assertIn("resource", entry)
 
-        endpoint_resource = first_entry["resource"]
-        self.assertEqual(endpoint_resource["resourceType"], "Endpoint")
-        self.assertIn("id", endpoint_resource)
-        self.assertIn("status", endpoint_resource)
-        self.assertIn("connectionType", endpoint_resource)
-        self.assertIn("address", endpoint_resource)
+            endpoint_resource = entry["resource"]
+            self.assertEqual(endpoint_resource["resourceType"], "Endpoint")
+            self.assertIn("id", endpoint_resource)
+            self.assertIn("status", endpoint_resource)
+            self.assertIn("connectionType", endpoint_resource)
+            self.assertIn("address", endpoint_resource)
 
     # Pagination tests
     def test_pagination_custom_page_size(self):
@@ -124,10 +124,11 @@ def test_filter_by_name(self):
 
         self.assertGreater(len(bundle["entry"]), 0)
 
-        first_endpoint = bundle["entry"][0]["resource"]
+        for entry in bundle["entry"]:
+            endpoint = entry["resource"]
 
-        self.assertIn("name", first_endpoint)
-        self.assertIn("Kansas City", first_endpoint["name"])
+            self.assertIn("name", endpoint)
+            self.assertIn("Kansas City Psychiatric Group", endpoint["name"])
 
     def test_filter_by_connection_type(self):
         connection_type = "hl7-fhir-rest"
@@ -139,11 +140,12 @@ def test_filter_by_connection_type(self):
         entries = bundle.get("entry", [])
         self.assertGreater(len(entries), 0)
 
-        first_endpoint = entries[0]["resource"]
-        self.assertIn("connectionType", first_endpoint)
+        for entry in bundle["entry"]:
+            endpoint = entry["resource"]
+            self.assertIn("connectionType", endpoint)
 
-        code = first_endpoint["connectionType"]["code"]
-        self.assertEqual(connection_type, code)
+            code = endpoint["connectionType"]["code"]
+            self.assertEqual(connection_type, code)
 
     def test_filter_by_payload_type(self):
         payload_type = "ccda-structuredBody:1.1"
@@ -155,11 +157,12 @@ def test_filter_by_payload_type(self):
         entries = bundle.get("entry", [])
         self.assertGreater(len(entries), 0)
 
-        first_endpoint = entries[0]["resource"]
-        self.assertIn("payloadType", first_endpoint)
+        for entry in bundle["entry"]:
+            endpoint = entry["resource"]
+            self.assertIn("payloadType", endpoint)
 
-        code = first_endpoint["payloadType"][0]["coding"][0]["display"]
-        self.assertEqual(payload_type, code)
+            code = endpoint["payloadType"][0]["coding"][0]["display"]
+            self.assertEqual(payload_type, code)
 
     def test_filter_returns_empty_for_nonexistent_name(self):
         response = self.client.get(self.list_url, {"name": "NonexistentEndpointName12345"})
@@ -171,10 +174,7 @@ def test_filter_returns_empty_for_nonexistent_name(self):
 
     # Retrieve tests
     def test_retrieve_specific_endpoint(self):
-        list_response = self.client.get(self.list_url, {"page_size": 1})
-        first_endpoint = list_response.data["results"]["entry"][0]["resource"]
-
-        endpoint_id = first_endpoint["id"]
+        endpoint_id = str(self.endpoints[0].endpoint_instance.id)
         detail_url = reverse("fhir-endpoint-detail", args=[endpoint_id])
 
         response = self.client.get(detail_url)
@@ -194,9 +194,3 @@ def test_retrieve_nonexistent_endpoint(self):
 
         self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND)
 
-    def test_retrieve_single_endpoint(self):
-        id = self.endpoints[0].endpoint_instance.id
-        url = reverse("fhir-endpoint-detail", args=[id])
-        response = self.client.get(url)
-        self.assertEqual(response.status_code, status.HTTP_200_OK)
-        self.assertEqual(response.data["id"], str(id))