commit cc80878381d589bce3b299eb2e5b538631c82d14
author Shreya Bhattarai <shreyax@google.com> Wed Feb 10 22:17:00 2016 +0000
committer Dmitry Lomov <dslomov@google.com> Thu Feb 11 11:49:51 2016 +0000
tree 31dbcd2f938c9ad0289add2bc963df0987b656ca
parent 55b3abb12da8b231623f1272f47875fa2634eb3b

Allow for more flexibility in choices of Version for the GraphConcurrencyTest by allowing us to set an initial version, as well as a method to return the version's successor.

--
MOS_MIGRATED_REVID=114367424

src/test/java/com/google/devtools/build/skyframe/InMemoryGraphConcurrencyTest.java

diff --git a/src/test/java/com/google/devtools/build/skyframe/InMemoryGraphConcurrencyTest.java b/src/test/java/com/google/devtools/build/skyframe/InMemoryGraphConcurrencyTest.java
index d02200e..28f0da3 100644
--- a/src/test/java/com/google/devtools/build/skyframe/InMemoryGraphConcurrencyTest.java
+++ b/src/test/java/com/google/devtools/build/skyframe/InMemoryGraphConcurrencyTest.java
@@ -13,6 +13,8 @@
 // limitations under the License.
 package com.google.devtools.build.skyframe;
 
+import com.google.devtools.build.lib.util.Preconditions;
+
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
 
@@ -22,6 +24,17 @@
   private ProcessableGraph graph;
 
   @Override
+  protected Version getStartingVersion() {
+    return IntVersion.of(0);
+  }
+
+  @Override
+  protected Version getNextVersion(Version v) {
+    Preconditions.checkState(v instanceof IntVersion);
+    return ((IntVersion) v).next();
+  }
+
+  @Override
   protected void makeGraph() {
     graph = new InMemoryGraph();
   }