TestPirasoRequest.java

/*
 * Copyright (c) 2012. Piraso Alvin R. de Leon. All Rights Reserved.
 *
 * See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The Piraso licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.piraso.server;

import org.piraso.api.JacksonUtils;
import org.piraso.api.Preferences;
import org.apache.commons.lang.Validate;
import org.codehaus.jackson.map.ObjectMapper;
import org.springframework.mock.web.MockHttpServletRequest;

import java.io.IOException;

import static org.piraso.api.PirasoConstants.*;

/**
 * Test piraso request.
 */
public class TestPirasoRequest implements PirasoEntryPoint, PirasoRequest {

    private MockHttpServletRequest request;

    public TestPirasoRequest(MockHttpServletRequest request) {
        this.request = request;
    }

    public String getPath() {
        return request.getRequestURI();
    }

    public Preferences getPreferences() {
        Validate.notNull(request.getParameter(PREFERENCES_PARAMETER), "");

        ObjectMapper mapper = JacksonUtils.createMapper();

        try {
            return mapper.readValue(request.getParameter(PREFERENCES_PARAMETER), Preferences.class);
        } catch (IOException e) {
            throw new IllegalArgumentException(e.getMessage(), e);
        }
    }

    public String getRemoteAddr() {
        return request.getRemoteAddr();
    }

    public String getWatchedAddr() {
        if(request.getParameter(WATCHED_ADDR_PARAMETER) != null) {
            return request.getParameter(WATCHED_ADDR_PARAMETER);
        } else {
            return request.getRemoteAddr();
        }
    }

    public String getActivityUuid() {
        return request.getParameter(ACTIVITY_PARAMETER);
    }

    public MockHttpServletRequest getMockRequest() {
        return request;
    }
}