com.splunk.shuttl.server.mbeans.ShuttlArchiverMBeanTest.java Source code

Introduction

Here is the source code for com.splunk.shuttl.server.mbeans.ShuttlArchiverMBeanTest.java

Source

// Copyright (C) 2011 Splunk Inc.
//
// Splunk Inc. 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.
package com.splunk.shuttl.server.mbeans;

import static com.splunk.shuttl.testutil.TUtilsFile.*;
import static java.util.Arrays.*;
import static org.testng.Assert.*;

import java.io.File;
import java.util.Arrays;
import java.util.List;

import org.apache.commons.io.FileUtils;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import com.splunk.shuttl.testutil.TUtilsMBean;
import com.splunk.shuttl.testutil.TUtilsString;

/**
 * White box testing of MBeans
 */
@Test(groups = { "fast-unit" })
public class ShuttlArchiverMBeanTest {
    ShuttlArchiver archiverMBean;

    @BeforeMethod(groups = { "fast-unit" })
    public void createMBean() throws Exception {
        File confFile = getTempFile();
        System.out.println("ShuttlArchiverMBeanTest - running " + confFile.getPath());
        archiverMBean = ShuttlArchiver.createWithConfFile(confFile);
    }

    @Test(groups = { "fast-unit" })
    public void setArchiverRoot_archiverRootIsSet_gotArchiverRoot() {
        String archiverRoot = "/test_archive_root";
        assertNotEquals(archiverMBean.getArchiverRootURI(), archiverRoot);
        archiverMBean.setArchiverRootURI(archiverRoot);
        assertEquals(archiverMBean.getArchiverRootURI(), archiverRoot);
    }

    public void setArchiveFormat_archiveFormatIsSet_gotArchiveFormat() {
        List<String> archiveFormats1 = asList("SPLUNK_BUCKET");
        assertNotEquals(archiverMBean.getArchiveFormats(), archiveFormats1);
        archiverMBean.setArchiveFormats(archiveFormats1);
        assertEquals(archiverMBean.getArchiveFormats(), archiveFormats1);
    }

    public void setClusterName_clusterNameIsSet_gotClusterName() {
        String clusterName = "test_cluster_name";
        assertNotEquals(archiverMBean.getClusterName(), clusterName);
        archiverMBean.setClusterName(clusterName);
        assertEquals(archiverMBean.getClusterName(), clusterName);
    }

    public void setServerName_serverNameIsSet_gotCluserName() {
        String serverName = "test_server_name";
        assertNotEquals(archiverMBean.getServerName(), serverName);
        archiverMBean.setServerName(serverName);
        assertEquals(archiverMBean.getServerName(), serverName);
    }

    public void setBucketFormatPriority_priorityIsSet_gotPriority() {
        List<String> bucketFormatPriority = Arrays.asList("SPLUNK_BUCKET", "UNKNOWN");
        assertNotEquals(archiverMBean.getBucketFormatPriority(), bucketFormatPriority);
        archiverMBean.setBucketFormatPriority(bucketFormatPriority);
        assertEquals(archiverMBean.getBucketFormatPriority(), bucketFormatPriority);
    }

    public void save_configured_producesCorrectXML() throws Exception {
        List<String> archiveFormats = asList("SPLUNK_BUCKET", "CSV");
        String clusterName = "some_cluster_name";
        String serverName = "some_server_name";
        String archiverRootURI = "hdfs://localhost:1234";
        String expectedConfigFile = TUtilsMBean.XML_HEADER
                + "<ns2:archiverConf xmlns:ns2=\"com.splunk.shuttl.server.model\">\n" + "<archiveFormats>\n"
                + "<archiveFormat>SPLUNK_BUCKET</archiveFormat>\n" + "<archiveFormat>CSV</archiveFormat>\n"
                + "</archiveFormats>\n" + "<clusterName>" + clusterName + "</clusterName>\n" + "<serverName>"
                + serverName + "</serverName>\n" + "<archiverRootURI>" + archiverRootURI + "</archiverRootURI>\n"
                + "<bucketFormatPriority>" + "SPLUNK_BUCKET" + "</bucketFormatPriority>\n"
                + "<bucketFormatPriority>" + "CSV" + "</bucketFormatPriority>\n" + "</ns2:archiverConf>\n";

        File file = getTempFile();
        archiverMBean = ShuttlArchiver.createWithConfFile(file);
        archiverMBean.setArchiveFormats(archiveFormats);
        archiverMBean.setClusterName(clusterName);
        archiverMBean.setServerName(serverName);
        archiverMBean.setArchiverRootURI(archiverRootURI);
        archiverMBean.setBucketFormatPriority(archiveFormats);
        archiverMBean.save();

        assertEquals(TUtilsString.noSpaces(FileUtils.readFileToString(file)),
                TUtilsString.noSpaces(expectedConfigFile));
    }

    public void load_preconfiguredFile_givesCorrectValues() throws Exception {
        List<String> archiveFormats = asList("SPLUNK_BUCKET", "CSV");
        String clusterName = "some_cluster_name";
        String serverName = "some_server_name";
        String archiverRootURI = "hdfs://localhost:1234";
        String configFilePreset = TUtilsMBean.XML_HEADER
                + "<ns2:archiverConf xmlns:ns2=\"com.splunk.shuttl.server.model\">\n" + "<archiveFormats>\n"
                + "<archiveFormat>SPLUNK_BUCKET</archiveFormat>\n" + "<archiveFormat>CSV</archiveFormat>\n"
                + "</archiveFormats>\n" + "<clusterName>" + clusterName + "</clusterName>\n" + "    <serverName>"
                + serverName + "</serverName>\n" + "    <archiverRootURI>" + archiverRootURI
                + "</archiverRootURI>\n" + "    <bucketFormatPriority>" + "SPLUNK_BUCKET"
                + "</bucketFormatPriority>\n" + "</ns2:archiverConf>";

        File file = createFile();
        file.deleteOnExit();
        FileUtils.writeStringToFile(file, configFilePreset);
        archiverMBean = ShuttlArchiver.createWithConfFile(file);
        assertEquals(archiverMBean.getArchiveFormats(), archiveFormats);
        assertEquals(archiverMBean.getClusterName(), clusterName);
        assertEquals(archiverMBean.getServerName(), serverName);
        assertEquals(archiverMBean.getArchiverRootURI(), archiverRootURI);
    }

    private File getTempFile() throws Exception {
        return TUtilsMBean.createEmptyConfInNamespace("archiverConf");
    }

}