def test_attachment():
user, password = "******", "p1"
resp = "data"
client = LathermailClient(user, password)
message_id = "aaaa"
attachment_index = 1
@httmock.all_requests
def messages_mock(url, request):
assert url.path == "/api/0/messages/{0}/attachments/{1}".format(message_id, attachment_index)
assert request.headers["X-Mail-Password"] == password
assert request.headers["X-Mail-Inbox"] == user
return resp
with httmock.HTTMock(messages_mock):
assert client.get_attachment(message_id, attachment_index) == resp
attachment_index = 2
assert client.get_attachment(message_id, attachment_index) == resp
