From 61f37b756a6464a25e001df49a2110f0bc2b5520 Mon Sep 17 00:00:00 2001
From: mmuller <mmuller@google.com>
Date: Thu, 16 Jun 2016 10:45:07 -0700
Subject: [PATCH] Add TimedTransitionProperty.getTransitionTime()

Add a method to get the next transition time so that we can return the expiry
date along with the EAP fee for a given time.
-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=125076484
---
 .../model/common/TimedTransitionProperty.java         | 10 ++++++++++
 .../model/common/TimedTransitionPropertyTest.java     | 11 +++++++++++
 2 files changed, 21 insertions(+)

diff --git a/java/google/registry/model/common/TimedTransitionProperty.java b/java/google/registry/model/common/TimedTransitionProperty.java
index b1eef1dfc..e261f1e9a 100644
--- a/java/google/registry/model/common/TimedTransitionProperty.java
+++ b/java/google/registry/model/common/TimedTransitionProperty.java
@@ -35,6 +35,8 @@ import org.joda.time.DateTime;
 import java.util.NavigableMap;
 import java.util.TreeMap;
 
+import javax.annotation.Nullable;
+
 /**
  * An entity property whose value transitions over time.  Each value it takes on becomes active
  * at a corresponding instant, and remains active until the next transition occurs.  At least one
@@ -203,4 +205,12 @@ public class TimedTransitionProperty<V, T extends TimedTransitionProperty.TimedT
     // where any given time earlier than START_OF_TIME is replaced by START_OF_TIME.
     return backingMap.floorEntry(latestOf(START_OF_TIME, time)).getValue().getValue();
   }
+
+  /**
+   * Returns the time of the next transition.  Returns null if there is no subsequent transition.
+   */
+  @Nullable
+  public DateTime getNextTransitionAfter(DateTime time) {
+    return backingMap.higherKey(latestOf(START_OF_TIME, time));
+  }
 }
diff --git a/javatests/google/registry/model/common/TimedTransitionPropertyTest.java b/javatests/google/registry/model/common/TimedTransitionPropertyTest.java
index 7cbb37c61..19263253e 100644
--- a/javatests/google/registry/model/common/TimedTransitionPropertyTest.java
+++ b/javatests/google/registry/model/common/TimedTransitionPropertyTest.java
@@ -106,6 +106,17 @@ public class TimedTransitionPropertyTest {
     testGetValueAtTime(timedString);
   }
 
+  @Test
+  public void testSuccess_getNextTransitionAfter() throws Exception {
+    assertThat(timedString.getNextTransitionAfter(A_LONG_TIME_AGO)).isEqualTo(DATE_1);
+    assertThat(timedString.getNextTransitionAfter(START_OF_TIME.plusMillis(1))).isEqualTo(DATE_1);
+    assertThat(timedString.getNextTransitionAfter(DATE_1.minusMillis(1))).isEqualTo(DATE_1);
+    assertThat(timedString.getNextTransitionAfter(DATE_1)).isEqualTo(DATE_2);
+    assertThat(timedString.getNextTransitionAfter(DATE_2.minusMillis(1))).isEqualTo(DATE_2);
+    assertThat(timedString.getNextTransitionAfter(DATE_2)).isEqualTo(DATE_3);
+    assertThat(timedString.getNextTransitionAfter(DATE_3)).isNull();
+  }
+
   @Test
   public void testSuccess_simulatedLoad() throws Exception {
     // Just for testing, don't extract transitions from a TimedTransitionProperty in real code.