Skip to content
Snippets Groups Projects
Commit ad905ce4 authored by Nico's avatar Nico
Browse files

Fix lint and tests

parent 6abae859
No related merge requests found
Pipeline #3959 passed with warnings
...@@ -19,12 +19,12 @@ class PrivateChat(Model): ...@@ -19,12 +19,12 @@ class PrivateChat(Model):
_on_message_received_callback = None _on_message_received_callback = None
def __init__(self, api, contact_id): def __init__(self, api, contact_id):
self._api = api super().__init__(api)
self._initialize_headers()
self._contact_id = contact_id self._contact_id = contact_id
def get(self): def get(self):
url = urljoin(BASE_HTTP_URL, self.API_ENDPOINT + "/%d" % self._contact_id) url = urljoin(BASE_HTTP_URL,
self.API_ENDPOINT + "/%d" % self._contact_id)
request = _get(url, headers=self._headers) request = _get(url, headers=self._headers)
return request.json() return request.json()
...@@ -37,5 +37,6 @@ class PrivateChat(Model): ...@@ -37,5 +37,6 @@ class PrivateChat(Model):
self._on_message_received_callback(event['data']) self._on_message_received_callback(event['data'])
def send(self, message): def send(self, message):
url = urljoin(BASE_HTTP_URL, self.API_ENDPOINT + "/%i" % self._contact_id) url = urljoin(BASE_HTTP_URL,
self.API_ENDPOINT + "/%i" % self._contact_id)
_post(url, headers=self._headers, json={"text": message}) _post(url, headers=self._headers, json={"text": message})
...@@ -16,23 +16,23 @@ TEST_TEXT = "Hello World" ...@@ -16,23 +16,23 @@ TEST_TEXT = "Hello World"
@requests_mock.Mocker(kw="requests_mock") @requests_mock.Mocker(kw="requests_mock")
def test_get_empty(api, request_headers, requests_mock): def test_get_empty(api, request_headers, requests_mock):
private_chat = PrivateChat(api) private_chat = PrivateChat(api, TEST_CONTACT_ID)
url = BASE_HTTP_URL % TEST_CONTACT_ID url = BASE_HTTP_URL % TEST_CONTACT_ID
response = [] response = []
requests_mock.register_uri("GET", url, request_headers=request_headers, requests_mock.register_uri("GET", url, request_headers=request_headers,
text=json.dumps(response)) text=json.dumps(response))
assert private_chat.get(TEST_CONTACT_ID) == response assert private_chat.get() == response
@requests_mock.Mocker(kw="requests_mock") @requests_mock.Mocker(kw="requests_mock")
def test_send_message(api, request_headers, requests_mock): def test_send_message(api, request_headers, requests_mock):
private_chat = PrivateChat(api) private_chat = PrivateChat(api, TEST_CONTACT_ID)
url = BASE_HTTP_URL % TEST_CONTACT_ID url = BASE_HTTP_URL % TEST_CONTACT_ID
requests_mock.register_uri("POST", url, request_headers=request_headers, requests_mock.register_uri("POST", url, request_headers=request_headers,
additional_matcher=match_request_send_message) additional_matcher=match_request_send_message)
private_chat.send(TEST_CONTACT_ID, TEST_TEXT) private_chat.send(TEST_TEXT)
def match_request_send_message(request): def match_request_send_message(request):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment