commit 2bf177cfa3fca0df86733043c8ee44aaac87aa52
author Jordan Halterman <jordan@onlab.us> Thu Jun 29 01:49:08 2017 -0700
committer Ray Milkey <ray@opennetworking.org> Thu Jul 13 20:29:21 2017 +0000
tree e0e9d7dbae547c52ffff0900e68073da2067c91a
parent 2cfe8ef782a45f4b9be720c321658d29790e9764

[ONOS-6594] Upgrade to Atomix 2.0.0

Change-Id: I6534bca1c8570b4e017f682953b876da29146675

apps/test/distributed-primitives/src/main/java/org/onosproject/distributedprimitives/cli/LeaderElectorTestCommand.java

diff --git a/apps/test/distributed-primitives/src/main/java/org/onosproject/distributedprimitives/cli/LeaderElectorTestCommand.java b/apps/test/distributed-primitives/src/main/java/org/onosproject/distributedprimitives/cli/LeaderElectorTestCommand.java
index ef5e45c..22849e8 100644
--- a/apps/test/distributed-primitives/src/main/java/org/onosproject/distributedprimitives/cli/LeaderElectorTestCommand.java
+++ b/apps/test/distributed-primitives/src/main/java/org/onosproject/distributedprimitives/cli/LeaderElectorTestCommand.java
@@ -21,8 +21,8 @@
 import org.onosproject.cluster.ClusterService;
 import org.onosproject.cluster.Leadership;
 import org.onosproject.cluster.NodeId;
+import org.onosproject.distributedprimitives.DistributedPrimitivesTest;
 import org.onosproject.store.service.LeaderElector;
-import org.onosproject.store.service.StorageService;
 
 import com.google.common.base.Joiner;
 
@@ -49,13 +49,10 @@
 
     @Override
     protected void execute() {
-        StorageService storageService = get(StorageService.class);
         ClusterService clusterService = get(ClusterService.class);
+        DistributedPrimitivesTest test = get(DistributedPrimitivesTest.class);
+        leaderElector = test.getLeaderElector(name);
         NodeId localNodeId = clusterService.getLocalNode().id();
-        leaderElector = storageService.leaderElectorBuilder()
-                .withName(name)
-                .build()
-                .asLeaderElector();
         if ("run".equals(operation)) {
             print(leaderElector.run(topic, localNodeId));
         } else if ("withdraw".equals(operation)) {