Refactor Timestamp class to be safer. Add methods to minus a timestam… (#367)

1 files changed, 1 insertions, 7 deletions

diff --git a/testing/src/main/java/io/opencensus/testing/common/TestClock.java b/testing/src/main/java/io/opencensus/testing/common/TestClock.java
index 2a604f2a..ef287456 100644
--- a/testing/src/main/java/io/opencensus/testing/common/TestClock.java
+++ b/testing/src/main/java/io/opencensus/testing/common/TestClock.java
@@ -66,14 +66,8 @@ public final class TestClock extends Clock {
    *
    * @param duration the increase in time.
    */
-  // TODO(sebright): Consider adding an 'addDuration' method to Timestamp.
   public synchronized void advanceTime(Duration duration) {
-    currentTime =
-        validateNanos(
-            Timestamp.create(
-                    LongMath.checkedAdd(currentTime.getSeconds(), duration.getSeconds()),
-                    currentTime.getNanos())
-                .addNanos(duration.getNanos()));
+    currentTime = validateNanos(currentTime.addDuration(duration));
   }
 
   @Override