info.fetter.logstashforwarder.FileWatcherTest.java Source code

Introduction

Here is the source code for info.fetter.logstashforwarder.FileWatcherTest.java

Source

package info.fetter.logstashforwarder;

/*
 * Copyright 2015 Didier Fetter
 *
 * 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.
 *
 */

import static org.apache.log4j.Level.*;

import java.io.File;
import java.io.IOException;

import org.apache.commons.io.FileUtils;
import org.apache.log4j.BasicConfigurator;
import org.apache.log4j.Logger;
import org.apache.log4j.spi.RootLogger;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;

public class FileWatcherTest {
    Logger logger = Logger.getLogger(FileWatcherTest.class);

    @BeforeClass
    public static void setUpBeforeClass() throws Exception {
        BasicConfigurator.configure();
        RootLogger.getRootLogger().setLevel(TRACE);
    }

    @AfterClass
    public static void tearDownAfterClass() throws Exception {
        BasicConfigurator.resetConfiguration();
    }

    //@Test
    public void testFileWatch() throws InterruptedException, IOException {
        FileWatcher watcher = new FileWatcher();
        watcher.addFilesToWatch("./test.txt", new Event().addField("test", "test"), FileWatcher.ONE_DAY);
        for (int i = 0; i < 100; i++) {
            Thread.sleep(1000);
            watcher.checkFiles();
        }
    }

    @Test
    public void testWildcardWatch() throws InterruptedException, IOException {
        if (System.getProperty("os.name").toLowerCase().contains("win")) {
            logger.warn("Not executing this test on windows");
            return;
        }
        FileWatcher watcher = new FileWatcher();
        watcher.addFilesToWatch("./testFileWatcher*.txt", new Event().addField("test", "test"),
                FileWatcher.ONE_DAY);
        watcher.initialize();

        File file1 = new File("testFileWatcher1.txt");
        File file2 = new File("testFileWatcher2.txt");
        //File file3 = new File("test3.txt");
        //File file4 = new File("test4.txt");

        //File testDir = new File("testFileWatcher");
        //FileUtils.forceMkdir(new File("test"));

        watcher.checkFiles();
        Thread.sleep(100);
        FileUtils.write(file1, "file 1 line 1\n", true);
        Thread.sleep(100);
        watcher.checkFiles();
        FileUtils.write(file1, "file 1 line 2\n", true);
        //FileUtils.write(file2, "file 2 line 1\n", true);
        Thread.sleep(1000);
        watcher.checkFiles();
        //      FileUtils.moveFileToDirectory(file1, testDir, true);
        //      FileUtils.write(file2, "file 2 line 2\n", true);
        FileUtils.moveFile(file1, file2);
        //      FileUtils.write(file2, "file 3 line 1\n", true);
        //
        Thread.sleep(1000);
        watcher.checkFiles();
        //      
        //      
        watcher.close();
        FileUtils.deleteQuietly(file1);
        FileUtils.deleteQuietly(file2);
        //      FileUtils.forceDelete(testDir);

    }

    @Test
    public void dummy() {
    }
}