com.vaushell.superpipes.transforms.bitly.T_ShortenExpandTest.java Source code

Introduction

Here is the source code for com.vaushell.superpipes.transforms.bitly.T_ShortenExpandTest.java

Source

/*
 * Copyright (C) 2013 Fabien Vauchelles (fabien_AT_vauchelles_DOT_com).
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3, 29 June 2007, of the License, or (at your option) any later version.
 *
 * 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., 51 Franklin Street, Fifth Floor, Boston,
 * MA 02110-1301  USA
 */

package com.vaushell.superpipes.transforms.bitly;

import com.vaushell.superpipes.dispatch.ConfigProperties;
import com.vaushell.superpipes.dispatch.Dispatcher;
import com.vaushell.superpipes.dispatch.Message;
import com.vaushell.superpipes.nodes.A_Node;
import com.vaushell.superpipes.nodes.dummy.N_Dummy;
import com.vaushell.superpipes.tools.scribe.code.VC_FileFactory;
import com.vaushell.superpipes.transforms.A_Transform;
import java.net.URI;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Arrays;
import org.apache.commons.configuration.XMLConfiguration;
import static org.testng.AssertJUnit.*;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;

/**
 * Unit test.
 *
 * @see T_Shorten
 * @see T_Expand
 * @author Fabien Vauchelles (fabien_AT_vauchelles_DOT_com)
 */
public class T_ShortenExpandTest {
    // PUBLIC
    public T_ShortenExpandTest() {
        this.dispatcher = new Dispatcher();
    }

    /**
     * Initialize the test.
     *
     * @throws Exception
     */
    @BeforeClass
    public void setUp() throws Exception {
        // My config
        String conf = System.getProperty("conf");
        if (conf == null) {
            conf = "conf-local/test/configuration.xml";
        }

        String datas = System.getProperty("datas");
        if (datas == null) {
            datas = "conf-local/test/datas";
        }

        final XMLConfiguration config = new XMLConfiguration(conf);

        final Path pDatas = Paths.get(datas);
        dispatcher.init(config, pDatas, new VC_FileFactory(pDatas));

        // Test if parameters are set.
        final ConfigProperties properties = dispatcher.getCommon("bitly");

        assertTrue("Parameter 'username' should exist", properties.containsKey("username"));
        assertTrue("Parameter 'apikey' should exist", properties.containsKey("apikey"));
    }

    /**
     * Test T_Shorten and T_Expand.
     *
     * @throws java.lang.Exception
     */
    @Test
    public void testTransform() throws Exception {
        final A_Node n = dispatcher.addNode("dummy", N_Dummy.class, ConfigProperties.EMPTY_COMMONS);

        final ConfigProperties common = dispatcher.getCommon("bitly");

        final A_Transform tShorten = n.addTransformIN(T_Shorten.class, Arrays.asList(common));
        final A_Transform tExpand = n.addTransformIN(T_Expand.class, Arrays.asList(common));

        // Prepare node
        n.prepare();

        // Test : shorten
        final URI refURI = new URI("http://fabien.vauchelles.com/");
        final Message m = Message.create(Message.KeyIndex.URI, refURI);

        assertNotNull("Transform should return a message", tShorten.transform(m));

        final URI shortenURL = (URI) m.getProperty(Message.KeyIndex.URI);

        assertTrue("Shorten URL should contains bit.ly domain", shortenURL.getHost().contains("bit.ly"));

        // Test : expand
        assertNotNull("Transform should return a message", tExpand.transform(m));

        final URI expandURL = (URI) m.getProperty(Message.KeyIndex.URI);

        assertEquals("URI must be the same after a reduce/expand", refURI, expandURL);

        // Terminate node
        n.terminate();
    }

    // PRIVATE
    private final Dispatcher dispatcher;
}