package org.webslinger.httpunit;

 import java.io.File;

 import java.util.Hashtable;

 import java.util.regex.Matcher;

 import java.util.regex.Pattern;

 import com.meterware.servletunit.ServletRunner;

 import com.meterware.servletunit.ServletUnitClient;

 import com.meterware.httpunit.ClientProperties;

 import junit.framework.TestCase;

 import org.webslinger.containers.TestContainer;

 import org.webslinger.servlet.WebslingerServlet;

 import org.webslinger.servlet.WebslingerServletContextFactoryHardcoded;

 import com.meterware.httpunit.HttpNotFoundException;

 import com.meterware.httpunit.WebResponse;

 public abstract class WebslingerTestCase extends TestContainer.TestContainerTestCase {

     protected ServletRunner runner;

     protected ServletUnitClient suc;

     protected ClientProperties cp;

     protected WebslingerTestCase(String name) {

         super(name);

     }

     protected Hashtable<String, String> getBaseServletParams() throws Exception {

         Hashtable<String, String> params = new Hashtable<String, String>();

         params.put(WebslingerServlet.class.getName() + ".WebslingerServletContextFactory", WebslingerServletContextFactoryHardcoded.class.getName());

         return params;

     }

     protected Hashtable<String, String> getServletParams(File testLocation, File moduleRoot) throws Exception {

         Hashtable<String, String> params = getBaseServletParams();

         params.put("mount-path", testLocation.toString());

         params.put("module-base", moduleRoot.toString());

         return params;

     }

     protected void setUp() throws Exception {

         super.setUp();

         runner = new ServletRunner();

         Hashtable<String, String> params = getServletParams(getTestLocation(), getModuleRoot());

         runner.registerServlet("/*", WebslingerServlet.class.getName(), params);

         suc = runner.newClient();

         cp = suc.getClientProperties();

         cp.setAcceptGzip(false);

         cp.setAutoRedirect(false);

         cp.setAutoRefresh(false);

         cp.setIframeSupported(false);

         cp.setUserAgent("webslinger-test-runner/0.0");

     }

     protected void tearDown() throws Exception {

         runner.shutDown();

         super.tearDown();

     }

     public void runSimpleTest(String path, boolean exists, String contentType) {

         try {

             suc.getResponse("http://localhost" + path);

             if (!exists) fail("HttpNotFoundException was *not* thrown for " + path);

         } catch (HttpNotFoundException e) {

             // FIXME httpunit doesn't allow for fetching the text from an error page

             if (!exists) return;

             fail("Couldn't find " + path);

         } catch (RuntimeException e) {

             throw e;

         } catch (Exception e) {

             e.printStackTrace();

             fail("Caught exception fetching " + path);

         }

         WebResponse res = suc.getCurrentPage();

         try {

             if (contentType != null) assertEquals(path + " content type", contentType, res.getContentType());

         } catch (RuntimeException e) {

             throw e;

         } catch (Exception e) {

             fail("Caught exception fetching " + path);

         }

     }

     public void runSimpleTest(String path, boolean exists, String text, String contentType) {

         runSimpleTest(path, exists, contentType);

         if (!exists) return;

         WebResponse res = suc.getCurrentPage();

         try {

             //System.err.println("{" + res.getText() + "}" + res.getContentType());

             if (text != null) assertEquals(path + " content", text, res.getText());

         } catch (RuntimeException e) {

             throw e;

         } catch (Exception e) {

             fail("Caught exception fetching " + path);

         }

     }

     public String[] runSimpleTest(String path, boolean exists, Pattern pattern, String contentType) {

         runSimpleTest(path, exists, contentType);

         if (!exists) return null;

         WebResponse res = suc.getCurrentPage();

         String text = null;

         try {

             //System.err.println("{" + res.getText() + "}" + res.getContentType());

             text = res.getText();

         } catch (RuntimeException e) {

             throw e;

         } catch (Exception e) {

             fail("Caught exception fetching " + path);

         }

         Matcher matcher = pattern.matcher(text);

         if (!matcher.matches()) fail(path + " content(" + pattern.toString() + "), got(" + text + ")");

         String[] parts = new String[matcher.groupCount()];

         for (int i = 0; i < parts.length; i++) {

             parts[i] = matcher.group(i + 1);

         }

         return parts;

     }

     public void runSimpleTest(String path, String text, String contentType) {

         runSimpleTest(path, true, text, contentType);

     }

     public String[] runSimpleTest(String path, Pattern pattern, String contentType) {

         return runSimpleTest(path, true, pattern, contentType);

     }

 }