Move GossipDeviceStore away from deprecated ClusterCommunicationService API

Change-Id: Ib0ca7125e17013156aac27f8437ca717a96a56f0

Move GossipDeviceStore away from deprecated ClusterCommunicationService API
Change-Id: Ib0ca7125e17013156aac27f8437ca717a96a56f0
Jonathan Hart · Gerrit Code Review
Commit 46ab5cc4bad9602485428da5675b7205d0b83c27 46ab5cc4 1 parent 714dd2bf
Showing 2 changed files with 160 additions and 227 deletions
core/store/dist/src/main/java/org/onosproject/store/device/impl/GossipDeviceStore.java
core/store/dist/src/test/java/org/onosproject/store/device/impl/GossipDeviceStoreTest.java
--- a/core/store/dist/src/main/java/org/onosproject/store/device/impl/GossipDeviceStore.java
View file @46ab5cc
+++ b/core/store/dist/src/main/java/org/onosproject/store/device/impl/GossipDeviceStore.java
View file @46ab5cc
@@ -15,26 +15,10 @@
  */
 package org.onosproject.store.device.impl;
-import java.io.IOException;
+import com.google.common.collect.FluentIterable;
-import java.util.ArrayList;
+import com.google.common.collect.ImmutableList;
-import java.util.Collection;
+import com.google.common.collect.Maps;
-import java.util.Collections;
+import com.google.common.collect.Sets;
-import java.util.HashMap;
-import java.util.HashSet;
-import java.util.Iterator;
-import java.util.List;
-import java.util.Map;
-import java.util.Map.Entry;
-import java.util.Objects;
-import java.util.Optional;
-import java.util.Set;
-import java.util.concurrent.ConcurrentHashMap;
-import java.util.concurrent.ConcurrentMap;
-import java.util.concurrent.ExecutorService;
-import java.util.concurrent.ScheduledExecutorService;
-import java.util.concurrent.TimeUnit;
-import java.util.stream.Stream;
-
 import org.apache.commons.lang3.RandomUtils;
 import org.apache.felix.scr.annotations.Activate;
 import org.apache.felix.scr.annotations.Component;
@@ -81,8 +65,6 @@ import org.onosproject.net.provider.ProviderId;
 import org.onosproject.store.AbstractStore;
 import org.onosproject.store.Timestamp;
 import org.onosproject.store.cluster.messaging.ClusterCommunicationService;
-import org.onosproject.store.cluster.messaging.ClusterMessage;
-import org.onosproject.store.cluster.messaging.ClusterMessageHandler;
 import org.onosproject.store.cluster.messaging.MessageSubject;
 import org.onosproject.store.impl.Timestamped;
 import org.onosproject.store.serializers.KryoNamespaces;
@@ -96,10 +78,26 @@ import org.onosproject.store.service.StorageService;
 import org.onosproject.store.service.WallClockTimestamp;
 import org.slf4j.Logger;
-import com.google.common.collect.FluentIterable;
+import java.io.IOException;
-import com.google.common.collect.ImmutableList;
+import java.util.ArrayList;
-import com.google.common.collect.Maps;
+import java.util.Collection;
-import com.google.common.collect.Sets;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Objects;
+import java.util.Optional;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ConcurrentMap;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+import java.util.function.Consumer;
+import java.util.stream.Stream;
 import static com.google.common.base.Preconditions.checkArgument;
 import static com.google.common.base.Predicates.notNull;
@@ -117,8 +115,13 @@ import static org.onosproject.net.device.DeviceEvent.Type.PORT_STATS_UPDATED;
 import static org.onosproject.net.device.DeviceEvent.Type.PORT_UPDATED;
 import static org.onosproject.store.device.impl.GossipDeviceStoreMessageSubjects.DEVICE_ADVERTISE;
 import static org.onosproject.store.device.impl.GossipDeviceStoreMessageSubjects.DEVICE_INJECTED;
+import static org.onosproject.store.device.impl.GossipDeviceStoreMessageSubjects.DEVICE_OFFLINE;
+import static org.onosproject.store.device.impl.GossipDeviceStoreMessageSubjects.DEVICE_REMOVED;
 import static org.onosproject.store.device.impl.GossipDeviceStoreMessageSubjects.DEVICE_REMOVE_REQ;
