Better: Add ruby 3.1 compatibility.

travisjeffery · Aug 10, 2023 · e3b4c9f · e3b4c9f
1 parent 0ad9c66
commit e3b4c9f
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/lib/timecop/time_extensions.rb b/lib/timecop/time_extensions.rb
@@ -10,16 +10,24 @@ def mock_time
 
     alias_method :now_without_mock_time, :now
 
-    def now_with_mock_time
-      mock_time || now_without_mock_time
+    def now_with_mock_time(*args)
+      mock_time || now_without_mock_time(*args)
     end
 
     alias_method :now, :now_with_mock_time
 
     alias_method :new_without_mock_time, :new
 
-    def new_with_mock_time(*args)
-      args.size <= 0 ? now : new_without_mock_time(*args)
+    def new_with_mock_time(*args, **kwargs)
+      if args.size <= 0
+        now(*args)
+      else
+        if kwargs == {}
+          new_without_mock_time(*args)
+        else
+          new_without_mock_time(*args, **kwargs)
+        end
+      end
     end
 
     ruby2_keywords :new_with_mock_time if Module.private_method_defined?(:ruby2_keywords)