com.datatorrent.demos.mobile.ApplicationTest.java Source code

Introduction

Here is the source code for com.datatorrent.demos.mobile.ApplicationTest.java

Source

/*
 * Copyright (c) 2013 DataTorrent, Inc. ALL Rights Reserved.
 *
 * 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 com.datatorrent.demos.mobile;

import java.net.URI;
import java.util.HashMap;
import java.util.Map;

import org.apache.hadoop.conf.Configuration;
import org.eclipse.jetty.server.Connector;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.servlet.ServletContextHandler;
import org.eclipse.jetty.servlet.ServletHolder;
import org.junit.Assert;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.datatorrent.api.LocalMode;

import com.datatorrent.lib.helper.SamplePubSubWebSocketServlet;
import com.datatorrent.lib.io.PubSubWebSocketInputOperator;
import com.datatorrent.lib.io.PubSubWebSocketOutputOperator;
import com.datatorrent.lib.testbench.CollectorTestSink;

public class ApplicationTest {
    private static final Logger LOG = LoggerFactory.getLogger(ApplicationTest.class);

    public ApplicationTest() {
    }

    /**
     * Test of getApplication method, of class Application.
     */
    @Test
    public void testGetApplication() throws Exception {
        Configuration conf = new Configuration(false);
        conf.addResource("dt-site-mobile.xml");
        Server server = new Server(0);
        SamplePubSubWebSocketServlet servlet = new SamplePubSubWebSocketServlet();
        ServletHolder sh = new ServletHolder(servlet);
        ServletContextHandler contextHandler = new ServletContextHandler(server, "/",
                ServletContextHandler.SESSIONS);
        contextHandler.addServlet(sh, "/pubsub");
        contextHandler.addServlet(sh, "/*");
        server.start();
        Connector connector[] = server.getConnectors();
        conf.set("dt.attr.GATEWAY_CONNECT_ADDRESS", "localhost:" + connector[0].getLocalPort());
        URI uri = URI.create("ws://localhost:" + connector[0].getLocalPort() + "/pubsub");

        PubSubWebSocketOutputOperator<Object> outputOperator = new PubSubWebSocketOutputOperator<Object>();
        outputOperator.setName("testOutputOperator");
        outputOperator.setUri(uri);
        outputOperator.setTopic(conf.get("dt.application.MobileDemo.operator.QueryLocation.topic"));

        PubSubWebSocketInputOperator<Map<String, String>> inputOperator = new PubSubWebSocketInputOperator<Map<String, String>>();
        inputOperator.setName("testInputOperator");
        inputOperator.setUri(uri);
        inputOperator.setTopic(conf.get("dt.application.MobileDemo.operator.LocationResults.topic"));

        CollectorTestSink<Object> sink = new CollectorTestSink<Object>();
        inputOperator.outputPort.setSink(sink);

        Map<String, String> data = new HashMap<String, String>();
        data.put("command", "add");
        data.put("phone", "5559990");

        Application app = new Application();
        LocalMode lma = LocalMode.newInstance();
        lma.prepareDAG(app, conf);
        LocalMode.Controller lc = lma.getController();
        lc.setHeartbeatMonitoringEnabled(false);
        lc.runAsync();
        Thread.sleep(5000);
        inputOperator.setup(null);
        outputOperator.setup(null);
        inputOperator.activate(null);
        outputOperator.beginWindow(0);
        outputOperator.input.process(data);
        outputOperator.endWindow();
        inputOperator.beginWindow(0);
        int timeoutMillis = 5000;
        while (sink.collectedTuples.size() < 5 && timeoutMillis > 0) {
            inputOperator.emitTuples();
            timeoutMillis -= 20;
            Thread.sleep(20);
        }
        inputOperator.endWindow();
        lc.shutdown();
        inputOperator.teardown();
        outputOperator.teardown();
        server.stop();
        Assert.assertTrue("size of output is 5 ", sink.collectedTuples.size() == 5);
        for (Object obj : sink.collectedTuples) {
            Assert.assertEquals("Expected phone number", "5559990", ((Map<String, String>) obj).get("phone"));
        }
    }
}