server/src/test/java/org/apache/abdera/protocol/server/test/custom/CustomProviderTest.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  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.  For additional information regarding
 * copyright in this work, please see the NOTICE file in the top level
 * directory of this distribution.
 */
package org.apache.abdera.protocol.server.test.custom;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.util.Date;

import org.apache.abdera.Abdera;
import org.apache.abdera.model.Base;
import org.apache.abdera.model.Categories;
import org.apache.abdera.model.Collection;
import org.apache.abdera.model.Document;
import org.apache.abdera.model.Entry;
import org.apache.abdera.model.Feed;
import org.apache.abdera.model.Service;
import org.apache.abdera.model.Workspace;
import org.apache.abdera.protocol.Response.ResponseType;
import org.apache.abdera.protocol.client.AbderaClient;
import org.apache.abdera.protocol.client.ClientResponse;
import org.apache.abdera.protocol.client.RequestOptions;
import org.apache.abdera.protocol.server.test.JettyServer;
import org.apache.abdera.util.Constants;
import org.apache.abdera.util.MimeTypeHelper;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;

public class CustomProviderTest {

    private static JettyServer server;
    private static Abdera abdera = Abdera.getInstance();
    private static AbderaClient client = new AbderaClient();

    private static String BASE = "http://localhost:9002/atom";

    @BeforeClass
    public static void setUp() throws Exception {
        server = new JettyServer();
        server.start(CustomProvider.class);
    }

    @AfterClass
    public static void tearDown() throws Exception {
        server.stop();
    }

    @Test
    public void testGetService() throws IOException {
        ClientResponse resp = client.get(BASE);
        assertNotNull(resp);
        assertEquals(ResponseType.SUCCESS, resp.getType());
        assertTrue(MimeTypeHelper.isMatch(resp.getContentType().toString(), Constants.APP_MEDIA_TYPE));
        Document<Service> doc = resp.getDocument();
        try {
            prettyPrint(doc);
        } catch (Exception e) {
        }
        Service service = doc.getRoot();
        prettyPrint(service);
        assertEquals(1, service.getWorkspaces().size());
        Workspace workspace = service.getWorkspaces().get(0);
        assertEquals(1, workspace.getCollections().size());
        Collection collection = workspace.getCollections().get(0);
        assertEquals(BASE + "/feed", collection.getResolvedHref().toString());
        assertEquals("A simple feed", collection.getTitle().toString());
        resp.release();
    }

    @Test
    public void testGetCategories() {
        ClientResponse resp = client.get(BASE + "/feed;categories");
        assertNotNull(resp);
        assertEquals(ResponseType.SUCCESS, resp.getType());
        assertTrue(MimeTypeHelper.isMatch(resp.getContentType().toString(), Constants.CAT_MEDIA_TYPE));
        Document<Categories> doc = resp.getDocument();
        Categories cats = doc.getRoot();
        assertEquals(3, cats.getCategories().size());
        assertEquals("foo", cats.getCategories().get(0).getTerm());
        assertEquals("bar", cats.getCategories().get(1).getTerm());
        assertEquals("baz", cats.getCategories().get(2).getTerm());
        assertFalse(cats.isFixed());
    }

    @Test
    public void testGetFeed() throws Exception {
        ClientResponse resp = client.get(BASE + "/feed");
        assertNotNull(resp);
        assertEquals(ResponseType.SUCCESS, resp.getType());
        assertTrue(MimeTypeHelper.isMatch(resp.getContentType().toString(), Constants.ATOM_MEDIA_TYPE));
        Document<Feed> doc = resp.getDocument();
        Feed feed = doc.getRoot();
        assertEquals("tag:example.org,2008:feed", feed.getId().toString());
        assertEquals("A simple feed", feed.getTitle());
        assertEquals("Simple McGee", feed.getAuthor().getName());
        assertEquals(0, feed.getEntries().size());
        resp.release();
    }

    protected void prettyPrint(Base doc) throws IOException {
        // WriterFactory writerFactory = abdera.getWriterFactory();
        // Writer writer = writerFactory.getWriter("prettyxml");
        // writer.writeTo(doc, System.out);
        // System.out.println();
    }

    @Test
    public void testPostEntry() {
        Entry entry = abdera.newEntry();
        entry.setId(BASE + "/feed/entries/1");
        entry.setTitle("test entry");
        entry.setContent("Test Content");
        entry.addLink("http://example.org");
        entry.setUpdated(new Date());
        entry.addAuthor("James");
        ClientResponse resp = client.post(BASE + "/feed", entry);
        assertNotNull(resp);
        assertEquals(ResponseType.SUCCESS, resp.getType());
        assertEquals(201, resp.getStatus());
        assertNotNull(resp.getLocation());
        resp.release();
        resp = client.get(BASE + "/feed");
        Document<Feed> feed_doc = resp.getDocument();
        Feed feed = feed_doc.getRoot();
        assertEquals(1, feed.getEntries().size());
    }

    @Test
    public void testPostMedia() {
        ByteArrayInputStream in = new ByteArrayInputStream(new byte[] {0x01, 0x02, 0x03, 0x04});
        RequestOptions options = client.getDefaultRequestOptions();
        options.setContentType("application/octet-stream");
        ClientResponse resp = client.post(BASE + "/feed", in, options);
        assertEquals(ResponseType.CLIENT_ERROR, resp.getType());
        assertEquals(405, resp.getStatus());
        resp.release();
    }

    @Test
    public void testPutEntry() throws IOException {
        ClientResponse resp = client.get(BASE + "/feed");
        Document<Feed> feed_doc = resp.getDocument();
        Feed feed = feed_doc.getRoot();
        prettyPrint(feed);
        Entry entry = feed.getEntries().get(0);
        String edit = entry.getEditLinkResolvedHref().toString();
        resp.release();
        resp = client.get(edit);
        Document<Entry> doc = resp.getDocument();
        doc.writeTo(System.out);
        prettyPrint(doc.getRoot());
        entry = doc.getRoot();
        entry.setTitle("This is the modified title");
        resp.release();
        resp = client.put(edit, entry);
        assertEquals(ResponseType.SUCCESS, resp.getType());
        assertEquals(204, resp.getStatus());
        resp.release();
        resp = client.get(edit);
        doc = resp.getDocument();
        entry = doc.getRoot();
        assertEquals("This is the modified title", entry.getTitle());
        resp.release();
        resp = client.get(BASE + "/feed");
        feed_doc = resp.getDocument();
        feed = feed_doc.getRoot();
        assertEquals(1, feed.getEntries().size());
        resp.release();
    }

    @Test
    public void testDeleteEntry() {
        ClientResponse resp = client.get(BASE + "/feed");
        Document<Feed> feed_doc = resp.getDocument();
        Feed feed = feed_doc.getRoot();
        Entry entry = feed.getEntries().get(0);
        String edit = entry.getEditLinkResolvedHref().toString();
        resp.release();
        resp = client.delete(edit);
        assertEquals(ResponseType.SUCCESS, resp.getType());
        resp.release();
        resp = client.get(BASE + "/feed");
        feed_doc = resp.getDocument();
        feed = feed_doc.getRoot();
        assertEquals(0, feed.getEntries().size());
        resp.release();
    }
}