Unit test for getIncomingHostByteCounts

[affinity.git] / analytics / implementation / src / main / java / org / opendaylight / affinity / analytics / internal / AnalyticsManager.java

diff --git a/analytics/implementation/src/main/java/org/opendaylight/affinity/analytics/internal/AnalyticsManager.java b/analytics/implementation/src/main/java/org/opendaylight/affinity/analytics/internal/AnalyticsManager.java
index 075d6ef925538b0029af65c7129f190557abdf3d..8d0669270494006062e97721f51cbd6c02806327 100644 (file)
--- a/analytics/implementation/src/main/java/org/opendaylight/affinity/analytics/internal/AnalyticsManager.java
+++ b/analytics/implementation/src/main/java/org/opendaylight/affinity/analytics/internal/AnalyticsManager.java
@@ -384,11 +384,16 @@ public class AnalyticsManager implements IReadServiceListener, IAnalyticsManager
         return getIncomingHostByteCounts(subnet, null);
     }
 
-    /* Returns all hosts that transferred data into this subnet. */
     public Map<Host, Long> getIncomingHostByteCounts(String subnet, Byte protocol) {
+         Set<HostNodeConnector> allHosts = this.hostTracker.getAllHosts();
+         return getIncomingHostByteCounts(subnet, protocol, allHosts);
+    }
+
+    /* Returns all hosts that transferred data into this subnet. */
+    public Map<Host, Long> getIncomingHostByteCounts(String subnet, Byte protocol, Set<HostNodeConnector> allHosts) {
         Map<Host, Long> hosts = new HashMap<Host, Long>();
-        Set<Host> dstHosts = getHostsInSubnet(subnet);
-        Set<Host> otherHosts = getHostsNotInSubnet(subnet);
+        Set<Host> dstHosts = getHostsInSubnet(subnet, allHosts);
+        Set<Host> otherHosts = getHostsNotInSubnet(subnet, allHosts);
         for (Host host : otherHosts) {
             for (Host targetHost : dstHosts) {
                 Long byteCount = getByteCount(host, targetHost, protocol);