commit 07f052bcf3bf6d27a8ad1b31077177fdf85ca8b9
author Jordan Halterman <jordan@opennetworking.org> Sun Oct 08 14:22:41 2017 -0700
committer Thomas Vachuska <tom@opennetworking.org> Sat Nov 04 01:27:43 2017 +0000
tree 247b032ef235a67c3091409339359e2feb0278e1
parent 61b0862d193692bab060ae86d49b7da2d3fd9d29

[ONOS-7088] Distribute storage partitions evenly during upgrades

Change-Id: Id82f86ddedbe6c7de2322717338c5c341177bc9e

core/net/src/test/java/org/onosproject/upgrade/impl/UpgradeManagerTest.java

diff --git a/core/net/src/test/java/org/onosproject/upgrade/impl/UpgradeManagerTest.java b/core/net/src/test/java/org/onosproject/upgrade/impl/UpgradeManagerTest.java
index 78127ee..4df9d46 100644
--- a/core/net/src/test/java/org/onosproject/upgrade/impl/UpgradeManagerTest.java
+++ b/core/net/src/test/java/org/onosproject/upgrade/impl/UpgradeManagerTest.java
@@ -35,6 +35,7 @@
 import org.onosproject.cluster.MembershipGroup;
 import org.onosproject.cluster.MembershipServiceAdapter;
 import org.onosproject.cluster.NodeId;
+import org.onosproject.common.event.impl.TestEventDispatcher;
 import org.onosproject.core.Version;
 import org.onosproject.core.VersionServiceAdapter;
 import org.onosproject.store.service.AsyncAtomicValue;
@@ -49,6 +50,7 @@
 import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertTrue;
 import static org.junit.Assert.fail;
+import static org.onosproject.net.NetTestTools.injectEventDispatcher;
 
 /**
  * Upgrade manager test.
@@ -66,6 +68,7 @@
     @SuppressWarnings("unchecked")
     private UpgradeManager createUpgradeManager(Version version, Upgrade state, List<Version> versions) {
         UpgradeManager upgradeManager = new UpgradeManager();
+        injectEventDispatcher(upgradeManager, new TestEventDispatcher());
         upgradeManager.membershipService = new MembershipServiceAdapter() {
             @Override
             public MembershipGroup getLocalGroup() {