Feature: parameter 'metadata' was missing from create_program

aleph-im · Mar 2, 2023 · bedffbb · bedffbb
1 parent 4fd8e7f
commit bedffbb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/aleph/sdk/client.py b/src/aleph/sdk/client.py
@@ -378,6 +378,7 @@ def create_program(
         encoding: Encoding = Encoding.zip,
         volumes: Optional[List[Mapping]] = None,
         subscriptions: Optional[List[Mapping]] = None,
+        metadata: Optional[Mapping[str, Any]] = None,
     ) -> Tuple[ProgramMessage, MessageStatus]:
         return self._wrap(
             self.async_session.create_program,
@@ -396,6 +397,7 @@ def create_program(
             encoding=encoding,
             volumes=volumes,
             subscriptions=subscriptions,
+            metadata=metadata,
         )
 
     def forget(
@@ -1171,6 +1173,7 @@ async def create_program(
         encoding: Encoding = Encoding.zip,
         volumes: Optional[List[Mapping]] = None,
         subscriptions: Optional[List[Mapping]] = None,
+        metadata: Optional[Mapping[str, Any]] = None,
     ) -> Tuple[ProgramMessage, MessageStatus]:
         """
         Post a (create) PROGRAM message.
@@ -1244,6 +1247,7 @@ async def create_program(
                 },
                 "volumes": volumes,
                 "time": time.time(),
+                "metadata": metadata,
             }
         )
 

diff --git a/tests/unit/test_asynchronous.py b/tests/unit/test_asynchronous.py
@@ -131,6 +131,7 @@ async def test_create_program(mock_session_with_post_success):
             entrypoint="main:app",
             runtime="facefacefacefacefacefacefacefacefacefacefacefacefacefacefaceface",
             channel="TEST",
+            metadata={"tags": ["test"]},
         )
 
     assert mock_session_with_post_success.http_session.post.called_once