unit testing time API

rusefi · Apr 27, 2024 · a61291f · a61291f
1 parent fef5317
commit a61291f
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/mock/lib-time-mocks.cpp b/mock/lib-time-mocks.cpp
@@ -1,6 +1,6 @@
 #include <rusefi/rusefi_time_math.h>
 
-int timeNowUs = 0;
+static int timeNowUs = 0;
 
 efitimeus_t getTimeNowUs() {
 	return timeNowUs;
@@ -13,3 +13,11 @@ efitimesec_t getTimeNowS() {
 efitick_t getTimeNowNt() {
 	return getTimeNowUs() * US_TO_NT_MULTIPLIER;
 }
+
+void setTimeNowUs(int us) {
+	timeNowUs = us;
+}
+
+void advanceTimeUs(int us) {
+	timeNowUs += us;
+}
diff --git a/util/include/rusefi/rusefi_time_math.h b/util/include/rusefi/rusefi_time_math.h
@@ -13,3 +13,9 @@
 #define NT2USF(x) (((float)(x)) / US_TO_NT_MULTIPLIER)
 
 #define NT_PER_SECOND (US2NT(US_PER_SECOND_LL))
+
+#if EFI_UNIT_TEST
+// In unit tests, we can time travel...
+void setTimeNowUs(int us);
+void advanceTimeUs(int us);
+#endif