package com.ziesemer.utils.web;
   
   import static org.junit.Assert.assertEquals;
   import static org.junit.Assert.fail;
   
   import java.io.DataInputStream;
   import java.io.File;
   import java.io.InputStream;
   import java.util.zip.ZipEntry;
  import java.util.zip.ZipFile;
  
  import org.eclipse.jetty.servlet.ServletContextHandler;
  import org.slf4j.LoggerFactory;
  
  /**
   * <p>Requires that &quot;<code>yui_2.9.0.zip</code>&quot; is downloaded from <a href=
   * 	"http://yui.github.io/yui2/">http://yui.github.io/yui2/</a>
   * 	and placed underneath <code>TestWebContent/WEB-INF/</code>.</p>
   * @author Mark A. Ziesemer
   * <a href="http://www.ziesemer.com.">&lt;www.ziesemer.com&gt;</a>
   */
  public class ZipServletTestSupport extends ServletTester{
  	public static final String WEB_CONTENT = "TestWebContent";
  	public static final String PREFIX = "yui_2.9.0";
  	
  	public static final String TEST_SINGLE_PATH =
  		"build/yahoo/yahoo.js";
  	public static final String TEST_COMBO_PATH =
  		"build/yahoo/yahoo-min.js&build/dom/dom-min.js&build/event/event-min.js";
  	
  	protected static final File ZIP_FILE_FILE;
  	
  	static{
  		ZIP_FILE_FILE = new File(WEB_CONTENT + "/WEB-INF/" + PREFIX + ".zip");
  		if(!ZIP_FILE_FILE.exists()){
  			final String err = ZIP_FILE_FILE
  				+ " must be provided to run unit tests.  Please download from http://yui.github.io/yui2/ .";
  			LoggerFactory.getLogger(ZipServletTestSupport.class).error(err);
  			throw new ExceptionInInitializerError(err);
  		}
  	}
  	
  	protected final ZipFile zipFile;
  	
  	public ZipServletTestSupport() throws Exception{
  		zipFile = new ZipFile(ZIP_FILE_FILE);
  	}
  	
  	@Override
  	protected ServletContextHandler initHandler(){
  		final ServletContextHandler c = initHandlerClass();
  		c.setResourceBase(WEB_CONTENT);
  		return c;
  	}
  	
  	protected ServletContextHandler initHandlerClass(){
  		return new ServletContextHandler();
  	}
  	
  	public ZipFile getZipFile(){
  		return zipFile;
  	}
  	
  	public File getZipFileFile(){
  		return ZIP_FILE_FILE;
  	}
  	
  	/**
  	 * Does not close or check for EOF on the InputStream.
  	 * Can be called repeatedly for testing combo mode.
  	 * Finish by calling {@link #testEndOfStream(InputStream)}.
  	 */
  	public void testStreamsEqual(final ZipEntry ze, final InputStream is) throws Exception{
  		long readSum = 0;
  		final InputStream zeIs = getZipFile().getInputStream(ze);
  		try{
  			final DataInputStream srvIs = new DataInputStream(is);
  			final byte[] zeBuffer = new byte[1 << 10];
  			final byte[] srvBuffer = new byte[zeBuffer.length];
  			int read;
  			while((read = zeIs.read(zeBuffer)) >= 0){
  				srvIs.readFully(srvBuffer, 0, read);
  				for(int i=0; i < read; i++){
  					if(zeBuffer[i] != srvBuffer[i]){
  						fail("Unmatched bytes at position: " + (readSum + i)
  							+ ".  expected:<" + zeBuffer[i] + "> but was:<" + srvBuffer[i] + ">");
  					}
  				}
  				readSum += read;
  			}
  		}finally{
  			zeIs.close();
  		}
  		
  		assertEquals("Didn't read all bytes.",
  			ze.getSize(), readSum);
  	}
  	
  	public void testEndOfStream(final InputStream is) throws Exception{
 		assertEquals(-1, is.read());
 	}
 
 	@Override
 	public void close() throws Exception{
 		zipFile.close();
 		super.close();
 	}
 }