def test_reject_message(mock_session):
response = Mock(status_code=200, spec=Response)
mock_session.return_value.request.return_value = response
data = {"data": {}}
response.json.return_value = data
hoot_suite = HootSweet("client_id", "client_secret", token=test_token)
message_id = "1234"
reason = "Message contains profanity"
sequence = 11
reviewer_type = Reviewer.EXTERNAL
hoot_suite.reject_message(
message_id=message_id,
reason=reason,
sequence=sequence,
reviewer_type=Reviewer.EXTERNAL,
)
expected_data = {
"sequenceNumber": sequence,
"reviewerType": reviewer_type.name,
"reason": reason,
}
expected_url = "https://platform.hootsuite.com/v1/messages/%s/reject" % message_id
mock_session.return_value.request.assert_called_once_with(
"POST", expected_url, data=expected_data
)
def test_get_outbound_messages(mock_session):
expected_url = "https://platform.hootsuite.com/v1/messages"
args = (
datetime.datetime(2020, 1, 1, 12, 1, 1),
datetime.datetime(2020, 1, 2, 12, 1, 1),
)
kwargs = {
"state": MessageState.SENT,
"social_profile_ids": [123, 456],
"include_unscheduled_review_messages": True,
}
response = Mock(status_code=200, spec=Response)
mock_session.return_value.request.return_value = response
data = {"data": {}}
response.json.return_value = data
token = {"access_token": "token"}
hoot_suite = HootSweet("client_id", "client_secret", token=token)
actual = hoot_suite.get_outbound_messages(*args, **kwargs)
expected_params = {
"startTime": args[0].strftime(ISO_FORMAT),
"endTime": args[1].strftime(ISO_FORMAT),
"state": kwargs["state"].name,
"limit": 50,
"socialProfileIds": kwargs["social_profile_ids"],
"includeUnscheduledReviewMsgs": True,
}
mock_session.return_value.request.assert_called_once_with(
"GET", expected_url, params=expected_params
)
assert actual == data["data"]
def test_negative_expires_in(mock_session):
attrs = [{"status_code": c, "json.return_value": {"data": {}}} for c in [200, 200]]
responses = [Mock(spec=Response, **attr) for attr in attrs]
mock_session.return_value.request.side_effect = responses
mock_session.return_value.token = {"expires_in": -10}
mock_refresh_cb = Mock(__name__="refresh_cb")
hoot_suite = HootSweet(
"client_id", "client_secret", token=test_token, refresh_cb=mock_refresh_cb
)
hoot_suite.get_me()
mock_session.return_value.refresh_token.assert_called_once_with(
HOOTSUITE_TOKEN_URL, auth=HTTPBasicAuth("client_id", "client_secret")
)
mock_refresh_cb.assert_called_once()
def test_delete_endpoint_urls(mock_session, func, args, expected_url):
response = Mock(status_code=200, spec=Response)
mock_session.return_value.request.return_value = response
data = {"data": {}}
response.json.return_value = data
hoot_suite = HootSweet("client_id", "client_secret", token=test_token)
actual = getattr(hoot_suite, func)(*args)
mock_session.return_value.request.assert_called_once_with("DELETE", expected_url)
assert actual == data["data"]
def test_refresh_token(mock_session, mock_refresh_cb):
attrs = [
{"status_code": c, "json.return_value": {"data": {}}} for c in [401, 200, 200]
]
responses = [Mock(spec=Response, **attr) for attr in attrs]
mock_session.return_value.request.side_effect = responses
hoot_suite = HootSweet("client_id", "client_secret", token=test_token)
expected_url = "https://platform.hootsuite.com/v1/me"
hoot_suite.get_me()
calls = [call("GET", expected_url), call("GET", expected_url)]
assert mock_session.return_value.request.mock_calls == calls
mock_session.return_value.refresh_token.assert_called_once_with(
HOOTSUITE_TOKEN_URL, auth=HTTPBasicAuth("client_id", "client_secret")
)
mock_refresh_cb.assert_called_once()
def test_schedule_message(mock_session):
response = Mock(status_code=200, spec=Response)
mock_session.return_value.request.return_value = response
data = {"data": {}}
response.json.return_value = data
hoot_suite = HootSweet("client_id", "client_secret", token=test_token)
text = "An example message."
ids_ = ["1234", "12345"]
send_time = datetime.datetime(2020, 1, 1, 13, 10, 14)
hoot_suite.schedule_message(text, ids_, send_time=send_time)
data = {
"text": text,
"socialProfileIds": ids_,
"scheduledSendTime": send_time.strftime(ISO_FORMAT),
"emailNotification": False,
}
expected_url = "https://platform.hootsuite.com/v1/messages"
mock_session.return_value.request.assert_called_once_with(
"POST", expected_url, json=data
)
def test_approve_message(mock_session):
response = Mock(status_code=200, spec=Response)
mock_session.return_value.request.return_value = response
data = {"data": {}}
response.json.return_value = data
hoot_suite = HootSweet("client_id", "client_secret", token=test_token)
message_id = "1234"
sequence = 11
reviewer_type = Reviewer.EXTERNAL
hoot_suite.approve_message(
message_id=message_id, sequence_number=sequence, reviewer_type=Reviewer.EXTERNAL
)
expected_json = json.dumps(
{"sequenceNumber": sequence, "reviewerType": reviewer_type.name}
)
expected_url = "https://platform.hootsuite.com/v1/messages/%s/approve" % message_id
mock_session.return_value.request.assert_called_once_with(
"POST", expected_url, data=expected_json
)
def test_create_member_endpoint_urls(mock_session):
response = Mock(status_code=200, spec=Response)
mock_session.return_value.request.return_value = response
mock_session.return_value.token = {"expires_in": 10}
data = {"data": {}}
response.json.return_value = data
hoot_suite = HootSweet("client_id", "client_secret", token=test_token)
args = ("Joe Bloggs", "*****@*****.**", ["1234"])
with pytest.raises(InvalidLanguage):
hoot_suite.create_member(*args, language="rr")
with pytest.raises(InvalidTimezone):
hoot_suite.create_member(*args, timezone="Mars/Europa")
assert mock_session.return_value.request.call_count == 0
hoot_suite.create_member(*args)
expected_url = "https://platform.hootsuite.com/v1/members"
expected_data = {
"fullName": "Joe Bloggs",
"email": "*****@*****.**",
"organizationIds": ["1234"],
"language": "en",
"timezone": "Europe/London",
}
mock_session.return_value.request.assert_called_once_with(
"POST", expected_url, data=expected_data
)
mock_session.return_value.request.reset_mock()
hoot_suite.create_member(*args, bio="a bio", company_name="ACompany")
expected_data["bio"] = "a bio"
expected_data["companyName"] = "ACompany"
mock_session.return_value.request.assert_called_once_with(
"POST", expected_url, data=expected_data
)
def test_create_media_upload_url(mock_session):
response = Mock(status_code=200, spec=Response)
mock_session.return_value.request.return_value = response
mock_session.return_value.token = {"expires_in": 10}
data = {"data": {}}
response.json.return_value = data
hoot_suite = HootSweet("client_id", "client_secret", token=test_token)
with pytest.raises(MIMETypeNotAllowed):
hoot_suite.create_media_upload_url(5000, "image/nnn")
args = (500, "image/png")
expected_url = "https://platform.hootsuite.com/v1/media"
expected_data = {"sizeBytes": args[0], "mimeType": args[1]}
hoot_suite.create_media_upload_url(*args)
mock_session.return_value.request.assert_called_once_with(
"POST", expected_url, json=expected_data
)