vincent/aptly
1
0
Fork 0
mirror of https://github.com/aptly-dev/aptly.git synced 2026-01-12 03:21:33 +00:00
Code Issues Projects Releases Wiki Activity

Fix system tests

Browse Source
This commit is contained in:
Lorenzo Bolla
2021-09-21 18:52:43 +02:00
parent ff51c46915
commit 787cc8e3ee
51 changed files with 607 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
system/api_lib.py
View File

@@ -47,7 +47,14 @@ class APITest(BaseTest):
kwargs["headers"]["Content-Type"] = "application/json"
return requests.post("http://%s%s" % (self.base_url, uri), *args, **kwargs)
def _ensure_async(self, kwargs):
# Make sure we run this as an async task
params = kwargs.get('params', {})
params.setdefault('_async', True)
kwargs['params'] = params
def post_task(self, uri, *args, **kwargs):
self._ensure_async(kwargs)
resp = self.post(uri, *args, **kwargs)
if resp.status_code != 202:
return resp
@@ -67,6 +74,7 @@ class APITest(BaseTest):
return requests.put("http://%s%s" % (self.base_url, uri), *args, **kwargs)
def put_task(self, uri, *args, **kwargs):
self._ensure_async(kwargs)
resp = self.put(uri, *args, **kwargs)
if resp.status_code != 202:
return resp
@@ -86,6 +94,7 @@ class APITest(BaseTest):
return requests.delete("http://%s%s" % (self.base_url, uri), *args, **kwargs)
def delete_task(self, uri, *args, **kwargs):
self._ensure_async(kwargs)
resp = self.delete(uri, *args, **kwargs)
if resp.status_code != 202:
return resp