chore: add tests for build/run time decorators

src/damavand/resource/resource.py

@@ -7,7 +7,7 @@
 def buildtime(func):
     def wrapper(self, *args, **kwargs):
         if not utils.is_building():
-            return lambda _: None
+            return None
 
         return func(self, *args, **kwargs)
 
@@ -17,7 +17,7 @@ def wrapper(self, *args, **kwargs):
 def runtime(func):
     def wrapper(self, *args, **kwargs):
         if utils.is_building():
-            return lambda _: None
+            return None
 
         return func(self, *args, **kwargs)
 

tests/resources/test_resource.py

@@ -0,0 +1,27 @@
+from _pytest.monkeypatch import MonkeyPatch
+
+from damavand.resource import buildtime, runtime
+
+
+def test_buildtime_decorator(monkeypatch: MonkeyPatch):
+    @buildtime
+    def test_func(_):
+        return "building"
+
+    monkeypatch.setattr("damavand.utils.is_building", lambda: True)
+    assert test_func(None) == "building"
+
+    monkeypatch.setattr("damavand.utils.is_building", lambda: False)
+    assert test_func(None) == None
+
+
+def test_runtime_decorator(monkeypatch: MonkeyPatch):
+    @runtime
+    def test_func(_):
+        return "running"
+
+    monkeypatch.setattr("damavand.utils.is_building", lambda: True)
+    assert test_func(None) == None
+
+    monkeypatch.setattr("damavand.utils.is_building", lambda: False)
+    assert test_func(None) == "running"