WarpExtension.java

/*
 * JBoss, Home of Professional Open Source
 * Copyright 2012, Red Hat Middleware LLC, and individual contributors
 * by the @authors tag. See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 *
 * 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.jboss.arquillian.warp;

import org.jboss.arquillian.container.test.impl.enricher.resource.URLResourceProvider;
import org.jboss.arquillian.container.test.spi.client.deployment.ProtocolArchiveProcessor;
import org.jboss.arquillian.core.spi.LoadableExtension;
import org.jboss.arquillian.test.spi.enricher.resource.ResourceProvider;
import org.jboss.arquillian.warp.client.filter.http.HttpFilterBuilder;
import org.jboss.arquillian.warp.impl.client.commandBus.CommandBusObserver;
import org.jboss.arquillian.warp.impl.client.commandBus.CommandServiceOnClient;
import org.jboss.arquillian.warp.impl.client.commandBus.RemoteSuiteLifecyclePropagation;
import org.jboss.arquillian.warp.impl.client.context.operation.OperationalContextInitializer;
import org.jboss.arquillian.warp.impl.client.deployment.DeploymentEnricher;
import org.jboss.arquillian.warp.impl.client.deployment.DeploymentValidator;
import org.jboss.arquillian.warp.impl.client.enrichment.HttpRequestEnrichmentService;
import org.jboss.arquillian.warp.impl.client.enrichment.HttpResponseDeenrichmentService;
import org.jboss.arquillian.warp.impl.client.enrichment.HttpResponseTransformationService;
import org.jboss.arquillian.warp.impl.client.execution.DefaultExecutionSynchronizer;
import org.jboss.arquillian.warp.impl.client.execution.DefaultHttpFiltersSource;
import org.jboss.arquillian.warp.impl.client.execution.DefaultHttpRequestEnrichmentService;
import org.jboss.arquillian.warp.impl.client.execution.DefaultRequestObserverChainManager;
import org.jboss.arquillian.warp.impl.client.execution.DefaultResponseDeenrichmentService;
import org.jboss.arquillian.warp.impl.client.execution.DefaultResponseTransformationService;
import org.jboss.arquillian.warp.impl.client.execution.DefaultWarpExecutor;
import org.jboss.arquillian.warp.impl.client.execution.DefaultWarpRequestSpecifier;
import org.jboss.arquillian.warp.impl.client.execution.EnrichmentObserver;
import org.jboss.arquillian.warp.impl.client.execution.ExecutionSynchronizer;
import org.jboss.arquillian.warp.impl.client.execution.WarpContext;
import org.jboss.arquillian.warp.impl.client.execution.WarpContextImpl;
import org.jboss.arquillian.warp.impl.client.execution.WarpExecutionInitializer;
import org.jboss.arquillian.warp.impl.client.execution.WarpExecutionObserver;
import org.jboss.arquillian.warp.impl.client.execution.WarpExecutor;
import org.jboss.arquillian.warp.impl.client.execution.WarpRequestSpecifier;
import org.jboss.arquillian.warp.impl.client.filter.http.DefaultHttpFilterBuilder;
import org.jboss.arquillian.warp.impl.client.proxy.DefaultProxyService;
import org.jboss.arquillian.warp.impl.client.proxy.DefaultURLMapping;
import org.jboss.arquillian.warp.impl.client.proxy.ProxyObserver;
import org.jboss.arquillian.warp.impl.client.proxy.ProxyService;
import org.jboss.arquillian.warp.impl.client.proxy.ProxyURLProvider;
import org.jboss.arquillian.warp.impl.client.proxy.ProxyUsageTracker;
import org.jboss.arquillian.warp.impl.client.proxy.URLMapping;
import org.jboss.arquillian.warp.impl.client.scope.WarpExecutionContextImpl;
import org.jboss.arquillian.warp.impl.client.verification.ResponsePayloadVerifier;
import org.jboss.arquillian.warp.impl.shared.command.CommandService;
import org.jboss.arquillian.warp.spi.observer.RequestObserverChainManager;
import org.littleshoot.proxy.HttpFiltersSource;

/**
 * <p>The Arquillian Warp extension - client-side.</p>
 * <p>
 * <p>For server-side, see {@link WarpRemoteExtension}.</p>
 *
 * @author Lukas Fryc
 */
public class WarpExtension implements LoadableExtension {

    @Override
    public void register(ExtensionBuilder builder) {

        // proxy
        builder.override(ResourceProvider.class, URLResourceProvider.class, ProxyURLProvider.class);

        // deployment enrichment
        builder.service(ProtocolArchiveProcessor.class, DeploymentEnricher.class);
        builder.observer(DeploymentValidator.class);

        // activity executor
        builder.service(WarpRequestSpecifier.class, DefaultWarpRequestSpecifier.class);
        builder.observer(WarpRuntimeInitializer.class);
        builder.observer(WarpExecutionObserver.class);
        builder.service(ExecutionSynchronizer.class, DefaultExecutionSynchronizer.class);
        builder.context(WarpExecutionContextImpl.class);
        builder.service(WarpExecutor.class, DefaultWarpExecutor.class);
        builder.observer(WarpExecutionInitializer.class);
        builder.service(URLMapping.class, DefaultURLMapping.class);
        builder.service(ProxyService.class, DefaultProxyService.class);
        builder.service(HttpFiltersSource.class, DefaultHttpFiltersSource.class);
        builder.observer(ProxyObserver.class);
        builder.observer(EnrichmentObserver.class);
        builder.service(HttpRequestEnrichmentService.class, DefaultHttpRequestEnrichmentService.class);
        builder.service(HttpResponseDeenrichmentService.class, DefaultResponseDeenrichmentService.class);
        builder.service(HttpResponseTransformationService.class, DefaultResponseTransformationService.class);
        builder.service(WarpContext.class, WarpContextImpl.class);
        builder.service(HttpFilterBuilder.class, DefaultHttpFilterBuilder.class);
        builder.service(RequestObserverChainManager.class, DefaultRequestObserverChainManager.class);
        builder.observer(ResponsePayloadVerifier.class);
        builder.observer(CommandBusObserver.class);
        builder.observer(RemoteSuiteLifecyclePropagation.class);
        builder.observer(OperationalContextInitializer.class);
        builder.observer(ProxyUsageTracker.class);
        builder.service(CommandService.class, CommandServiceOnClient.class);
    }
}