Added probation approve/reject methods from the R client.

Author: LTLA

src/pygobbler/__init__.py

@@ -37,3 +37,5 @@
 from .version_path import version_path
 from .refresh_latest import refresh_latest
 from .refresh_usage import refresh_usage
+from .approve_probation import approve_probation
+from .reject_probation import reject_probation

src/pygobbler/approve_probation.py

@@ -0,0 +1,24 @@
+from . import _utils as ut
+
+
+def approve_probation(project: str, asset: str, version: str, staging: str, url: str):
+    """
+    Approve a probational upload of a version of a project's asset.
+
+    Args:
+        project:
+            The name of the project.
+
+        asset:
+            The name of the asset of the ``project``.
+
+        version:
+            The name of the version of the ``asset`` to approve.
+
+        staging:
+            Path to the staging directory.
+
+        url:
+            URL to the Gobbler REST API.
+    """
+    ut.dump_request(staging, url, "approve_probation", { "project": project, "asset": asset, "version": version })

src/pygobbler/reject_probation.py

@@ -0,0 +1,24 @@
+from . import _utils as ut
+
+
+def reject_probation(project: str, asset: str, version: str, staging: str, url: str):
+    """
+    Reject a probational upload of a version of a project's asset.
+
+    Args:
+        project:
+            The name of the project.
+
+        asset:
+            The name of the asset of the ``project``.
+
+        version:
+            The name of the version of the ``asset`` to reject.
+
+        staging:
+            Path to the staging directory.
+
+        url:
+            URL to the Gobbler REST API.
+    """
+    ut.dump_request(staging, url, "reject_probation", { "project": project, "asset": asset, "version": version })

tests/test_probation.py

@@ -0,0 +1,34 @@
+import pygobbler as pyg
+import pytest
+import os
+
+
+@pytest.fixture(scope="module")
+def setup():
+    _, staging, registry, url = pyg.start_gobbler()
+    pyg.remove_project("test", staging=staging, url=url)
+    pyg.create_project("test", staging=staging, url=url)
+
+    src = pyg.allocate_upload_directory(staging)
+    with open(os.path.join(src, "foo"), "w") as handle:
+        handle.write("BAR")
+    pyg.upload_directory("test", "probation", "good", src, staging=staging, url=url, probation=True)
+    pyg.upload_directory("test", "probation", "bad", src, staging=staging, url=url, probation=True)
+
+
+def test_approve_probation(setup):
+    _, staging, registry, url = pyg.start_gobbler()
+    assert pyg.fetch_summary("test", "probation", "good", registry=registry, url=url)["on_probation"]
+    assert pyg.fetch_latest("test", "probation", registry=registry, url=url) is None
+
+    pyg.approve_probation("test", "probation", "good", staging=staging, url=url)
+    assert "on_probation" not in pyg.fetch_summary("test", "probation", "good", registry=registry, url=url)
+    assert pyg.fetch_latest("test", "probation", registry=registry, url=url) == "good"
+
+
+def test_reject_probation(setup):
+    _, staging, registry, url = pyg.start_gobbler()
+    assert pyg.fetch_summary("test", "probation", "bad", registry=registry, url=url)["on_probation"]
+
+    pyg.reject_probation("test", "probation", "bad", staging=staging, url=url)
+    assert "bad" not in pyg.list_versions("test", "probation", registry=registry, url=url)