From e27d015ff99d6dff41e5d2c06c23f6dd3b09d107 Mon Sep 17 00:00:00 2001
From: Damien Elmes <gpg@ankiweb.net>
Date: Thu, 30 Apr 2020 09:33:02 +1000
Subject: [PATCH] expose clock hack via env var

closes #594
---
 pylib/Makefile         |  2 +-
 rslib/src/timestamp.rs | 40 ++++++++++++++++++++--------------------
 2 files changed, 21 insertions(+), 21 deletions(-)

diff --git a/pylib/Makefile b/pylib/Makefile
index 5e5cba96f..b9e6aa633 100644
--- a/pylib/Makefile
+++ b/pylib/Makefile
@@ -79,7 +79,7 @@ CHECKDEPS := $(shell "${FIND}" anki tests -name '*.py' | grep -v buildinfo.py)
 	@touch $@
 
 .build/test: $(CHECKDEPS)
-	python -m pytest --durations=1 -s
+	SHIFT_CLOCK_HACK=1 python -m pytest --durations=1 -s
 	@touch $@
 
 .build/lint: $(CHECKDEPS)
diff --git a/rslib/src/timestamp.rs b/rslib/src/timestamp.rs
index b1a419579..e09e4e50b 100644
--- a/rslib/src/timestamp.rs
+++ b/rslib/src/timestamp.rs
@@ -2,6 +2,8 @@
 // License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
 
 use crate::define_newtype;
+use lazy_static::lazy_static;
+use std::env;
 use std::time;
 
 define_newtype!(TimestampSecs, i64);
@@ -19,28 +21,26 @@ impl TimestampMillis {
     }
 }
 
-#[cfg(not(test))]
-fn elapsed() -> time::Duration {
-    time::SystemTime::now()
-        .duration_since(time::SystemTime::UNIX_EPOCH)
-        .unwrap()
+lazy_static! {
+    static ref TESTING: bool = env::var("SHIFT_CLOCK_HACK").is_ok();
 }
 
-// when running in CI, shift the current time away from the cutoff point
-// to accomodate unit tests that depend on the current time
-#[cfg(test)]
 fn elapsed() -> time::Duration {
-    use chrono::{Local, Timelike};
-
-    let now = Local::now();
-
-    let mut elap = time::SystemTime::now()
-        .duration_since(time::SystemTime::UNIX_EPOCH)
-        .unwrap();
-
-    if now.hour() >= 2 && now.hour() < 4 {
-        elap -= time::Duration::from_secs(60 * 60 * 2);
+    if *TESTING {
+        // shift clock around rollover time to accomodate Python tests that make bad assumptions.
+        // we should update the tests in the future and remove this hack.
+        use chrono::{Local, Timelike};
+        let mut elap = time::SystemTime::now()
+            .duration_since(time::SystemTime::UNIX_EPOCH)
+            .unwrap();
+        let now = Local::now();
+        if now.hour() >= 2 && now.hour() < 4 {
+            elap -= time::Duration::from_secs(60 * 60 * 2);
+        }
+        elap
+    } else {
+        time::SystemTime::now()
+            .duration_since(time::SystemTime::UNIX_EPOCH)
+            .unwrap()
     }
-
-    elap
 }