com.splunk.shuttl.testutil.SplunkTestUtils.java Source code

Introduction

Here is the source code for com.splunk.shuttl.testutil.SplunkTestUtils.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.testutil;

import java.io.IOException;
import java.io.InputStream;
import java.util.List;

import org.apache.commons.io.IOUtils;

import com.splunk.EntityCollection;
import com.splunk.Index;
import com.splunk.Job;
import com.splunk.Service;

/**
 * Util class for tests that are run against Splunk.
 */
public class SplunkTestUtils {

    private static final int SLEEP_IN_MILLIS_BETWEEN_JOB_REFRESH = 150;

    public static final String TEST_RESOURCES_PATH = "test/resources";

    public static void waitWhileJobFinishes(Job job) {
        while (!job.isDone()) {
            sleep(SLEEP_IN_MILLIS_BETWEEN_JOB_REFRESH);
            job.refresh();
        }
    }

    private static void sleep(int millis) {
        try {
            Thread.sleep(millis);
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        }
    }

    public static List<String> readSearchResults(InputStream results) {
        try {
            return IOUtils.readLines(results);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    static public void waitForIndexing(Index index, int value, int seconds) {
        int indexedEventCount = 0;
        while (seconds > 0)
            try {
                // 5000ms (5 second sleep)
                Thread.sleep(5000);
                seconds = seconds - 5;
                indexedEventCount = index.getTotalEventCount();
                if (indexedEventCount == value)
                    return;
                index.refresh();
            } catch (InterruptedException e) {
                continue;
            }
        String msg = "Indexing incomplete for index " + index + " after " + seconds + "seconds, expectedCount="
                + value + " actualCount=" + indexedEventCount;
        throw new RuntimeException(msg);
    }

    public static Index createSplunkIndex(Service service, String name) {
        Index index;
        EntityCollection<Index> indexes = service.getIndexes();
        if (indexes.containsKey(name)) {
            System.out.println("Index " + name + " already exists");
            index = indexes.get(name);
        } else {
            indexes.create(name);
            index = indexes.get(name);
            index.refresh();
            indexes.refresh();
            System.out.println("Index " + name + " created");
        }

        return index;
    }
}