def test_push_msg_app_ids(self, mock):
"""Test push_msg with array app_ids works."""
client = PybossaOneSignal(app_ids=["1", "2"], api_key="key")
fakeRequest = MagicMock()
fakeRequest.status_code = 200
fakeRequest.reason = 'OK'
fakeRequest.json.return_value = self.valid_notification
mock.return_value = fakeRequest
tmp = client.push_msg()
assert tmp[0] == 200
assert tmp[1] == 'OK'
assert tmp[2] == self.valid_notification
self.payload['app_ids'] = ["1", "2"]
mock.assert_called_with(client.api_url,
headers=client.header("key"),
json=self.payload)
def test_push_msg_include_send_after(self, mock):
"""Test push_msg with array send_after works."""
client = PybossaOneSignal(app_id="1", api_key="key")
fakeRequest = MagicMock()
fakeRequest.status_code = 200
fakeRequest.reason = 'OK'
fakeRequest.json.return_value = self.valid_notification
mock.return_value = fakeRequest
tmp = client.push_msg(send_after="Thu Sep 24 2015 14:00:00 GMT-0700 (PDT)")
assert tmp[0] == 200
assert tmp[1] == 'OK'
assert tmp[2] == self.valid_notification
self.payload['app_id'] = "1"
self.payload['send_after'] = "Thu Sep 24 2015 14:00:00 GMT-0700 (PDT)"
mock.assert_called_with(client.api_url,
headers=client.header("key"),
json=self.payload)
def test_push_msg_include_priority(self, mock):
"""Test push_msg with array priority works."""
client = PybossaOneSignal(app_id="1", api_key="key")
fakeRequest = MagicMock()
fakeRequest.status_code = 200
fakeRequest.reason = 'OK'
fakeRequest.json.return_value = self.valid_notification
mock.return_value = fakeRequest
tmp = client.push_msg(priority="10")
assert tmp[0] == 200
assert tmp[1] == 'OK'
assert tmp[2] == self.valid_notification
self.payload['app_id'] = "1"
self.payload['priority'] = "10"
mock.assert_called_with(client.api_url,
headers=client.header("key"),
json=self.payload)
def test_headers(self):
"""Test headers method."""
client = PybossaOneSignal(app_id="1", api_key="key")
headers = client.header("key")
assert headers['Content-Type'] == 'application/json; charset=utf-8'
assert headers['Authorization'] == 'Basic key'
