org.atomserver.core.dbstore.CRUDDBSTest.java Source code

Introduction

Here is the source code for org.atomserver.core.dbstore.CRUDDBSTest.java

Source

/* Copyright (c) 2007 HomeAway, Inc.
 *  All rights reserved.  http://www.atomserver.org
 *
 * 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.atomserver.core.dbstore;

import junit.framework.Test;
import junit.framework.TestSuite;
import org.apache.abdera.model.Document;
import org.apache.abdera.model.Entry;
import org.apache.abdera.protocol.client.ClientResponse;
import org.apache.commons.lang.StringUtils;
import org.atomserver.core.filestore.FileBasedContentStorage;
import org.atomserver.testutils.client.MockRequestContext;
import org.atomserver.uri.EntryTarget;

import java.io.File;

/**
 */
public class CRUDDBSTest extends CRUDDBSTestCase {

    public static Test suite() {
        return new TestSuite(CRUDDBSTest.class);
    }

    public void setUp() throws Exception {
        super.setUp();
    }

    public void tearDown() throws Exception {
        super.tearDown();
        EntryTarget entryTarget = widgetURIHelper
                .getEntryTarget(new MockRequestContext(serviceContext, "GET", getEntryIRI().toString()), true);
        entriesDAO.obliterateEntry(entryTarget);
    }

    // --------------------
    //       tests
    //---------------------

    public void testCRUD() throws Exception {

        // run the tests up to some point
        // INSERT/SELECT/UPDATE/SELECT/DELETE
        String finalEditLink = runCRUDTest();
        String url = getSelfUriFromEditUri(finalEditLink);
        int rev = extractRevisionFromURI(finalEditLink) - 1;

        // SELECT against the just deleted entry
        ClientResponse response = clientGet(url, null, 200, true);

        Document<Entry> doc = response.getDocument();
        Entry entryOut = doc.getRoot();
        log.debug("CONTENT= " + entryOut.getContent());
        assertEquals(
                "<deletion xmlns=\"http://schemas.atomserver.org/atomserver/v1/rev0\" collection=\"acme\" id=\"12345\" locale=\"en\" workspace=\"widgets\"><property xmlns=\"http://schemas.atomserver.org/widgets/v1/rev0\" systemId=\"acme\" id=\"12345\" inNetwork=\"false\">\n"
                        + "<colors>" + "<color isDefault=\"true\">teal</color>" + "</colors>" + "<contact>"
                        + "<contactId>1638</contactId>" + "<displayName>This is an update 3</displayName>"
                        + "<hasEmail>true</hasEmail>" + "</contact>" + "</property></deletion>",
                entryOut.getContent());

        response.release();
        if (contentStorage instanceof FileBasedContentStorage) {
            File pFile = getEntryFile(rev);
            assertTrue(pFile != null && pFile.exists());
        }
    }

    public void testDeleteNonExistent() throws Exception {
        String fullURL = getServerURL() + "widgets/acme/678901234.en.xml";
        delete(fullURL, 404);
    }

    public void testMultipleDeletions() throws Exception {
        String fullURL = getServerURL() + getURLPath();
        String id = getURLPath();

        // INSERT
        String editURI = insert(id, fullURL, getFileXMLInsert(), false, false);

        // DELETE
        editURI = delete(editURI);

        // SELECT
        selectAndCheckDeletion(fullURL);

        // DELETE again
        editURI = delete(editURI);

        // SELECT
        selectAndCheckDeletion(fullURL);

        // DELETE again
        editURI = delete(editURI);

        // SELECT
        selectAndCheckDeletion(fullURL);
    }

    public void testObliterate() throws Exception {
        String fullURL = getServerURL() + getURLPath();
        String id = getURLPath();

        // INSERT
        String editURI = insert(id, fullURL, getFileXMLInsert(), false, false);

        // OBLITERATE
        editURI = delete(editURI + "?obliterate=true");

        // SELECT
        selectAndCheckObliteration(fullURL);
    }

    private void selectAndCheckDeletion(String url) {
        ClientResponse response = clientGetWithFullURL(url, 200);
        Document<Entry> doc = response.getDocument();
        Entry entry = doc.getRoot();
        String content = entry.getContent();
        assertFalse(StringUtils.isEmpty(content));
        int count = StringUtils.countMatches(content, "<deletion xmlns");
        log.debug("Count= " + count + "\nContent= \n" + content);
        assertEquals(1, count);
        response.release();
    }

    private void selectAndCheckObliteration(String url) {
        ClientResponse response = clientGetWithFullURL(url, 404);
        //redundant because clientGetWithFullURL() also does this check
        assertEquals(response.getStatus(), 404);
        response.release();
    }
}