Add test

onaio · Jul 30, 2024 · cfd53a9 · cfd53a9
1 parent 6fe6d4d
commit cfd53a9
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/...test/java/org/smartregister/fhir/gateway/plugins/LocationHierarchyEndpointHelperTest.java b/...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");