com.agapsys.agreste.test.ServletContainerBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of agreste-test Show documentation
Testing utilities for AGRESTE-based applications
There is a newer version: 6.0.0
/*
 * Copyright 2016 Agapsys Tecnologia Ltda-ME.
 *
 * 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 com.agapsys.agreste.test;

import com.agapsys.agreste.AbuseCheckFilter;
import com.agapsys.agreste.ClientExceptionFilter;
import com.agapsys.agreste.JpaTransactionFilter;
import com.agapsys.rcf.Controller;
import com.agapsys.rcf.WebController;
import com.agapsys.sevlet.container.ServletContainer;
import com.agapsys.web.toolkit.AbstractWebApplication;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;

/**
 * Servlet container builder for AGRESTE applications
 * @author Leandro Oliveira ([email protected])
 */
public class ServletContainerBuilder extends com.agapsys.web.toolkit.ServletContainerBuilder {
	// STATIC SCOPE ============================================================
	public static ServletContainer getControllerContainer(Class...controllers) {
		return getControllerContainer(MockedWebApplication.class, controllers);
	}

	public static ServletContainer getControllerContainer(Class webApp, Class...controllers) {
		ServletContainerBuilder builder = new ServletContainerBuilder(webApp);
		for (Class controller : controllers) {
			builder.registerController(controller);
		}
		return builder.build();
	}

	private static final String EMBEDDED_CONTROLER_INFO = "/META-INF/rcf.info";
	private static final String EMBEDDED_CONTROLER_INFO_ENCODING = "utf-8";
	// =========================================================================

	// INSTANCE SCOPE ==========================================================
	private void registerScannedControllers() {
		InputStream is = null;
		try {

			is = ServletContainerBuilder.class.getResourceAsStream(EMBEDDED_CONTROLER_INFO);

			if (is == null)
				return;

			BufferedReader in = new BufferedReader(new InputStreamReader(is, EMBEDDED_CONTROLER_INFO_ENCODING));

			String readLine;

			while ((readLine = in.readLine()) != null) {
				readLine = readLine.trim();

				if (readLine.isEmpty() || readLine.startsWith("#"))
					continue;

				registerController((Class) Class.forName(readLine));
			}

			in.close();

		} catch (IOException | ClassNotFoundException ex) {
			throw new RuntimeException(ex);
		} finally {
			if (is != null)
				try {
					is.close();
			} catch (IOException ex) {
				throw new RuntimeException(ex);
			}
		}
	}

	private void init() {
		super.registerFilter(AbuseCheckFilter.class, "/*");
		super.registerFilter(ClientExceptionFilter.class, "/*");
		super.registerFilter(JpaTransactionFilter.class, "/*");

		registerScannedControllers();
	}

	public ServletContainerBuilder(Class webApp) {
		super(webApp);
		init();
	}

	public ServletContainerBuilder() {
		this(MockedWebApplication.class);
	}

	public ServletContainerBuilder registerController(Class controllerClass, String name) {
		return (ServletContainerBuilder) super.registerServlet(controllerClass, String.format("/%s/*", name));
	}

	public ServletContainerBuilder registerController(Class controllerClass) {
		WebController annotation = controllerClass.getAnnotation(WebController.class);

		if (annotation == null)
			throw new IllegalArgumentException(String.format("Missing annotation '%s' for '%s'", WebController.class.getName(), controllerClass.getName()));

		String name = annotation.value();
		if (name.trim().isEmpty())
			name = controllerClass.getSimpleName();

		registerController(controllerClass, name);

		return this;
	}
	// =========================================================================
}