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'