io.viewserver.server.steps.ViewServerSteps.java Source code

Introduction

Here is the source code for io.viewserver.server.steps.ViewServerSteps.java

Source

/*
 * Copyright 2016 Claymore Minds Limited and Niche Solutions (UK) Limited
 *
 * 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 io.viewserver.server.steps;

import io.viewserver.adapters.csv.CsvDataAdapter;
import io.viewserver.core.JacksonSerialiser;
import io.viewserver.datasource.DataSource;
import io.viewserver.network.netty.inproc.NettyInProcEndpoint;
import io.viewserver.reactor.EventLoopReactor;
import io.viewserver.report.ReportDefinition;
import io.viewserver.server.ViewServerMasterTest;
import io.viewserver.server.ViewServerSlave;
import io.viewserver.server.setup.BootstrapperBase;
import com.google.common.base.Charsets;
import com.google.common.io.Resources;
import cucumber.api.java.After;
import cucumber.api.java.en.And;
import cucumber.api.java.en.Given;

import java.io.IOException;

/**
 * Created by nick on 10/02/2015.
 */
public class ViewServerSteps {
    private InProcessViewServerContext viewServerContext;

    public ViewServerSteps(InProcessViewServerContext viewServerContext) {
        this.viewServerContext = viewServerContext;
    }

    @After
    public void afterScenario() {
        for (ViewServerSlave slave : viewServerContext.slaves) {
            EventLoopReactor reactor = slave.getServerReactor();
            reactor.shutDown();
            reactor.waitForShutdown();
        }
        viewServerContext.slaves.clear();

        if (viewServerContext.master != null) {
            EventLoopReactor reactor = viewServerContext.master.getServerReactor();
            reactor.shutDown();
            reactor.waitForShutdown();
            viewServerContext.master = null;
        }
    }

    @Given("^an in-process viewserver with ([0-9]+) slave nodes$")
    public void an_in_process_viewserver_with_slave_nodes(int numSlaves) throws Throwable {
        //        System.setProperty("server.bypassDataSources", "true");
        BootstrapperBase.bootstrap(viewServerContext.masterConfiguration);

        viewServerContext.master = new ViewServerMasterTest("master", viewServerContext.masterConfiguration);

        viewServerContext.master.run();

        for (int i = 0; i < numSlaves; i++) {
            ViewServerSlave slave = new ViewServerSlave("slave-" + i, new NettyInProcEndpoint("inproc://master"));
            slave.run();
            viewServerContext.slaves.add(slave);
        }
    }

    @Given("^a data source defined by \"([^\"]*)\"$")
    public void a_data_source_defined_by(String dataSourceDefinitionFile) throws Exception {
        viewServerContext.dataSource = getDataSource(dataSourceDefinitionFile);
        viewServerContext.master.getDataSourceRegistry().register(viewServerContext.dataSource);
    }

    @Given("^a CSV data source defined by \"([^\"]*)\" and loaded from \"([^\"]*)\"$")
    public void a_csv_data_source_defined_by_and_loaded_from(String dataSourceDefinitionFile, String dataFile)
            throws Exception {
        DataSource dataSource = getDataSource(dataSourceDefinitionFile);
        if (!(dataSource.getDataLoader().getDataAdapter() instanceof CsvDataAdapter)) {
            throw new Exception("Data source must use a CsvDataAdapter to use this step");
        }
        ((CsvDataAdapter) dataSource.getDataLoader().getDataAdapter()).setFileName(dataFile);
        viewServerContext.master.getDataSourceRegistry().register(dataSource);
    }

    private DataSource getDataSource(String dataSourceFile) throws IOException {
        JacksonSerialiser serialiser = new JacksonSerialiser();
        String json = Resources.toString(Resources.getResource(dataSourceFile), Charsets.UTF_8);
        DataSource dataSource = serialiser.deserialise(json, DataSource.class);
        return dataSource;
    }

    @Given("^a report defined by \"([^\"]*)\"$")
    public void a_report_defined_by(String reportDefinitionFile) throws Throwable {
        JacksonSerialiser serialiser = new JacksonSerialiser();
        String json = Resources.toString(Resources.getResource(reportDefinitionFile), Charsets.UTF_8);
        ReportDefinition reportDefinition = serialiser.deserialise(json, ReportDefinition.class);
        viewServerContext.master.getReportRegistry().register(reportDefinition);
    }

    @And("^all slave nodes are initialised$")
    public void all_slave_nodes_are_initialised() throws Throwable {
        viewServerContext.master.waitForNodes(viewServerContext.slaves.size());
    }

    @Given("^the viewserver has been bootstrapped from \"([^\"]*)\"$")
    public void the_viewserver_has_been_bootstrapped_from(String bootstrapperClassName) throws Throwable {
        viewServerContext.bootstrapperClass = bootstrapperClassName;
    }
}