diff --git a/briar_wrapper/models/contacts.py b/briar_wrapper/models/contacts.py index 81d6304621f135ce78a1a8904b2024901f67bc7d..5ae6a6d7740e5c39d75eba72af9e28d20a46b30c 100644 --- a/briar_wrapper/models/contacts.py +++ b/briar_wrapper/models/contacts.py @@ -5,6 +5,7 @@ from operator import itemgetter from urllib.parse import urljoin +from requests import delete as _delete from requests import get as _get from requests import post as _post @@ -23,6 +24,10 @@ class Contacts(Model): url = urljoin(BASE_HTTP_URL, self.API_ENDPOINT + "add/pending/") _post(url, headers=self._headers, json={"link": link, "alias": alias}) + def delete(self, contact_id): + url = urljoin(BASE_HTTP_URL, self.API_ENDPOINT + str(contact_id)) + _delete(url, headers=self._headers) + def get(self): url = urljoin(BASE_HTTP_URL, self.API_ENDPOINT) request = _get(url, headers=self._headers) diff --git a/tests/briar_wrapper/models/test_contacts.py b/tests/briar_wrapper/models/test_contacts.py index 3af6d2e7e91231090a60dfa911cc103c833da468..caaa2478b500450412f469a5bebaf9c13f06966a 100644 --- a/tests/briar_wrapper/models/test_contacts.py +++ b/tests/briar_wrapper/models/test_contacts.py @@ -48,6 +48,16 @@ def match_request_add_pending(request): return {"alias": TEST_ALIAS, "link": TEST_LINK} == request.json() +@requests_mock.Mocker(kw="requests_mock") +def test_delete(api, request_headers, requests_mock): + contacts = Contacts(api) + contact_id = 137 + requests_mock.register_uri("DELETE", + BASE_HTTP_URL + str(contact_id), + request_headers=request_headers) + contacts.delete(contact_id) + + @requests_mock.Mocker(kw='requests_mock') def test_get(api, request_headers, requests_mock): contacts = Contacts(api)