FBReaderJ/test/org/test/fbreader/formats/fb2/TestFB2Reader.java

package org.test.fbreader.formats.fb2;

import org.fbreader.bookmodel.BookModel;
import org.fbreader.bookmodel.ContentsModel;
import org.fbreader.formats.fb2.FB2Reader;
import org.zlibrary.text.model.ZLTextModel;
import org.zlibrary.text.model.ZLTextParagraph;
import org.zlibrary.text.model.ZLTextTreeParagraph;
import org.zlibrary.text.model.entry.ZLTextEntry;

import junit.framework.TestCase;

import java.io.*;

public class TestFB2Reader extends TestCase {
	
	private String myDirectory = "test/data/fb2/";
	
	private boolean compareFiles(String f1, String f2) {
		int i1;
		int i2;
		FileInputStream fis1 = null;
		FileInputStream fis2 = null;
		try {
			fis1 = new FileInputStream(f1);
			fis2 = new FileInputStream(f2);
		} catch (FileNotFoundException e) {
			return false;
		}
		try{
			try{			
				i1 = fis1.read();
				i2 = fis2.read();
				while ((i1 == i2) && (i1 != -1)){
					i1 = fis1.read();
					i2 = fis2.read();
				}	
			} finally {
				fis1.close();
				fis2.close();
			}		
		} catch (IOException e) {
			return false;
		}
		return i1 == i2;
	}
	
	private void writeDumpToFile(ZLTextModel model, String outputFile) {
		FileOutputStream fos;
		try {
			fos = new FileOutputStream(outputFile);
			fos.write(model.dump().getBytes());
			fos.close();			
		} catch (Exception e) {
			fail();
		}		
	}
	
	private void writeTextModelDumpToFile(String inputFile, String outputFile) {
		FB2Reader reader = new FB2Reader(inputFile);
		ZLTextModel model = reader.read().getBookModel();
		writeDumpToFile(model, outputFile);
	}
	
	private void writeTreeModelDumpToFile(String inputFile, String outputFile) {
		FB2Reader reader = new FB2Reader(inputFile);
		ZLTextModel model = reader.read().getContentsModel();
		writeDumpToFile(model, outputFile);
	}
	
	private void doTreeModelTest(String test) {
		String test_result = myDirectory + test + "_act.txt";
		writeTreeModelDumpToFile(myDirectory + test + ".fb2", test_result);
		assertTrue(compareFiles(myDirectory + test + "_exp.txt", test_result));
		(new File(test_result)).delete();
	}
	
	private void doTest(String test) {
		String test_result = myDirectory + test + "_act.txt";
		writeTextModelDumpToFile(myDirectory + test + ".fb2", test_result);
		assertTrue(compareFiles(myDirectory + test + "_exp.txt", test_result));
		(new File(test_result)).delete();
	}
	
	public void testOneParagraph() {
		doTest("test1");
	}
	
	public void testTwoParagraphs() {
		doTest("test2");
	}
	
	public void testControlSup() {
		doTest("test3");
	}
	
	public void testControlSub() {
		doTest("test4");
	}
	
	public void testControlEmphasis() {
		doTest("test5");
	}

	public void testControlStrong() {
		doTest("test6");
	}
	
	public void testControlStrikeThrough() {
		doTest("test7");
	}
	
	public void testControlCode() {
		doTest("test8");
	}
	
	public void testControlVerse() {
		doTest("verse");
	}
	
	public void testControlDate() {
		doTest("date");
	}
	
	public void testControlSubtitle() {
		doTest("subtitle");
	}
	
	public void testControlTextAuthor() {
		doTest("text_author");
	}
	
	public void testEmptyLine() {
		doTest("empty_line");
	}
	
	public void testEmptyLineParagraphKind() {
		FB2Reader reader = new FB2Reader(myDirectory + "empty_line.fb2");
		ZLTextModel model = reader.read().getBookModel();
		assertEquals(model.getParagraph(0).getKind(),
				ZLTextParagraph.Kind.EMPTY_LINE_PARAGRAPH);
	}
	
