Added tests and initial fix for opening uncompressed WARCs

This initial unit test just tests that the first record from
compressed and uncompressed WARCs can be accessed.

src/main/java/org/archive/io/warc/WARCReaderFactory.java

@@ -103,9 +103,13 @@ public static ArchiveReader get(final String s, final InputStream is,
     protected ArchiveReader getArchiveReader(final String f,
 			final InputStream is, final boolean atFirstRecord)
 			throws IOException {
-		// For now, assume stream is compressed. Later add test of input
-		// stream or handle exception thrown when figure not compressed stream.
-		return new CompressedWARCReader(f, is, atFirstRecord);
+    	// Check if it's compressed:
+    	// TODO Currently relies on the file extension, but this should all really sniff the content properly.
+    	if( f.endsWith(".gz") ) {
+    		return new CompressedWARCReader(f, is, atFirstRecord);
+    	} else {
+    		return new UncompressedWARCReader(f, is);
+    	}
 	}
 
     public static WARCReader get(final URL arcUrl, final long offset)

src/test/java/org/archive/io/warc/WARCReaderFactoryTest.java

@@ -0,0 +1,34 @@
+package org.archive.io.warc;
+
+import java.io.FileInputStream;
+import java.io.IOException;
+
+import org.archive.format.warc.WARCConstants;
+import org.archive.format.warc.WARCConstants.WARCRecordType;
+import org.archive.io.ArchiveReader;
+import org.archive.io.ArchiveRecord;
+
+import junit.framework.TestCase;
+
+public class WARCReaderFactoryTest extends TestCase {
+
+	// Test files:
+	String[] files = new String[] {
+			"src/test/resources/org/archive/format/gzip/IAH-urls-wget.warc.gz",
+			"src/test/resources/org/archive/format/warc/IAH-urls-wget.warc"
+	};
+
+	public void testGetStringInputstreamBoolean() throws IOException {
+		// Check the test files can be opened:
+		for( String file : files ) {
+			FileInputStream is = new FileInputStream(file);
+			ArchiveReader ar = WARCReaderFactory.get(file, is, true);
+			ArchiveRecord r = ar.get();
+			String type = (String) r.getHeader().getHeaderValue(WARCConstants.HEADER_KEY_TYPE);
+			// Check the first record comes out as a 'warcinfo' record.
+			assertEquals(WARCRecordType.warcinfo.name(), type);
+		}
+	}
+
+
+}