commit 61aeb356774fcaa478b2ded0ae505086daae13e0
author Jordan Halterman <jordan@opennetworking.org> Wed Oct 18 16:22:17 2017 -0700
committer Thomas Vachuska <tom@opennetworking.org> Wed Nov 01 18:19:36 2017 +0000
tree 9458548af433b87de5f5c95c5426cec428b40c73
parent 64047219f92938d0222c566c60a59155379595f8

Add upgrade event listener to UpgradeService on initialization of MastershipManager

Change-Id: Ic1d419d1c5908bddee633897391fa6ef11c525b3

core/net/src/test/java/org/onosproject/cluster/impl/MastershipManagerTest.java

diff --git a/core/net/src/test/java/org/onosproject/cluster/impl/MastershipManagerTest.java b/core/net/src/test/java/org/onosproject/cluster/impl/MastershipManagerTest.java
index 14c40fd..0f75445 100644
--- a/core/net/src/test/java/org/onosproject/cluster/impl/MastershipManagerTest.java
+++ b/core/net/src/test/java/org/onosproject/cluster/impl/MastershipManagerTest.java
@@ -49,6 +49,7 @@
 
 import com.google.common.collect.Sets;
 import com.google.common.util.concurrent.Futures;
+import org.onosproject.upgrade.impl.UpgradeServiceAdapter;
 
 import static org.easymock.EasyMock.anyObject;
 import static org.easymock.EasyMock.expect;
@@ -104,6 +105,7 @@
         injectEventDispatcher(mgr, new TestEventDispatcher());
         testClusterService = new TestClusterService();
         mgr.clusterService = testClusterService;
+        mgr.upgradeService = new UpgradeServiceAdapter();
         mgr.store = new TestSimpleMastershipStore(mgr.clusterService);
         regionStore = new DistributedRegionStore();
         TestUtils.setField(regionStore, "storageService", new TestStorageService());