client + tests

Author: barthalion

flat-manager-client

@@ -1096,6 +1096,11 @@ if __name__ == "__main__":
     purge_parser.add_argument("build_url", help="remote build url")
     purge_parser.set_defaults(func=purge_command)
 
+    prune_parser = subparsers.add_parser("prune", help="Prune repository")
+    prune_parser.add_argument("manager_url", help="remote repo manager url")
+    prune_parser.add_argument("repo", help="repo name")
+    prune_parser.set_defaults(func=prune_command)
+
     create_token_parser = subparsers.add_parser(
         "create-token", help="Create subset token"
     )
@@ -1199,3 +1204,17 @@ if __name__ == "__main__":
             f.write("\n")
             f.close()
     exit(res)
+
+
+async def prune_command(args, session):
+    resp = await session.post(
+        urljoin(args.manager_url, "api/v1/prune"),
+        headers={"Authorization": "Bearer " + args.token},
+        json={"repo": args.repo},
+    )
+    async with resp:
+        if resp.status >= 500:
+            raise ServerApiError(resp, await resp.text())
+        elif resp.status != 200:
+            raise ApiError(resp, await resp.text())
+        return await resp.json()

tests/run-test.py

@@ -4,6 +4,7 @@
 import subprocess
 import sys
 from subprocess import PIPE
+import time
 
 
 def exec(cmd):
@@ -62,3 +63,10 @@ def exec(cmd):
 )
 exec(["flatpak", "update", "-y"])
 exec(["flatpak", "install", "-y", "flat-manager", "org.flatpak.FlatManagerCI"])
+
+# Test prune functionality
+print("Testing prune functionality")
+exec(["./flat-manager-client", "prune", "http://127.0.0.1:8080", "stable"])
+
+# Wait a bit to ensure the prune job completes
+time.sleep(5)