From 1850ad0fb7e4cebc250b77f0c5dc18da204f796f Mon Sep 17 00:00:00 2001
From: phi <phi.friday@gmail.com>
Date: Fri, 13 Sep 2024 13:14:45 +0900
Subject: [PATCH] test: init_func

---
 tests/test_executor.py | 41 ++++++++++++++++++++++++++++++++++-------
 1 file changed, 34 insertions(+), 7 deletions(-)

diff --git a/tests/test_executor.py b/tests/test_executor.py
index 0c4fec3..95abe2b 100644
--- a/tests/test_executor.py
+++ b/tests/test_executor.py
@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import time
+import uuid
 from collections.abc import Awaitable
 from itertools import product
 from typing import Any
@@ -15,6 +16,18 @@
 pytestmark = pytest.mark.anyio
 
 
+def sample_func(*args: Any, **kwargs: Any) -> tuple[tuple[Any, ...], dict[str, Any]]:
+    return args, kwargs
+
+
+async def sample_async_func(
+    *args: Any, **kwargs: Any
+) -> tuple[tuple[Any, ...], dict[str, Any]]:
+    await anyio.sleep(0.1)
+
+    return sample_func(*args, **kwargs)
+
+
 class SomeAwaitable:
     def __init__(self, size: int = 5, value: Any = None) -> None:
         self.size = max(size, 5)
@@ -178,13 +191,27 @@ def awaitable_func() -> Awaitable[Any]:
         assert result == expect
 
 
-def sample_func(*args: Any, **kwargs: Any) -> tuple[tuple[Any, ...], dict[str, Any]]:
-    return args, kwargs
+def test_init_func():
+    key, value = str(uuid.uuid4()), str(uuid.uuid4())
 
+    def sample_init(key: str, value: str) -> None:
+        import os
 
-async def sample_async_func(
-    *args: Any, **kwargs: Any
-) -> tuple[tuple[Any, ...], dict[str, Any]]:
-    await anyio.sleep(0.1)
+        os.environ[key.upper()] = value
 
-    return sample_func(*args, **kwargs)
+    def get_init_if_exist(key: str) -> str:
+        import os
+
+        return os.environ.get(key.upper(), "")
+
+    executor = TimeoutExecutor(1)
+    executor.set_initializer(sample_init, key, value=value)
+    assert executor.initializer is not None
+    assert executor.initializer.function is sample_init
+    assert executor.initializer.args == (key,)
+    assert executor.initializer.kwargs == {"value": value}
+
+    async_result = executor.apply(get_init_if_exist, key)
+    result = async_result.result()
+    assert isinstance(result, str)
+    assert result == value