diff --git a/bramble-core/src/main/java/org/briarproject/bramble/plugin/PollerImpl.java b/bramble-core/src/main/java/org/briarproject/bramble/plugin/PollerImpl.java index c42e49a8c..a0b72dc55 100644 --- a/bramble-core/src/main/java/org/briarproject/bramble/plugin/PollerImpl.java +++ b/bramble-core/src/main/java/org/briarproject/bramble/plugin/PollerImpl.java @@ -215,7 +215,7 @@ class PollerImpl implements Poller, EventListener { Map remote = transportPropertyManager.getRemoteProperties(t); Collection connected = - connectionRegistry.getConnectedContacts(t); + connectionRegistry.getConnectedOrBetterContacts(t); Collection> properties = new ArrayList<>(); for (Entry e : remote.entrySet()) { diff --git a/bramble-core/src/test/java/org/briarproject/bramble/plugin/PollerImplTest.java b/bramble-core/src/test/java/org/briarproject/bramble/plugin/PollerImplTest.java index 522c9aad9..9d56b2ef4 100644 --- a/bramble-core/src/test/java/org/briarproject/bramble/plugin/PollerImplTest.java +++ b/bramble-core/src/test/java/org/briarproject/bramble/plugin/PollerImplTest.java @@ -354,7 +354,7 @@ public class PollerImplTest extends BrambleMockTestCase { // Get the transport properties and connected contacts oneOf(transportPropertyManager).getRemoteProperties(transportId); will(returnValue(singletonMap(contactId, properties))); - oneOf(connectionRegistry).getConnectedContacts(transportId); + oneOf(connectionRegistry).getConnectedOrBetterContacts(transportId); will(returnValue(emptyList())); // Poll the plugin oneOf(plugin).poll(with(collectionOf( @@ -397,7 +397,7 @@ public class PollerImplTest extends BrambleMockTestCase { // Get the transport properties and connected contacts oneOf(transportPropertyManager).getRemoteProperties(transportId); will(returnValue(singletonMap(contactId, properties))); - oneOf(connectionRegistry).getConnectedContacts(transportId); + oneOf(connectionRegistry).getConnectedOrBetterContacts(transportId); will(returnValue(singletonList(contactId))); // All contacts are connected, so don't poll the plugin }});