	public void testControlCite() {
		doTest("cite");
	}
	
	public void testControlEpigraph() {
		doTest("epigraph");
	}
	
	public void testNotesSimple() {
		FB2Reader reader = new FB2Reader(myDirectory + "simple_notes.fb2");
		assertTrue(reader.read().getFootnotes().containsKey("1"));
		assertNotNull(reader.read().getFootnotes().get("1"));
	}
	
	public void testOneNote() {
		FB2Reader reader = new FB2Reader(myDirectory + "one_note.fb2");
		ZLTextModel model = reader.read().getFootnoteModel("1");
		String output = myDirectory + "one_note_act.txt";
		writeDumpToFile(model, output);
		assertTrue(compareFiles(myDirectory + "one_note_exp.txt", output));
		(new File(output)).delete();
	}
	
	public void testStanza() {
		doTest("stanza");
	}
	
	public void testStanzaParagraphKinds() {
		FB2Reader reader = new FB2Reader(myDirectory + "stanza.fb2");
		ZLTextModel model = reader.read().getBookModel();
		assertEquals(model.getParagraph(0).getKind(), ZLTextParagraph.Kind.BEFORE_SKIP_PARAGRAPH);
		assertEquals(model.getParagraph(2).getKind(), ZLTextParagraph.Kind.AFTER_SKIP_PARAGRAPH);
		assertEquals(model.getParagraph(1).getKind(), ZLTextParagraph.Kind.TEXT_PARAGRAPH);
	}
	
	public void testAnnotation() {
		doTest("annotation");
	}
	
	public void testAnnotationBeforeBodyParagraph() {
		FB2Reader reader = new FB2Reader(myDirectory + "annotation_before.fb2");
		ZLTextModel model = reader.read().getBookModel();
		assertEquals(model.getParagraph(1).getKind(),
				ZLTextParagraph.Kind.END_OF_SECTION_PARAGRAPH);
	}
	
	public void testAnnotationBeforeBody() {
		doTest("annotation_before");
	}
	
	public void testEndOfSection() {
		doTest("section");
	}
	
	public void testEndOfSectionParagraph() {		
		FB2Reader reader = new FB2Reader(myDirectory + "section.fb2");
		ZLTextModel model = reader.read().getBookModel();
		assertEquals(model.getParagraph(1).getKind(),
				ZLTextParagraph.Kind.END_OF_SECTION_PARAGRAPH);
	}
	
	public void testTitle() {
		doTest("title");
	}
	
	public void testPoemTitle() {
		doTest("poem_title");
	}
	
	public void testSectionTitle() {
		doTest("section_title");
	}
	
	public void testFootnote() {
		doTest("footnote");
	}
	
	public void testFootnote1() {
		FB2Reader reader = new FB2Reader(myDirectory + "footnote1.fb2");
		BookModel model = reader.read();
		assertEquals("footnote", 
				((ZLTextEntry)model.getParagraphByLink("note1").getEntries().get(1)).getData());
	}
	
	public void testExternalHyperlink() {
		doTest("ext_hyperlink");
	}
	
	public void testTreeParagraph() {
		doTreeModelTest("tree1");
	}
	
	public void testTreeParagraphRef() {
		FB2Reader reader = new FB2Reader(myDirectory + "tree1.fb2");
		ContentsModel model = reader.read().getContentsModel();
		assertTrue(model.getReference((ZLTextTreeParagraph) model.getParagraph(0)) == 0);
	}
	
	public void testTreeParagraphTitle() {
		doTreeModelTest("tree2");
	}
	
	public void test2TreeParagraphs() {
		doTreeModelTest("tree2sections");
	}
	
	// big tests
	public void testPoem() {
		doTest("poem");
	}
	
	public void testText1() {
		doTest("karenina");
	}
	
	public void testTextWithNotes() {
		doTest("whiteguard");
	}
	
}