org.osaf.cosmo.atom.provider.mock.BaseMockRequestContext.java Source code

Introduction

Here is the source code for org.osaf.cosmo.atom.provider.mock.BaseMockRequestContext.java

Source

/*
 * Copyright 2007 Open Source Applications Foundation
 * 
 * Licensed 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.osaf.cosmo.atom.provider.mock;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Properties;

import org.apache.abdera.Abdera;
import org.apache.abdera.model.Entry;
import org.apache.abdera.protocol.server.Provider;
import org.apache.abdera.protocol.server.RequestContext;
import org.apache.abdera.protocol.server.ServiceManager;
import org.apache.abdera.protocol.server.servlet.ServletRequestContext;
import org.apache.abdera.util.Constants;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.osaf.cosmo.atom.AtomConstants;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.mock.web.MockServletContext;

/**
 * Mock implementation of {@link RequestContext}.
 */
public class BaseMockRequestContext extends ServletRequestContext implements AtomConstants, Constants {

    private static final Log log = LogFactory.getLog(BaseMockRequestContext.class);

    private HashMap<String, String> params = new HashMap<String, String>();

    public BaseMockRequestContext(Provider provider, String method, String uri) {
        super(provider, createRequest(method, uri));
    }

    private static MockHttpServletRequest createRequest(String method, String uri) {
        MockServletContext ctx = new MockServletContext();
        return new MockHttpServletRequest(ctx, method, uri);
    }

    public MockHttpServletRequest getMockRequest() {
        return (MockHttpServletRequest) getRequest();
    }

    public void setParameter(String name, String value) {
        params.put(name, value);
    }

    @Override
    public String getParameter(String name) {
        String val = params.get(name);
        if (val != null)
            return val;
        else
            return super.getParameter(name);
    }

    @Override
    public String[] getParameterNames() {
        HashSet<String> names = new HashSet<String>();
        names.addAll(params.keySet());
        for (String name : super.getParameterNames())
            names.add(name);

        return names.toArray(new String[0]);
    }

    @Override
    public List<String> getParameters(String name) {
        // TODO override, but for now ignore
        return super.getParameters(name);
    }

    public void setContent(String content, String mediaType) throws IOException {
        if (content != null) {
            byte[] bytes = content.getBytes("UTF-8");
            getMockRequest().setContent(bytes);
            getMockRequest().addHeader("Content-Length", bytes.length);
        }
        getMockRequest().setContentType(mediaType);
        getMockRequest().addHeader("Content-Type", mediaType);
    }

    public void setContent(Entry entry) throws IOException {
        String xml = getAbdera().getWriter().write(entry).toString();
        setContent(xml, ATOM_MEDIA_TYPE);
    }

    public void setContentAsText(String content) throws IOException {
        setContent(content, "text/plain");
    }

    public void setContentAsEntry(String content) throws IOException {
        Entry entry = getAbdera().getFactory().newEntry();
        entry.setContent(content);
        setContent(entry);
    }

    public void setContentAsXhtml(String content) throws IOException {
        setContent(content, "application/xhtml+xml");
    }

    public void setContentAsFormEncoded(String content) throws IOException {
        setContent(content, "application/x-www-form-urlencoded");
    }

    public void setContentAsCalendar(String content) throws IOException {
        setContent(content, "text/calendar");
    }

    public void setXhtmlContentAsEntry(String content) throws IOException {
        Entry entry = getAbdera().getFactory().newEntry();
        entry.setContentAsXhtml(content);
        setContent(entry);
    }

    public void setProperties(Properties props, String mediaType) throws IOException {
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        props.store(out, null);
        setContent(out.toString(), mediaType);
    }

    public void setPropertiesAsText(Properties props) throws IOException {
        setProperties(props, "text/plain");
    }

    public void setPropertiesAsEntry(Properties props) throws IOException {
        Entry entry = getAbdera().getFactory().newEntry();
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        props.store(out, null);
        entry.setContent(out.toString());
        setContent(entry);
    }

    public Abdera getAbdera() {
        return ServiceManager.getAbdera();
    }
}