Forward arguments Task#wait -> Promise#wait including timeout:.

Author: ioquatix

lib/async/task.rb

@@ -264,14 +264,15 @@ def async(*arguments, **options, &block)
 		#
 		# Conceptually speaking, waiting on a task should return a result, and if it throws an exception, this is certainly an exceptional case that should represent a failure in your program, not an expected outcome. In other words, you should not design your programs to expect exceptions from `#wait` as a normal flow control, and prefer to catch known exceptions within the task itself and return a result that captures the intention of the failure, e.g. a `TimeoutError` might simply return `nil` or `false` to indicate that the operation did not generate a valid result (as a timeout was an expected outcome of the internal operation in this case).
 		#
+		# @parameter timeout [Numeric] The maximum number of seconds to wait for the result before raising a `TimeoutError`, if specified.
 		# @raises [RuntimeError] If the task's fiber is the current fiber.
 		# @returns [Object] The final expression/result of the task's block.
 		# @asynchronous This method is thread-safe.
-		def wait
+		def wait(...)
 			raise "Cannot wait on own fiber!" if Fiber.current.equal?(@fiber)
 
 			# Wait for the task to complete:
-			@promise.wait
+			@promise.wait(...)
 		end
 
 		# For compatibility with `Thread#join` and similar interfaces.

test/async/task.rb

@@ -868,6 +868,46 @@ def sleep_forever
 			expect(task.wait).to be == error
 			expect(task.result).to be == error
 		end
+		
+		it "raises TimeoutError if timeout expires before task completes" do
+			reactor.run do
+				slow_task = reactor.async do
+					sleep(10)
+					:done
+				end
+				
+				expect do
+					slow_task.wait(timeout: 0.001)
+				end.to raise_exception(Async::TimeoutError)
+				
+				slow_task.cancel
+			end
+		end
+		
+		it "returns the result if the task completes within the timeout" do
+			reactor.run do
+				fast_task = reactor.async do
+					:done
+				end
+				
+				expect(fast_task.wait(timeout: 10)).to be == :done
+			end
+		end
+		
+		it "raises TimeoutError immediately if timeout is zero and task is not complete" do
+			reactor.run do
+				slow_task = reactor.async do
+					sleep(10)
+					:done
+				end
+				
+				expect do
+					slow_task.wait(timeout: 0)
+				end.to raise_exception(Async::TimeoutError)
+				
+				slow_task.cancel
+			end
+		end
 	end
 
 	with "#wait_all" do