add vpn capability properties

examples/ssh/ssh.py

@@ -15,6 +15,7 @@
 )
 from yapapi.log import enable_default_logger, log_summary, log_event_repr  # noqa
 from yapapi.payload import vm
+from yapapi.payload.vm import VM_CAPS_VPN
 from yapapi.services import Service
 
 examples_dir = pathlib.Path(__file__).resolve().parent.parent
@@ -36,6 +37,7 @@ async def get_payload():
             image_hash="ea233c6774b1621207a48e10b46e3e1f944d881911f499f5cbac546a",
             min_mem_gib=0.5,
             min_storage_gib=2.0,
+            capabilities=[VM_CAPS_VPN],
         )
 
     async def run(self):

tests/props/test_base.py

@@ -9,6 +9,7 @@ class Foo(props_base.Model):
     bar: str = props_base.prop("bar", "cafebiba")
     max_baz: int = props_base.constraint("baz", "<=", 100)
     min_baz: int = props_base.constraint("baz", ">=", 1)
+    lst: list = props_base.constraint("lst", "=", default_factory=list)
 
 
 @dataclass
@@ -23,7 +24,7 @@ class FooZero(props_base.Model):
 
 def test_constraint_fields():
     fields = Foo.constraint_fields()
-    assert len(fields) == 2
+    assert len(fields) == 3
     assert any(f.name == "max_baz" for f in fields)
     assert all(f.name != "bar" for f in fields)
 
@@ -41,79 +42,103 @@ def test_constraint_to_str():
     assert props_base.constraint_to_str(foo.max_baz, max_baz) == "(baz<=42)"
 
 
+@pytest.mark.parametrize(
+    "value, constraint_str", [
+        (["one"], "(lst=one)"),
+        (["one", "two"], "(&(lst=one)\n\t(lst=two))"),
+    ]
+)
+def test_constraint_to_str_list(value, constraint_str):
+    foo = Foo(lst=value)
+    lst = [f for f in foo.constraint_fields() if f.name == "lst"][0]
+    assert props_base.constraint_to_str(foo.lst, lst) == constraint_str
+
+
 def test_constraint_model_serialize():
     foo = Foo()
     constraints = props_base.constraint_model_serialize(foo)
-    assert constraints == ["(baz<=100)", "(baz>=1)"]
+    assert constraints == ["(baz<=100)", "(baz>=1)", ""]
 
 
 @pytest.mark.parametrize(
     "model, operator, result, error",
     [
         (
-            Foo,
+            Foo(),
             None,
             "(&(baz<=100)\n\t(baz>=1))",
             False,
         ),
         (
-            Foo,
+            Foo(),
             "&",
             "(&(baz<=100)\n\t(baz>=1))",
             False,
         ),
         (
-            Foo,
+            Foo(lst=["one"]),
+            None,
+            "(&(baz<=100)\n\t(baz>=1)\n\t(lst=one))",
+            False,
+        ),
+        (
+            Foo(lst=["one", "other"]),
+            None,
+            "(&(baz<=100)\n\t(baz>=1)\n\t(&(lst=one)\n\t(lst=other)))",
+            False,
+        ),
+        (
+            Foo(),
             "|",
             "(|(baz<=100)\n\t(baz>=1))",
             False,
         ),
         (
-            Foo,
+            Foo(),
             "!",
             None,
             True,
         ),
         (
-            FooToo,
+            FooToo(),
             "!",
             "(!(baz=21))",
             False,
         ),
         (
-            FooToo,
+            FooToo(),
             "&",
             "(baz=21)",
             False,
         ),
         (
-            FooToo,
+            FooToo(),
             "|",
             "(baz=21)",
             False,
         ),
         (
-            FooZero,
+            FooZero(),
             None,
             "(&)",
             False,
         ),
         (
-            FooZero,
+            FooZero(),
             "&",
             "(&)",
             False,
         ),
         (
-            FooZero,
+            FooZero(),
             "|",
             "(|)",
             False,
         ),
     ],
 )
 def test_join_str_constraints(model, operator, result, error):
-    args = [props_base.constraint_model_serialize(model())]
+    args = [props_base.constraint_model_serialize(model)]
     if operator:
         args.append(operator)
     try:

