tinyfan 0.2.3__tar.gz → 0.2.4__tar.gz

{tinyfan-0.2.3 → tinyfan-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tinyfan
-Version: 0.2.3
+Version: 0.2.4
 Summary: Tiny Argo Workflows python devkit
 Project-URL: Documentation, https://github.com/eunchuldev/tinyfan#readme
 Project-URL: Issues, https://github.com/eunchuldev/tinyfan/issues

{tinyfan-0.2.3 → tinyfan-0.2.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "tinyfan"
-version = "0.2.3"
+version = "0.2.4"
 description = 'Tiny Argo Workflows python devkit'
 readme = "README.md"
 requires-python = ">=3.12"

{tinyfan-0.2.3 → tinyfan-0.2.4}/src/tinyfan/resources/base.py

@@ -1,10 +1,11 @@
 from ..config import Config, ConfigValue
+import inspect
 
 
 class ResourceBase:
     def get_refs(self) -> set[Config | ConfigValue]:
         refs = set()
-        for _, value in vars(self).items():
+        for _, value in inspect.getmembers(self):
             if isinstance(value, (Config, ConfigValue)):  # or a common base 'Reference'
                 refs.add(value)
             elif isinstance(value, ResourceBase):

tinyfan-0.2.4/tests/test_config.py

@@ -0,0 +1,60 @@
+from tinyfan import Flow, asset, ResourceBase, ConfigMapRef, ConfigMapKeyRef, SecretRef, SecretKeyRef
+from tinyfan.flow import FLOW_REGISTER
+from tinyfan.codegen import codegen
+from typing import Generator
+from unittest import mock
+import pytest
+import os
+
+
+# type: ignore
+@pytest.fixture()
+def sample_flow() -> Generator:
+    class Resource(ResourceBase):
+        cm = ConfigMapRef("cm")
+        cm_key = ConfigMapKeyRef("cm", "key")
+        cm_default = ConfigMapKeyRef("cm", "non_key", "default")
+        sec = SecretRef("secret")
+        sec_key = SecretKeyRef("secret", "key")
+        sec_default = SecretKeyRef("secret", "non_key", "default")
+
+    flow: Flow = Flow(
+        "test",
+        resources={
+            "res": Resource(),
+        },
+    )
+
+    @asset(flow=flow, schedule="@daily")
+    def test_asset(res: Resource):  # type: ignore
+        return (
+            res.cm.get("key"),
+            res.cm_key.get_value(),
+            res.cm_default.get_value(),
+            res.sec.get("key"),
+            res.sec_key.get_value(),
+            res.sec_default.get_value(),
+        )
+
+    yield flow
+    FLOW_REGISTER.clear()
+
+
+# HACK IT: skip it at this stage. too tired.
+def test_resource_config_codgen(validate_crds, sample_flow):
+    res = codegen(flow=sample_flow)
+    validate_crds(res)
+
+
+def test_resource_config_run(sample_flow: Flow):
+    env = {
+        "TINYFAN_CM__cm__key": "cmkey",
+        "TINYFAN_SC__secret__key": "sekey",
+        "key": "key",
+    }
+    with mock.patch.dict(os.environ, env):
+        _ = codegen(flow=sample_flow)
+        asset = list(sample_flow.assets.values())[0]
+        actual, _ = asset.run()
+        expected = ("key", "cmkey", "default", "key", "sekey", "default")
+        assert actual == expected

{tinyfan-0.2.3 → tinyfan-0.2.4}/uv.lock

@@ -784,7 +784,7 @@ wheels = [
 
 [[package]]
 name = "tinyfan"
-version = "0.2.3"
+version = "0.2.4"
 source = { editable = "." }
 dependencies = [
     { name = "croniter" },