Reorganise test system via Downloader/Cache

[gofetch.git] / src / be / nikiroo / gofetch / test / TestBase.java

package be.nikiroo.gofetch.test;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.Map;

import be.nikiroo.gofetch.data.Story;
import be.nikiroo.gofetch.output.Gopher;
import be.nikiroo.gofetch.output.Html;
import be.nikiroo.gofetch.output.Output;
import be.nikiroo.gofetch.support.BasicSupport;
import be.nikiroo.utils.Cache;
import be.nikiroo.utils.Downloader;
import be.nikiroo.utils.IOUtils;
import be.nikiroo.utils.test.TestCase;

/**
 * Base class for {@link BasicSupport}s testing.
 * <p>
 * It will use the paths:
 * <ul>
 * <li><tt>test/???/source</tt>: the html source files</li>
 * <li><tt>test/???/expected</tt>: the expected output</li>
 * <li><tt>test/???/actual</tt>: the actual output of the last test</li>
 * </ul>
 * 
 * @author niki
 */
abstract class TestBase extends TestCase {
	private BasicSupport support;
	private Cache cache;
	private Downloader downloader;

	public TestBase(BasicSupport support, String[] args) {
		super(support.getType().toString());
		this.support = support;
		try {
			cache = new Cache(new File("test/source/" + support.getType()), -1,
					-1);
			downloader = new Downloader("gofetch", cache);
		} catch (IOException e) {
			throw new RuntimeException(e);
		}
	}

	protected InputStream download(URL url) throws IOException {
		return downloader.open(url);
	}

	static protected InputStream doOpen(BasicSupport support,
			Map<URL, File> map, URL url) throws IOException {
		File file = map.get(url);
		if (file == null) {
			throw new FileNotFoundException("Test file not found for URL: "
					+ url);
		}

		return new FileInputStream("test/source/" + support.getType() + "/"
				+ file);

	}

	@Override
	public void test() throws Exception {
		File expected = new File("test/expected/" + support.getType());
		File actual = new File("test/result/" + support.getType());

		IOUtils.deltree(actual);
		expected.mkdirs();
		actual.mkdirs();

		Output gopher = new Gopher(support.getType(), "", "", 70);
		Output html = new Html(support.getType(), "", "", 80);

		for (Story story : support.list()) {
			support.fetch(story);
			IOUtils.writeSmallFile(new File(actual, story.getId() + ".header"),
					gopher.exportHeader(story));
			IOUtils.writeSmallFile(new File(actual, story.getId() + ""),
					gopher.export(story));
			IOUtils.writeSmallFile(new File(actual, story.getId()
					+ ".header.html"), html.exportHeader(story));
			IOUtils.writeSmallFile(new File(actual, story.getId() + ".html"),
					html.export(story));
		}

		assertEquals(expected, actual);
	}
}