test fixups

Signed-off-by: Paul Spooren <[email protected]>
openwrt · Apr 16, 2023 · 55aca0a · 55aca0a
1 parent 0e7b592
commit 55aca0a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/asu/api.py b/asu/api.py
@@ -109,7 +109,9 @@ def validate_request(req):
         )
 
     if req.get("packages_versions") and not req.get("packages"):
-        req["packages"] = req["packages_versions"].keys()
+        req["packages"] = set(req["packages_versions"].keys())
+    else:
+        req["packages"] = set(req.get("packages", []))
 
     r = get_redis()
 

diff --git a/asu/common.py b/asu/common.py
@@ -197,7 +197,7 @@ def get_container_version_tag(version: str) -> str:
     return version
 
 
-def diff_packages(requested_packages: list, default_packages: set):
+def diff_packages(requested_packages: set, default_packages: set):
     """Return a list of packages to install and remove
 
     Args:
@@ -206,7 +206,6 @@ def diff_packages(requested_packages: list, default_packages: set):
 
     Returns:
         set: Set of packages to install and remove"""
-    requested_packages = set(requested_packages)
     remove_packages = default_packages - requested_packages
     return requested_packages | set(
         map(lambda p: f"-{p}".replace("--", "-"), remove_packages)