From cfd53a92b0a42f227b92ff1babc2b2a0453e6f56 Mon Sep 17 00:00:00 2001
From: lincmba <lincolncmba@gmail.com>
Date: Tue, 30 Jul 2024 11:00:23 +0300
Subject: [PATCH] Add test

---
 .../LocationHierarchyEndpointHelperTest.java       | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/plugins/src/test/java/org/smartregister/fhir/gateway/plugins/LocationHierarchyEndpointHelperTest.java b/plugins/src/test/java/org/smartregister/fhir/gateway/plugins/LocationHierarchyEndpointHelperTest.java
index 95c22be..184d8d0 100644
--- a/plugins/src/test/java/org/smartregister/fhir/gateway/plugins/LocationHierarchyEndpointHelperTest.java
+++ b/plugins/src/test/java/org/smartregister/fhir/gateway/plugins/LocationHierarchyEndpointHelperTest.java
@@ -285,6 +285,20 @@ public void testGetDescendantsWithAdminLevelFiltersReturnsLocationsWithinAdminLe
         verify(queryMock, times(2)).execute();
     }
 
+    @Test
+    public void testFilterLocationsByAdminLevelsBasic() {
+        List<Location> locations = createLocationList(5, true);
+        List<String> adminLevels = List.of("1", "3");
+
+        List<Location> filteredLocations =
+                locationHierarchyEndpointHelper.filterLocationsByAdminLevels(
+                        locations, adminLevels);
+
+        Assert.assertEquals(2, filteredLocations.size());
+        Assert.assertEquals("1", filteredLocations.get(0).getId());
+        Assert.assertEquals("3", filteredLocations.get(1).getId());
+    }
+
     private Bundle getLocationBundle() {
         Bundle bundleLocation = new Bundle();
         bundleLocation.setId("Location/1234");