gistfile1.java

package org.richfaces.test.application.push;

import static org.junit.Assert.assertTrue;

import java.net.URL;

import javax.servlet.http.HttpServletRequest;

import org.jboss.arquillian.container.test.api.Deployment;
import org.jboss.arquillian.container.test.api.RunAsClient;
import org.jboss.arquillian.drone.api.annotation.Drone;
import org.jboss.arquillian.junit.Arquillian;
import org.jboss.arquillian.test.api.ArquillianResource;
import org.jboss.arquillian.warp.ClientAction;
import org.jboss.arquillian.warp.HttpRequest;
import org.jboss.arquillian.warp.RequestFilter;
import org.jboss.arquillian.warp.ServerAssertion;
import org.jboss.arquillian.warp.Warp;
import org.jboss.arquillian.warp.WarpTest;
import org.jboss.arquillian.warp.extension.servlet.AfterServlet;
import org.jboss.arquillian.warp.extension.servlet.BeforeServlet;
import org.jboss.shrinkwrap.api.spec.WebArchive;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.openqa.selenium.WebDriver;
import org.richfaces.application.push.MessageException;
import org.richfaces.application.push.PushContext;
import org.richfaces.application.push.PushContextFactory;
import org.richfaces.application.push.Session;
import org.richfaces.application.push.TopicKey;
import org.richfaces.application.push.TopicsContext;
import org.richfaces.shrinkwrap.descriptor.FaceletAsset;
import org.richfaces.test.deployment.CoreDeployment;
import org.richfaces.webapp.PushHandlerFilter;

@RunWith(Arquillian.class)
@WarpTest
public class PushTest {
    
    @Drone
    WebDriver driver;

    @ArquillianResource
    URL contextPath;

    @Deployment
    public static WebArchive createDeployment() {

        CoreDeployment deployment = new CoreDeployment(PushTest.class);
        
        
        deployment.addMavenDependency(
                "org.richfaces.ui.common:richfaces-ui-common-api",
                "org.richfaces.ui.common:richfaces-ui-common-ui",
                "org.richfaces.ui.core:richfaces-ui-core-api",
                "org.richfaces.ui.core:richfaces-ui-core-ui");
       
        FaceletAsset pushPage = new FaceletAsset().body("<a4j:push address=\"" + Commons.TOPIC + "\" />");
        
        deployment.archive()
                /** ROOT */
                .addAsWebResource(pushPage, "index.xhtml")
                .addAsResource("META-INF/resources/richfaces-event.js")
                .addAsResource("META-INF/resources/jquery.js")
                .addAsResource("META-INF/resources/richfaces.js")
                .addAsResource("META-INF/resources/richfaces-queue.js")
                .addAsResource("META-INF/resources/richfaces-base-component.js");

        deployment.withResourceHandler();
        deployment.withDependencyInjector();
        deployment.withResourceCodec();
        deployment.withResourceLibraries();
        deployment.withPush();

        return deployment.getFinalArchive();
    }

    @Test
    @RunAsClient
    public void test_push() {
        Warp.filter(new UriRequestFilter("__richfaces_push")).execute(new ClientAction() {
            
            @Override
            public void action() {
                driver.navigate().to(contextPath);
            }
        }).verify(new PushServletAssertion());
    }
    
    // TODO should be part of Phaser 
    public static class UriRequestFilter implements RequestFilter<HttpRequest> {

        private String uriPart;

        public UriRequestFilter(String uriPart) {
            this.uriPart = uriPart;
        }

        @Override
        public boolean matches(HttpRequest httpRequest) {
            String uri = httpRequest.getUri();
            return uri.contains(uriPart);
        }
    }
    
    public static class PushServletAssertion extends ServerAssertion {

        private static final long serialVersionUID = 1L;

        @ArquillianResource
        HttpServletRequest httpReq;
        
        @ArquillianResource
        PushContextFactory pushContextFactory;
        
        @BeforeServlet
        public void beforeServlet() throws Exception {
            // TODO should be invokable by separate session 
            sendMessage("foo");
        }
        
        @AfterServlet
        public void afterServlet() {
            Session session = getCurrentSession();
            assertTrue(session.getMessages().isEmpty());
        }
        
        private Session getCurrentSession() {
            PushContext pushContext = pushContextFactory.getPushContext();
            String pushSessionId = httpReq.getParameter(PushHandlerFilter.PUSH_SESSION_ID_PARAM);
            Session session = pushContext.getSessionManager().getPushSession(pushSessionId);
            return session;
        }
        
        private void sendMessage(String message) throws MessageException {
            TopicsContext topicsContext = TopicsContext.lookup();
            // TODO move name of the topic to Common static class
            TopicKey topicKey = new TopicKey(Commons.TOPIC);
            topicsContext.getOrCreateTopic(topicKey);
            topicsContext.publish(topicKey, message);
        }
    }
    
    public static class Commons {
        public static final String TOPIC = "testingTopic"; 
    }
}