se.vgregion.pubsub.push.impl.DefaultPushSubscriberPublishTest.java Source code

Introduction

Here is the source code for se.vgregion.pubsub.push.impl.DefaultPushSubscriberPublishTest.java

Source

/**
 * Copyright 2010 Vstra Gtalandsregionen
 *
 *   This library is free software; you can redistribute it and/or modify
 *   it under the terms of version 2.1 of the GNU Lesser General Public
 *   License as published by the Free Software Foundation.
 *
 *   This library is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU Lesser General Public License for more details.
 *
 *   You should have received a copy of the GNU Lesser General Public
 *   License along with this library; if not, write to the
 *   Free Software Foundation, Inc., 59 Temple Place, Suite 330,
 *   Boston, MA 02111-1307  USA
 *
 */

package se.vgregion.pubsub.push.impl;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.TimeUnit;

import nu.xom.Builder;
import nu.xom.Document;

import org.apache.http.HttpEntity;
import org.apache.http.HttpEntityEnclosingRequest;
import org.apache.http.HttpException;
import org.apache.http.HttpRequest;
import org.apache.http.HttpResponse;
import org.apache.http.localserver.LocalTestServer;
import org.apache.http.protocol.HttpContext;
import org.apache.http.protocol.HttpRequestHandler;
import org.joda.time.DateTime;
import org.joda.time.DateTimeUtils;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

import se.vgregion.pubsub.ContentType;
import se.vgregion.pubsub.Feed;
import se.vgregion.pubsub.Namespaces;
import se.vgregion.pubsub.impl.DefaultEntry.EntryBuilder;
import se.vgregion.pubsub.impl.DefaultFeed.FeedBuilder;
import se.vgregion.pubsub.push.UnitTestConstants;

public class DefaultPushSubscriberPublishTest {

    private DefaultPushSubscriber subscriber;
    private LocalTestServer server = new LocalTestServer(null, null);

    @Before
    public void before() throws Exception {
        DateTimeUtils.setCurrentMillisFixed(new DateTime().getMillis());
        server.start();
    }

    private URI buildTestUrl(String path) throws URISyntaxException {
        return new URI("http://" + server.getServiceHostName() + ":" + server.getServicePort() + path);
    }

    @Test
    public void publish() throws Exception {

        subscriber = new DefaultPushSubscriber(UnitTestConstants.TOPIC, buildTestUrl("/"), UnitTestConstants.FUTURE,
                UnitTestConstants.UPDATED1, 100, "verify", UnitTestConstants.SECRET, true);

        final LinkedBlockingQueue<HttpRequest> issuedRequests = new LinkedBlockingQueue<HttpRequest>();
        final LinkedBlockingQueue<byte[]> issuedRequestBodies = new LinkedBlockingQueue<byte[]>();
        server.register("/*", new HttpRequestHandler() {
            @Override
            public void handle(HttpRequest request, HttpResponse response, HttpContext context)
                    throws HttpException, IOException {
                issuedRequests.add(request);

                HttpEntity entity = ((HttpEntityEnclosingRequest) request).getEntity();
                ByteArrayOutputStream buffer = new ByteArrayOutputStream();
                entity.writeTo(buffer);
                issuedRequestBodies.add(buffer.toByteArray());
            }
        });

        Feed feed = new FeedBuilder(ContentType.ATOM).id("e1")
                .entry(new EntryBuilder().id("f1").updated(new DateTime()).build())
                .entry(new EntryBuilder().id("f2").updated(UnitTestConstants.UPDATED1.minusHours(1)).build())
                .build();

        subscriber.publish(feed, null);

        // subscriber should be updated
        Assert.assertEquals(new DateTime(), subscriber.getLastUpdated());

        HttpRequest request = issuedRequests.poll(10000, TimeUnit.MILLISECONDS);
        Assert.assertNotNull(request);
        Assert.assertEquals(ContentType.ATOM.toString(), request.getFirstHeader("Content-Type").getValue());

        // verify HMAC header
        Assert.assertEquals("sha1=1356b52665408a17af46803a7988e48d40d1fb75",
                request.getFirstHeader("X-Hub-Signature").getValue());

        // verify content
        Assert.assertTrue(request instanceof HttpEntityEnclosingRequest);

        HttpEntity entity = ((HttpEntityEnclosingRequest) request).getEntity();

        Assert.assertNotNull(entity);

        Document actualAtom = new Builder().build(new ByteArrayInputStream(issuedRequestBodies.poll()));

        Assert.assertEquals(1, actualAtom.getRootElement().getChildElements("entry", Namespaces.ATOM).size());

    }

    @After
    public void after() {
        DateTimeUtils.setCurrentMillisSystem();

        try {
            server.stop();
        } catch (Exception e) {
            // ignore
        }
    }
}