Merge commit 'edcd53bbbba9f94e21f43fd03d3a2febcc2b1564'

[fanfix.git] / src / be / nikiroo / fanfix / test / Test.java

package be.nikiroo.fanfix.test;

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

import be.nikiroo.fanfix.Instance;
import be.nikiroo.fanfix.bundles.Config;
import be.nikiroo.fanfix.bundles.ConfigBundle;
import be.nikiroo.utils.IOUtils;
import be.nikiroo.utils.TempFiles;
import be.nikiroo.utils.resources.Bundles;
import be.nikiroo.utils.test.TestLauncher;

/**
 * Tests for Fanfix.
 * 
 * @author niki
 */
public class Test extends TestLauncher {
	//
	// 4 files can control the test:
	// - test/VERBOSE: enable verbose mode
	// - test/OFFLINE: to forbid any downloading
	// - test/URLS: to allow testing URLs
	// - test/FORCE_REFRESH: to force a clear of the cache
	//
	// Note that test/CACHE can be kept, as it will contain all internet related
	// files you need (if you allow URLs, run the test once which will populate
	// the CACHE then go OFFLINE, it will still work).
	//
	// The test files will be:
	// - test/*.url: URL to download in text format, content = URL
	// - test/*.story: text mode story, content = story
	//

	/**
	 * The temporary files handler.
	 */
	static TempFiles tempFiles;

	/**
	 * Create the Fanfix {@link TestLauncher}.
	 * 
	 * @param args
	 *            the arguments to configure the number of columns and the ok/ko
	 *            {@link String}s
	 * @param urlsAllowed
	 *            allow testing URLs (<tt>.url</tt> files)
	 * 
	 * @throws IOException
	 */
	public Test(String[] args, boolean urlsAllowed) throws IOException {
		super("Fanfix", args);
		Instance.setTraceHandler(null);
		addSeries(new BasicSupportUtilitiesTest(args));
		addSeries(new BasicSupportDeprecatedTest(args));
		addSeries(new LibraryTest(args));

		File sources = new File("test/");
		if (sources.isDirectory()) {
			for (File file : sources.listFiles()) {
				if (file.isDirectory()) {
					continue;
				}

				String expectedDir = new File(file.getParentFile(), "expected_"
						+ file.getName()).getAbsolutePath();
				String resultDir = new File(file.getParentFile(), "result_"
						+ file.getName()).getAbsolutePath();

				String uri;
				if (urlsAllowed && file.getName().endsWith(".url")) {
					uri = IOUtils.readSmallFile(file).trim();
				} else if (file.getName().endsWith(".story")) {
					uri = file.getAbsolutePath();
				} else {
					continue;
				}

				addSeries(new ConversionTest(file.getName(), uri, expectedDir,
						resultDir, args));
			}
		}
	}

	/**
	 * Main entry point of the program.
	 * 
	 * @param args
	 *            the arguments passed to the {@link TestLauncher}s.
	 * @throws IOException
	 *             in case of I/O error
	 */
	static public void main(String[] args) throws IOException {
		Instance.init();

		// Verbose mode:
		boolean verbose = new File("test/VERBOSE").exists();

		// Can force refresh
		boolean forceRefresh = new File("test/FORCE_REFRESH").exists();

		// Allow URLs:
		boolean urlsAllowed = new File("test/URLS").exists();

		
		// Only download files if allowed:
		boolean offline = new File("test/OFFLINE").exists();
		Instance.getCache().setOffline(offline);


		
		int result = 0;
		tempFiles = new TempFiles("fanfix-test");
		try {
			File tmpConfig = tempFiles.createTempDir("fanfix-config");
			File localCache = new File("test/CACHE");
			prepareCache(localCache, forceRefresh);

			ConfigBundle config = new ConfigBundle();
			Bundles.setDirectory(tmpConfig.getAbsolutePath());
			config.setString(Config.CACHE_DIR, localCache.getAbsolutePath());
			config.setInteger(Config.CACHE_MAX_TIME_STABLE, -1);
			config.setInteger(Config.CACHE_MAX_TIME_CHANGING, -1);
			config.updateFile(tmpConfig.getPath());
			System.setProperty("CONFIG_DIR", tmpConfig.getAbsolutePath());

			Instance.init(true);
			Instance.getCache().setOffline(offline);

			TestLauncher tests = new Test(args, urlsAllowed);
			tests.setDetails(verbose);

			result = tests.launch();

			IOUtils.deltree(tmpConfig);
			prepareCache(localCache, forceRefresh);
		} finally {
			// Test temp files
			tempFiles.close();

			// This is usually done in Fanfix.Main:
			Instance.getTempFiles().close();
		}

		System.exit(result);
	}

	/**
	 * Prepare the cache (or clean it up).
	 * <p>
	 * The cache directory will always exist if this method succeed
	 * 
	 * @param localCache
	 *            the cache directory
	 * @param forceRefresh
	 *            TRUE to force acache refresh (delete all files)
	 * 
	 * @throw IOException if the cache cannot be created
	 */
	private static void prepareCache(File localCache, boolean forceRefresh)
			throws IOException {
		// if needed
		localCache.mkdirs();

		if (!localCache.isDirectory()) {
			throw new IOException("Cannot get a cache");
		}

		// delete local cached files (_*) or all files if forceRefresh
		for (File f : localCache.listFiles()) {
			if (forceRefresh || f.getName().startsWith("_")) {
				IOUtils.deltree(f);
			}
		}
	}
}