FileWatcherTest.java

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() {}
}