yapapi/payload/vm.py

@@ -2,8 +2,14 @@
 from dataclasses import dataclass, field
 from enum import Enum
 import logging
-from typing import Optional
+import sys
+from typing import Optional, List
 from typing_extensions import Final
+if sys.version_info > (3, 8):
+    from typing import Literal
+else:
+    from typing_extensions import Literal
+
 from srvresolver.srv_resolver import SRVResolver, SRVRecord  # type: ignore
 
 from yapapi.payload.package import (
@@ -22,6 +28,9 @@
 
 logger = logging.getLogger(__name__)
 
+VM_CAPS_VPN: str = "vpn"
+
+VmCaps = Literal[VM_CAPS_VPN,]
 
 @dataclass
 class InfVm(InfBase):
@@ -47,7 +56,8 @@ class _VmConstraints:
     min_mem_gib: float = prop_base.constraint(inf.INF_MEM, operator=">=")
     min_storage_gib: float = prop_base.constraint(inf.INF_STORAGE, operator=">=")
     min_cpu_threads: int = prop_base.constraint(inf.INF_THREADS, operator=">=")
-    # cores: int = prop_base.constraint(inf.INF_CORES, operator=">=")
+
+    capabilities: List[VmCaps] = prop_base.constraint("golem.runtime.capabilities", operator="=", default_factory=list)
 
     runtime: str = prop_base.constraint(inf.INF_RUNTIME_NAME, operator="=", default=RUNTIME_VM)
 
@@ -80,6 +90,7 @@ async def repo(
     min_mem_gib: float = 0.5,
     min_storage_gib: float = 2.0,
     min_cpu_threads: int = 1,
+    capabilities: Optional[List[VmCaps]] = None,
 ) -> Package:
     """
     Build a reference to application package.
@@ -89,13 +100,15 @@ async def repo(
     :param min_mem_gib: minimal memory required to execute application code
     :param min_storage_gib: minimal disk storage to execute tasks
     :param min_cpu_threads: minimal available logical CPU cores
+    :param capabilities: an optional list of required vm capabilities
     :return: the payload definition for the given VM image
     """
+    capabilities = capabilities or list()
     return _VmPackage(
         repo_url=resolve_repo_srv(_DEFAULT_REPO_SRV),
         image_hash=image_hash,
         image_url=image_url,
-        constraints=_VmConstraints(min_mem_gib, min_storage_gib, min_cpu_threads),
+        constraints=_VmConstraints(min_mem_gib, min_storage_gib, min_cpu_threads, capabilities),
     )
 
 

yapapi/props/base.py

@@ -205,7 +205,7 @@ class ModelFieldType(enum.Enum):
     property = "property"
 
 
-def constraint(key: str, operator: ConstraintOperator = "=", default=MISSING):
+def constraint(key: str, operator: ConstraintOperator = "=", default=MISSING, default_factory=MISSING):
     """
     Return a constraint-type dataclass field for a Model.
 
@@ -229,6 +229,7 @@ def constraint(key: str, operator: ConstraintOperator = "=", default=MISSING):
     """
     return field(
         default=default,
+        default_factory=default_factory,
         metadata={
             PROP_KEY: key,
             PROP_OPERATOR: operator,
@@ -271,7 +272,10 @@ def constraint_to_str(value, f: Field) -> str:
     :param value: the value of the the constraint field
     :param f: the dataclass field for this constraint
     """
-    return f"({f.metadata[PROP_KEY]}{f.metadata[PROP_OPERATOR]}{value})"
+    if type(value) == list:
+        return join_str_constraints([constraint_to_str(v, f) for v in value]) if value else ""
+    else:
+        return f"({f.metadata[PROP_KEY]}{f.metadata[PROP_OPERATOR]}{value})"
 
 
 def constraint_model_serialize(m: Model) -> List[str]:
@@ -316,6 +320,8 @@ def join_str_constraints(constraints: List[str], operator: ConstraintGroupOperat
         (bar<=128))
     ```
     """
+    constraints = list(filter(bool, constraints))
+
     if operator == "!":
         if len(constraints) == 1:
             return f"({operator}{constraints[0]})"