ServletTestClient.java

/**
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF 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.apache.openejb.test.servlet;

import org.apache.openejb.test.TestClient;
import org.apache.openejb.test.TestManager;

import javax.naming.InitialContext;
import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.net.Authenticator;
import java.net.MalformedURLException;
import java.net.PasswordAuthentication;
import java.net.URL;
import java.net.URLConnection;
import java.util.Properties;

public abstract class ServletTestClient extends TestClient {
    protected URL serverUrl;
    private final String servletName;

    public ServletTestClient(final String servletName) {
        super("Servlet." + servletName + ".");
        this.servletName = servletName;
        final String serverUri = System.getProperty("openejb.server.uri", "http://127.0.0.1:8080/tomee/ejb");
        try {
            serverUrl = new URL(serverUri);
        } catch (final MalformedURLException e) {
            throw new RuntimeException(e);
        }

        // install authenticator for protected urls
        Authenticator.setDefault(new StaticAuthenticator());
    }

    private static class StaticAuthenticator extends Authenticator {
        protected PasswordAuthentication getPasswordAuthentication() {
            return new PasswordAuthentication("user", "user".toCharArray());
        }
    }

    /**
     * Sets up the fixture, for example, open a network connection.
     * This method is called before a test is executed.
     */
    protected void setUp() throws Exception {

        final Properties properties = TestManager.getServer().getContextEnvironment();
        //properties.put(Context.SECURITY_PRINCIPAL, "STATEFUL_test00_CLIENT");
        //properties.put(Context.SECURITY_CREDENTIALS, "STATEFUL_test00_CLIENT");

        initialContext = new InitialContext(properties);
    }

    protected Object invoke(final String methodName) {
        InputStream in = null;
        try {
            final URL url = new URL(serverUrl, "/itests/" + servletName + "?method=" + methodName);
            final URLConnection connection = url.openConnection();
            connection.connect();
            in = connection.getInputStream();
            String response = readAll(in);
            if (response.startsWith("FAILED")) {
                response = response.substring("FAILED".length()).trim();
                fail(response);
            }
        } catch (final Exception e) {
            fail("Received Exception " + e.getClass() + " : " + e.getMessage());
        } finally {
            if (in != null) {
                try {
                    in.close();
                } catch (final Exception ignored) {
                }
            }
        }
        return null;
    }

    private static String readAll(final InputStream in) throws IOException {
        // SwizzleStream block read methods are broken so read byte at a time
        final StringBuilder sb = new StringBuilder();
        int i = in.read();
        while (i != -1) {
            sb.append((char) i);
            i = in.read();
        }
        return sb.toString();
    }

    @SuppressWarnings({"unchecked"})
    protected <T> T newServletProxy(final Class<T> clazz) {
        final Object proxy = Proxy.newProxyInstance(getClass().getClassLoader(), new Class[]{clazz}, new ServletInvocationHandler());
        return (T) proxy;
    }

    private class ServletInvocationHandler implements InvocationHandler {
        public Object invoke(final Object proxy, final Method method, final Object[] args) throws Throwable {
            if (method.getParameterTypes().length != 0)
                throw new IllegalArgumentException("ServletProxy only supports no-argument methods: " + method);

            final String methodName = method.getName();

            return ServletTestClient.this.invoke(methodName);
        }
    }
}