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