+import static org.onosproject.store.device.impl.GossipDeviceStoreMessageSubjects.DEVICE_UPDATE;
 import static org.onosproject.store.device.impl.GossipDeviceStoreMessageSubjects.PORT_INJECTED;
+import static org.onosproject.store.device.impl.GossipDeviceStoreMessageSubjects.PORT_STATUS_UPDATE;
+import static org.onosproject.store.device.impl.GossipDeviceStoreMessageSubjects.PORT_UPDATE;
 import static org.onosproject.store.service.EventuallyConsistentMapEvent.Type.PUT;
 import static org.slf4j.LoggerFactory.getLogger;
@@ -208,29 +211,15 @@ public class GossipDeviceStore
         backgroundExecutor =
                 newSingleThreadScheduledExecutor(minPriority(groupedThreads("onos/device", "bg-%d", log)));
-        clusterCommunicator.addSubscriber(
+        addSubscriber(DEVICE_UPDATE, this::handleDeviceEvent);
-                GossipDeviceStoreMessageSubjects.DEVICE_UPDATE, new InternalDeviceEventListener(), executor);
+        addSubscriber(DEVICE_OFFLINE, this::handleDeviceOfflineEvent);
-        clusterCommunicator.addSubscriber(
+        addSubscriber(DEVICE_REMOVE_REQ, this::handleRemoveRequest);
-                GossipDeviceStoreMessageSubjects.DEVICE_OFFLINE,
+        addSubscriber(DEVICE_REMOVED, this::handleDeviceRemovedEvent);
-                new InternalDeviceOfflineEventListener(),
+        addSubscriber(PORT_UPDATE, this::handlePortEvent);
-                executor);
+        addSubscriber(PORT_STATUS_UPDATE, this::handlePortStatusEvent);
-        clusterCommunicator.addSubscriber(DEVICE_REMOVE_REQ,
+        addSubscriber(DEVICE_ADVERTISE, this::handleDeviceAdvertisement);
-                                          new InternalRemoveRequestListener(),
+        addSubscriber(DEVICE_INJECTED, this::handleDeviceInjectedEvent);
-                                          executor);
+        addSubscriber(PORT_INJECTED, this::handlePortInjectedEvent);
-        clusterCommunicator.addSubscriber(
-                GossipDeviceStoreMessageSubjects.DEVICE_REMOVED, new InternalDeviceRemovedEventListener(), executor);
-        clusterCommunicator.addSubscriber(
-                GossipDeviceStoreMessageSubjects.PORT_UPDATE, new InternalPortEventListener(), executor);
-        clusterCommunicator.addSubscriber(
-                GossipDeviceStoreMessageSubjects.PORT_STATUS_UPDATE, new InternalPortStatusEventListener(), executor);
-        clusterCommunicator.addSubscriber(
-                GossipDeviceStoreMessageSubjects.DEVICE_ADVERTISE,
-                new InternalDeviceAdvertisementListener(),
-                backgroundExecutor);
-        clusterCommunicator.addSubscriber(
-                GossipDeviceStoreMessageSubjects.DEVICE_INJECTED, new DeviceInjectedEventListener(), executor);
-        clusterCommunicator.addSubscriber(
-                GossipDeviceStoreMessageSubjects.PORT_INJECTED, new PortInjectedEventListener(), executor);
         // start anti-entropy thread
         backgroundExecutor.scheduleAtFixedRate(new SendAdvertisementTask(),
@@ -261,6 +250,10 @@ public class GossipDeviceStore
         log.info("Started");
     }
