ATLAS-5147: Relationship Search API approximateCount returns pageLimit value instead of total relationship count#2

Author: aditya-gupta36

graphdb/janus/src/main/java/org/apache/atlas/repository/graphdb/janus/AtlasJanusIndexQuery.java

@@ -114,7 +114,7 @@ public Iterator<Result<AtlasJanusVertex, AtlasJanusEdge>> edges(int offset, int
 
     @Override
     public Long edgeTotals() {
-        return query.edgeTotals();
+        return query.offset(0).limit(Integer.MAX_VALUE).edgeTotals();
     }
 
     /**

repository/src/test/java/org/apache/atlas/discovery/RelationshipSearchProcessorTest.java

@@ -29,6 +29,7 @@
 import org.apache.commons.collections.CollectionUtils;
 import org.testng.annotations.AfterClass;
 import org.testng.annotations.BeforeClass;
+import org.testng.annotations.DataProvider;
 import org.testng.annotations.Guice;
 import org.testng.annotations.Test;
 
@@ -65,6 +66,23 @@ public void totalRelationships() throws AtlasBaseException {
         executeAndAssert(params, 17);
     }
 
+    @DataProvider(name = "limitProvider")
+    public Integer[] limitProvider() {
+        return new Integer[] {15, 0, 1};
+    }
+
+    @Test(dataProvider = "limitProvider")
+    public void testRelationshipsApproxCount_WithVariousLimits(int limit) throws AtlasBaseException {
+        SearchParameters params = new SearchParameters();
+        params.setRelationshipName("user_post");
+        params.setLimit(limit);
+
+        SearchContext context = new SearchContext(params, typeRegistry, graph, Collections.emptySet());
+        RelationshipSearchProcessor processor = new RelationshipSearchProcessor(context, indexer.getEdgeIndexKeys());
+
+        assertEquals(processor.getResultCount(), 17);
+    }
+
     @Test
     public void searchByAttribute() throws AtlasBaseException {
         SearchParameters params = new SearchParameters();