Skip to content

Commit 1df8687

Browse files
committed
Add TimeTravel.return method and preserve float precision
1 parent 1c8fafa commit 1df8687

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

spec/support/time_travel.rb

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ class << self
88
def freeze(time = Time.now)
99
if block_given?
1010
Timecop.freeze(time) do
11-
redis_time_ms = (Time.now.to_f * 1000).to_i
11+
redis_time_ms = Time.now.to_f * 1000.0
1212
redis.rpush(STACK_KEY, redis_time_ms)
1313
begin
1414
yield
@@ -19,11 +19,16 @@ def freeze(time = Time.now)
1919
else
2020
redis.del(STACK_KEY)
2121
Timecop.freeze(time)
22-
redis_time_ms = (Time.now.to_f * 1000).to_i
22+
redis_time_ms = Time.now.to_f * 1000.0
2323
redis.rpush(STACK_KEY, redis_time_ms)
2424
end
2525
end
2626

27+
def return
28+
redis.del(STACK_KEY)
29+
Timecop.return
30+
end
31+
2732
private
2833

2934
def redis

0 commit comments

Comments
 (0)