+    private <M> void addSubscriber(MessageSubject subject, Consumer<M> handler) {
+        clusterCommunicator.addSubscriber(subject, SERIALIZER::decode, handler, executor);
+    }
+
     @Deactivate
     public void deactivate() {
         devicePortStats.removeListener(portStatsListener);
@@ -281,24 +274,15 @@ public class GossipDeviceStore
         devices.clear();
         devicePorts.clear();
         availableDevices.clear();
-        clusterCommunicator.removeSubscriber(
+        clusterCommunicator.removeSubscriber(DEVICE_UPDATE);
-                GossipDeviceStoreMessageSubjects.DEVICE_UPDATE);
+        clusterCommunicator.removeSubscriber(DEVICE_OFFLINE);
-        clusterCommunicator.removeSubscriber(
+        clusterCommunicator.removeSubscriber(DEVICE_REMOVE_REQ);
-                GossipDeviceStoreMessageSubjects.DEVICE_OFFLINE);
+        clusterCommunicator.removeSubscriber(DEVICE_REMOVED);
-        clusterCommunicator.removeSubscriber(
+        clusterCommunicator.removeSubscriber(PORT_UPDATE);
-                GossipDeviceStoreMessageSubjects.DEVICE_REMOVE_REQ);
+        clusterCommunicator.removeSubscriber(PORT_STATUS_UPDATE);
-        clusterCommunicator.removeSubscriber(
+        clusterCommunicator.removeSubscriber(DEVICE_ADVERTISE);
-                GossipDeviceStoreMessageSubjects.DEVICE_REMOVED);
+        clusterCommunicator.removeSubscriber(DEVICE_INJECTED);
-        clusterCommunicator.removeSubscriber(
+        clusterCommunicator.removeSubscriber(PORT_INJECTED);
-                GossipDeviceStoreMessageSubjects.PORT_UPDATE);
-        clusterCommunicator.removeSubscriber(
-                GossipDeviceStoreMessageSubjects.PORT_STATUS_UPDATE);
-        clusterCommunicator.removeSubscriber(
-                GossipDeviceStoreMessageSubjects.DEVICE_ADVERTISE);
-        clusterCommunicator.removeSubscriber(
-                GossipDeviceStoreMessageSubjects.DEVICE_INJECTED);
-        clusterCommunicator.removeSubscriber(
-                GossipDeviceStoreMessageSubjects.PORT_INJECTED);
         log.info("Stopped");
     }
@@ -1336,7 +1320,7 @@ public class GossipDeviceStore
     }
     private void notifyPeers(InternalDeviceEvent event) {
-        broadcastMessage(GossipDeviceStoreMessageSubjects.DEVICE_UPDATE, event);
+        broadcastMessage(DEVICE_UPDATE, event);
     }
     private void notifyPeers(InternalDeviceOfflineEvent event) {
@@ -1357,7 +1341,7 @@ public class GossipDeviceStore
     private void notifyPeer(NodeId recipient, InternalDeviceEvent event) {
         try {
-            unicastMessage(recipient, GossipDeviceStoreMessageSubjects.DEVICE_UPDATE, event);
+            unicastMessage(recipient, DEVICE_UPDATE, event);
         } catch (IOException e) {
             log.error("Failed to send" + event + " to " + recipient, e);
         }
@@ -1631,188 +1615,124 @@ public class GossipDeviceStore
         }
     }
