コード例 #1
0
def test_single_message():
    user, password = "******", "p1"
    resp = {"message_info": {"a": "b"}}
    client = LathermailClient(user, password)
    message_id = "aaaa"
    last_request = [None]

    @httmock.all_requests
    def messages_mock(url, request):
        assert url.path == "/api/0/messages/{0}".format(message_id)
        assert request.headers["X-Mail-Password"] == password
        assert request.headers["X-Mail-Inbox"] == user
        last_request[0] = request
        return json_response(resp)


    with httmock.HTTMock(messages_mock):
        assert client.get_single_message(message_id) == resp["message_info"]
        assert last_request[0].method == "GET"

        message_id = "bbb"
        client.delete_single_message(message_id)
        assert last_request[0].method == "DELETE"