Changeset View
Changeset View
Standalone View
Standalone View
swh/deposit/tests/loader/common.py
| Show All 29 Lines | def archive_get(self, archive_update_url, archive_path, log=None): | ||||
| f.write(chunk) | f.write(chunk) | ||||
| return archive_path | return archive_path | ||||
| def metadata_get(self, metadata_url, log=None): | def metadata_get(self, metadata_url, log=None): | ||||
| r = self.client.get(metadata_url) | r = self.client.get(metadata_url) | ||||
| return json.loads(r.content.decode('utf-8')) | return json.loads(r.content.decode('utf-8')) | ||||
| def status_update(self, update_status_url, status, revision_id=None): | def status_update(self, update_status_url, status, | ||||
| revision_id=None, directory_id=None): | |||||
| payload = {'status': status} | payload = {'status': status} | ||||
| if revision_id: | if revision_id: | ||||
| payload['revision_id'] = revision_id | payload['revision_id'] = revision_id | ||||
| if directory_id: | |||||
| payload['directory_id'] = directory_id | |||||
| self.client.put(update_status_url, | self.client.put(update_status_url, | ||||
| content_type='application/json', | content_type='application/json', | ||||
| data=json.dumps(payload)) | data=json.dumps(payload)) | ||||
| def check(self, check_url): | def check(self, check_url): | ||||
| r = self.client.get(check_url) | r = self.client.get(check_url) | ||||
| data = json.loads(r.content.decode('utf-8')) | data = json.loads(r.content.decode('utf-8')) | ||||
| return data['status'] | return data['status'] | ||||