-    private final class InternalDeviceEventListener
+    private void handleDeviceEvent(InternalDeviceEvent event) {
-            implements ClusterMessageHandler {
+        ProviderId providerId = event.providerId();
-        @Override
+        DeviceId deviceId = event.deviceId();
-        public void handle(ClusterMessage message) {
+        Timestamped<DeviceDescription> deviceDescription = event.deviceDescription();
-            log.debug("Received device update event from peer: {}", message.sender());
-            InternalDeviceEvent event = SERIALIZER.decode(message.payload());
-
-            ProviderId providerId = event.providerId();
-            DeviceId deviceId = event.deviceId();
-            Timestamped<DeviceDescription> deviceDescription = event.deviceDescription();
-            try {
+        try {
-                notifyDelegateIfNotNull(createOrUpdateDeviceInternal(providerId, deviceId,
+            notifyDelegateIfNotNull(createOrUpdateDeviceInternal(providerId, deviceId,
-                                                                     deviceDescription));
+                    deviceDescription));
-            } catch (Exception e) {
+        } catch (Exception e) {
-                log.warn("Exception thrown handling device update", e);
+            log.warn("Exception thrown handling device update", e);
-            }
         }
     }
-    private final class InternalDeviceOfflineEventListener
+    private void handleDeviceOfflineEvent(InternalDeviceOfflineEvent event) {
-            implements ClusterMessageHandler {
+        DeviceId deviceId = event.deviceId();
-        @Override
+        Timestamp timestamp = event.timestamp();
-        public void handle(ClusterMessage message) {
-            log.debug("Received device offline event from peer: {}", message.sender());
-            InternalDeviceOfflineEvent event = SERIALIZER.decode(message.payload());
-            DeviceId deviceId = event.deviceId();
+        try {
-            Timestamp timestamp = event.timestamp();
+            notifyDelegateIfNotNull(markOfflineInternal(deviceId, timestamp));
-
+        } catch (Exception e) {
-            try {
+            log.warn("Exception thrown handling device offline", e);
-                notifyDelegateIfNotNull(markOfflineInternal(deviceId, timestamp));
-            } catch (Exception e) {
-                log.warn("Exception thrown handling device offline", e);
-            }
         }
     }
-    private final class InternalRemoveRequestListener
+    private void handleRemoveRequest(DeviceId did) {
-            implements ClusterMessageHandler {
+        try {
-        @Override
+            removeDevice(did);
-        public void handle(ClusterMessage message) {
+        } catch (Exception e) {
-            log.debug("Received device remove request from peer: {}", message.sender());
+            log.warn("Exception thrown handling device remove", e);
-            DeviceId did = SERIALIZER.decode(message.payload());
-
-            try {
-                removeDevice(did);
-            } catch (Exception e) {
-                log.warn("Exception thrown handling device remove", e);
-            }
         }
     }
-    private final class InternalDeviceRemovedEventListener
+    private void handleDeviceRemovedEvent(InternalDeviceRemovedEvent event) {
-            implements ClusterMessageHandler {
+        DeviceId deviceId = event.deviceId();
-        @Override
+        Timestamp timestamp = event.timestamp();
-        public void handle(ClusterMessage message) {
-            log.debug("Received device removed event from peer: {}", message.sender());
-            InternalDeviceRemovedEvent event = SERIALIZER.decode(message.payload());
-
-            DeviceId deviceId = event.deviceId();
-            Timestamp timestamp = event.timestamp();
-            try {
+        try {
-                notifyDelegateIfNotNull(removeDeviceInternal(deviceId, timestamp));
+            notifyDelegateIfNotNull(removeDeviceInternal(deviceId, timestamp));
-            } catch (Exception e) {
+        } catch (Exception e) {
-                log.warn("Exception thrown handling device removed", e);
+            log.warn("Exception thrown handling device removed", e);
-            }
         }
     }
-    private final class InternalPortEventListener
+    private void handlePortEvent(InternalPortEvent event) {
-            implements ClusterMessageHandler {
+        ProviderId providerId = event.providerId();
-        @Override
+        DeviceId deviceId = event.deviceId();
-        public void handle(ClusterMessage message) {
+        Timestamped<List<PortDescription>> portDescriptions = event.portDescriptions();
-
-            log.debug("Received port update event from peer: {}", message.sender());
-            InternalPortEvent event = SERIALIZER.decode(message.payload());
-
-            ProviderId providerId = event.providerId();
-            DeviceId deviceId = event.deviceId();
-            Timestamped<List<PortDescription>> portDescriptions = event.portDescriptions();
-            if (getDevice(deviceId) == null) {
+        if (getDevice(deviceId) == null) {
-                log.debug("{} not found on this node yet, ignoring.", deviceId);
+            log.debug("{} not found on this node yet, ignoring.", deviceId);
-                // Note: dropped information will be recovered by anti-entropy
+            // Note: dropped information will be recovered by anti-entropy
-                return;
+            return;
-            }
+        }
-            try {
+        try {
-                notifyDelegate(updatePortsInternal(providerId, deviceId, portDescriptions));
+            notifyDelegate(updatePortsInternal(providerId, deviceId, portDescriptions));
-            } catch (Exception e) {
+        } catch (Exception e) {
-                log.warn("Exception thrown handling port update", e);
+            log.warn("Exception thrown handling port update", e);
-            }
         }
     }
-    private final class InternalPortStatusEventListener
+    private void handlePortStatusEvent(InternalPortStatusEvent event) {
-            implements ClusterMessageHandler {
+        ProviderId providerId = event.providerId();
-        @Override
+        DeviceId deviceId = event.deviceId();
-        public void handle(ClusterMessage message) {
+        Timestamped<PortDescription> portDescription = event.portDescription();
-
-            log.debug("Received port status update event from peer: {}", message.sender());
-            InternalPortStatusEvent event = SERIALIZER.decode(message.payload());
-            ProviderId providerId = event.providerId();
+        if (getDevice(deviceId) == null) {
-            DeviceId deviceId = event.deviceId();
+            log.debug("{} not found on this node yet, ignoring.", deviceId);
-            Timestamped<PortDescription> portDescription = event.portDescription();
+            // Note: dropped information will be recovered by anti-entropy
-
+            return;
-            if (getDevice(deviceId) == null) {
+        }
-                log.debug("{} not found on this node yet, ignoring.", deviceId);
-                // Note: dropped information will be recovered by anti-entropy
-                return;
-            }
-            try {
+        try {
-                notifyDelegateIfNotNull(updatePortStatusInternal(providerId, deviceId, portDescription));
+            notifyDelegateIfNotNull(updatePortStatusInternal(providerId, deviceId, portDescription));
-            } catch (Exception e) {
+        } catch (Exception e) {
-                log.warn("Exception thrown handling port update", e);
+            log.warn("Exception thrown handling port update", e);
-            }
         }
     }
-    private final class InternalDeviceAdvertisementListener
+    private void handleDeviceAdvertisement(DeviceAntiEntropyAdvertisement advertisement) {
-            implements ClusterMessageHandler {
+        try {
-        @Override
+            handleAdvertisement(advertisement);
-        public void handle(ClusterMessage message) {
+        } catch (Exception e) {
-            log.trace("Received Device Anti-Entropy advertisement from peer: {}", message.sender());
+            log.warn("Exception thrown handling Device advertisements.", e);
-            DeviceAntiEntropyAdvertisement advertisement = SERIALIZER.decode(message.payload());
-            try {
-                handleAdvertisement(advertisement);
-            } catch (Exception e) {
-                log.warn("Exception thrown handling Device advertisements.", e);
-            }
         }
     }
-    private final class DeviceInjectedEventListener
+    private void handleDeviceInjectedEvent(DeviceInjectedEvent event) {
-            implements ClusterMessageHandler {
+        ProviderId providerId = event.providerId();
-        @Override
+        DeviceId deviceId = event.deviceId();
-        public void handle(ClusterMessage message) {
+        DeviceDescription deviceDescription = event.deviceDescription();
-            log.debug("Received injected device event from peer: {}", message.sender());
+        if (!deviceClockService.isTimestampAvailable(deviceId)) {
-            DeviceInjectedEvent event = SERIALIZER.decode(message.payload());
+            // workaround for ONOS-1208
-
+            log.warn("Not ready to accept update. Dropping {}", deviceDescription);
-            ProviderId providerId = event.providerId();
+            return;
-            DeviceId deviceId = event.deviceId();
+        }
-            DeviceDescription deviceDescription = event.deviceDescription();
-            if (!deviceClockService.isTimestampAvailable(deviceId)) {
-                // workaround for ONOS-1208
-                log.warn("Not ready to accept update. Dropping {}", deviceDescription);
-                return;
-            }
-            try {
+        try {
-                createOrUpdateDevice(providerId, deviceId, deviceDescription);
+            createOrUpdateDevice(providerId, deviceId, deviceDescription);
-            } catch (Exception e) {
+        } catch (Exception e) {
-                log.warn("Exception thrown handling device injected event.", e);
+            log.warn("Exception thrown handling device injected event.", e);
-            }
         }
     }
-    private final class PortInjectedEventListener
+    private void handlePortInjectedEvent(PortInjectedEvent event) {
-            implements ClusterMessageHandler {
+        ProviderId providerId = event.providerId();
-        @Override
+        DeviceId deviceId = event.deviceId();
-        public void handle(ClusterMessage message) {
+        List<PortDescription> portDescriptions = event.portDescriptions();
-            log.debug("Received injected port event from peer: {}", message.sender());
+        if (!deviceClockService.isTimestampAvailable(deviceId)) {
-            PortInjectedEvent event = SERIALIZER.decode(message.payload());
+            // workaround for ONOS-1208
-
+            log.warn("Not ready to accept update. Dropping {}", portDescriptions);
-            ProviderId providerId = event.providerId();
+            return;
-            DeviceId deviceId = event.deviceId();
+        }
-            List<PortDescription> portDescriptions = event.portDescriptions();
-            if (!deviceClockService.isTimestampAvailable(deviceId)) {
-                // workaround for ONOS-1208
-                log.warn("Not ready to accept update. Dropping {}", portDescriptions);
-                return;
-            }
-            try {
+        try {
-                updatePorts(providerId, deviceId, portDescriptions);
+            updatePorts(providerId, deviceId, portDescriptions);
-            } catch (Exception e) {
+        } catch (Exception e) {
-                log.warn("Exception thrown handling port injected event.", e);
+            log.warn("Exception thrown handling port injected event.", e);
-            }
         }
     }
--- a/core/store/dist/src/test/java/org/onosproject/store/device/impl/GossipDeviceStoreTest.java
View file @46ab5cc
+++ b/core/store/dist/src/test/java/org/onosproject/store/device/impl/GossipDeviceStoreTest.java
View file @46ab5cc
@@ -17,7 +17,6 @@ package org.onosproject.store.device.impl;
 import com.google.common.collect.Iterables;
 import com.google.common.collect.Sets;
-
 import org.easymock.Capture;
 import org.junit.After;
 import org.junit.AfterClass;
@@ -55,12 +54,12 @@ import org.onosproject.store.Timestamp;
 import org.onosproject.store.cluster.StaticClusterService;
 import org.onosproject.store.cluster.messaging.ClusterCommunicationService;
 import org.onosproject.store.cluster.messaging.ClusterMessage;
-import org.onosproject.store.cluster.messaging.ClusterMessageHandler;
 import org.onosproject.store.cluster.messaging.MessageSubject;
 import org.onosproject.store.impl.MastershipBasedTimestamp;
 import org.onosproject.store.service.EventuallyConsistentMap;
 import org.onosproject.store.service.EventuallyConsistentMapBuilder;
 import org.onosproject.store.service.StorageService;
+
 import java.io.IOException;
 import java.util.Arrays;
 import java.util.Collections;
@@ -70,20 +69,37 @@ import java.util.Map;
 import java.util.Set;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.CountDownLatch;
-import java.util.concurrent.ExecutorService;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.AtomicLong;
 import java.util.function.BiFunction;
 import java.util.function.Function;
 import static java.util.Arrays.asList;
-import static org.easymock.EasyMock.*;
+import static org.easymock.EasyMock.anyObject;
-import static org.junit.Assert.*;
+import static org.easymock.EasyMock.capture;
+import static org.easymock.EasyMock.createNiceMock;
+import static org.easymock.EasyMock.expect;
+import static org.easymock.EasyMock.expectLastCall;
+import static org.easymock.EasyMock.replay;
+import static org.easymock.EasyMock.reset;
+import static org.easymock.EasyMock.verify;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertNull;
+import static org.junit.Assert.assertTrue;
+import static org.junit.Assert.fail;
 import static org.onosproject.cluster.ControllerNode.State.ACTIVE;
 import static org.onosproject.net.DefaultAnnotations.union;
 import static org.onosproject.net.Device.Type.SWITCH;
 import static org.onosproject.net.DeviceId.deviceId;
-import static org.onosproject.net.device.DeviceEvent.Type.*;
+import static org.onosproject.net.device.DeviceEvent.Type.DEVICE_ADDED;
+import static org.onosproject.net.device.DeviceEvent.Type.DEVICE_AVAILABILITY_CHANGED;
+import static org.onosproject.net.device.DeviceEvent.Type.DEVICE_REMOVED;
+import static org.onosproject.net.device.DeviceEvent.Type.DEVICE_UPDATED;
+import static org.onosproject.net.device.DeviceEvent.Type.PORT_ADDED;
+import static org.onosproject.net.device.DeviceEvent.Type.PORT_REMOVED;
+import static org.onosproject.net.device.DeviceEvent.Type.PORT_UPDATED;
 // TODO add tests for remote replication
@@ -157,9 +173,6 @@ public class GossipDeviceStoreTest {
     @Before
     public void setUp() throws Exception {
         clusterCommunicator = createNiceMock(ClusterCommunicationService.class);
-        clusterCommunicator.addSubscriber(anyObject(MessageSubject.class),
-                                          anyObject(ClusterMessageHandler.class), anyObject(ExecutorService.class));
-        expectLastCall().anyTimes();
         replay(clusterCommunicator);
         ClusterService clusterService = new TestClusterService();