Add support for setting a custom monotonic time function in mono_time

1 files changed, 24 insertions, 0 deletions

diff --git a/toxcore/mono_time_test.cc b/toxcore/mono_time_test.cc
index 4ed59406..6e81ac7e 100644
--- a/toxcore/mono_time_test.cc
+++ b/toxcore/mono_time_test.cc
@@ -35,4 +35,28 @@ TEST(MonoTime, IsTimeout) {
   mono_time_free(mono_time);
 }
 
+static uint64_t test_current_time_callback(void *user_data) { return *(uint64_t *)user_data; }
+
+TEST(MonoTime, CustomTime) {
+  Mono_Time *mono_time = mono_time_new();
+
+  uint64_t test_time = current_time_monotonic(mono_time) + 42137;
+
+  mono_time_set_current_time_callback(mono_time, test_current_time_callback, &test_time);
+  mono_time_update(mono_time);
+
+  EXPECT_EQ(current_time_monotonic(mono_time), test_time);
+
+  uint64_t const start = mono_time_get(mono_time);
+
+  test_time += 7000;
+
+  mono_time_update(mono_time);
+  EXPECT_EQ(mono_time_get(mono_time) - start, 7);
+
+  EXPECT_EQ(current_time_monotonic(mono_time), test_time);
+
+  mono_time_free(mono_time);
+}
+
 }  // namespace