fix and test for timeout

mishankov · Jan 21, 2024 · ce207bc · ce207bc
1 parent 5777281
commit ce207bc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/src/yahttp/internal/utils.nim b/src/yahttp/internal/utils.nim
@@ -16,6 +16,7 @@ macro http_method_gen*(name: untyped): untyped =
         query = query,
         body = body,
         auth = auth,
+        timeout = timeout,
         ignoreSsl = ignoreSsl
       )
 
@@ -34,5 +35,6 @@ macro http_method_no_body_gen*(name: untyped): untyped =
         headers = headers,
         query = query,
         auth = auth,
+        timeout = timeout,
         ignoreSsl = ignoreSsl
       )
diff --git a/tests/int/test_yahttp.nim b/tests/int/test_yahttp.nim
@@ -38,3 +38,10 @@ test "Test JSON body":
   let jsonResp = put(BASE_URL & "/put", headers = {"Content-Type": "application/json"}, body = $ %*{"key": "value"}).json()
 
   check jsonResp["json"]["key"].getStr() == "value"
+
+test "Test timeout":
+  expect TimeoutError:
+    discard get(BASE_URL & "/delay/5", timeout = 100)
+
+  # No exception
+  discard get(BASE_URL & "/delay/5", timeout = -1)