commit f70bf462f8ac59446d399be8866b2647d75ca4c8
author Jordan Halterman <jordan@opennetworking.org> Sat Jul 29 13:12:00 2017 -0700
committer Ray Milkey <ray@opennetworking.org> Tue Aug 01 16:03:57 2017 +0000
tree 8135610f8b9ff40585107d6a55a6b81affc15379
parent a966c34402da0f306576ce4cd90c8e2d418cb47d

[ONOS-6756] Replicate node version information for ISSU

Change-Id: Ibd31c573990f2732b7abf8615ca914ffb77615ec

core/common/src/test/java/org/onosproject/store/trivial/SimpleMastershipStore.java

diff --git a/core/common/src/test/java/org/onosproject/store/trivial/SimpleMastershipStore.java b/core/common/src/test/java/org/onosproject/store/trivial/SimpleMastershipStore.java
index ad4386e..2a8caf0 100644
--- a/core/common/src/test/java/org/onosproject/store/trivial/SimpleMastershipStore.java
+++ b/core/common/src/test/java/org/onosproject/store/trivial/SimpleMastershipStore.java
@@ -45,6 +45,8 @@
 import org.onosproject.cluster.DefaultControllerNode;
 import org.onosproject.cluster.NodeId;
 import org.onosproject.cluster.RoleInfo;
+import org.onosproject.core.Version;
+import org.onosproject.core.VersionService;
 import org.onosproject.mastership.MastershipEvent;
 import org.onosproject.mastership.MastershipStore;
 import org.onosproject.mastership.MastershipStoreDelegate;
@@ -75,6 +77,9 @@
     @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY)
     protected ClusterService clusterService;
 
+    @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY)
+    protected VersionService versionService;
+
     //devices mapped to their masters, to emulate multiple nodes
     protected final Map<DeviceId, NodeId> masterMap = new HashMap<>();
     //emulate backups with pile of nodes
@@ -122,6 +127,14 @@
                 }
 
                 @Override
+                public Version getVersion(NodeId nodeId) {
+                    if (instance.id().equals(nodeId)) {
+                        return versionService.version();
+                    }
+                    return null;
+                }
+
+                @Override
                 public DateTime getLastUpdated(NodeId nodeId) {
                     return creationTime;
                 }