test/com/google/enterprise/adaptor/fs/HtmlResponseWriterTest.java - plexi/fs - Git at Google

 // Copyright 2012 Google Inc. All Rights Reserved.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
 //
 //      http://www.apache.org/licenses/LICENSE-2.0
 //
 // Unless required by applicable law or agreed to in writing, software
 // distributed under the License is distributed on an "AS IS" BASIS,
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 // See the License for the specific language governing permissions and
 // limitations under the License.

 package com.google.enterprise.adaptor.fs;

 import static org.junit.Assert.*;

 import com.google.enterprise.adaptor.DocId;
 import com.google.enterprise.adaptor.DocIdEncoder;

 import org.junit.*;
 import org.junit.rules.ExpectedException;

 import java.io.StringWriter;
 import java.net.URI;
 import java.net.URISyntaxException;
 import java.util.Locale;

 /**
  * Test cases for {@link HtmlResponseWriter}.
  */
 public class HtmlResponseWriterTest {
   @Rule
   public ExpectedException thrown = ExpectedException.none();

   private final DocIdEncoder docIdEncoder = new DocIdEncoder() {
     @Override
     public URI encodeDocId(DocId docId) {
       URI base = URI.create("http://localhost/");
       URI resource;
       try {
         resource = new URI(null, null, "/" + docId.getUniqueId(), null);
       } catch (URISyntaxException ex) {
         throw new AssertionError();
       }
       return base.resolve(resource);
     }
   };

   private StringWriter output = new StringWriter(1024);
   private HtmlResponseWriter writer = new HtmlResponseWriter(
       output, docIdEncoder, Locale.ENGLISH);

   @Test
   public void testConstructorNullWriter() {
     thrown.expect(NullPointerException.class);
     new HtmlResponseWriter(null, docIdEncoder, Locale.ENGLISH);
   }

   @Test
   public void testConstructorNullDocIdEncoder() {
     thrown.expect(NullPointerException.class);
     new HtmlResponseWriter(output, null, Locale.ENGLISH);
   }

   @Test
   public void testConstructorNullLocale() {
     thrown.expect(NullPointerException.class);
     new HtmlResponseWriter(output, docIdEncoder, null);
   }

   @Test
   public void testBasicFlow() throws Exception {
     final String golden = "<!DOCTYPE html>\n"
         + "<html><head><title>Folder root</title></head>"
         + "<body><h1>Folder root</h1>"
         + "<li><a href=\"root/foo\">foo</a></li>"
         + "<li><a href=\"root/bar\">bar</a></li>"
         + "</body></html>";
     writer.start(new DocId("root"), null);
     writer.addLink(new DocId("root/foo"), "foo");
     writer.addLink(new DocId("root/bar"), "bar");
     writer.finish();
     assertEquals(golden, output.toString());
   }

   @Test
   public void testStartTwice() throws Exception {
     writer.start(new DocId(""), null);
     thrown.expect(IllegalStateException.class);
     writer.start(new DocId(""), null);
   }

   @Test
   public void testAddLinkPremature() throws Exception {
     thrown.expect(IllegalStateException.class);
     writer.addLink(new DocId(""), null);
   }

   @Test
   public void testAddLinkNullDocId() throws Exception {
     writer.start(new DocId(""), null);
     thrown.expect(NullPointerException.class);
     writer.addLink(null, null);
   }

   @Test
   public void testFinishPremature() throws Exception {
     thrown.expect(IllegalStateException.class);
     writer.finish();
   }

   @Test
   public void testFinishNoSections() throws Exception {
     final String golden = "<!DOCTYPE html>\n"
         + "<html><head><title>Folder root</title></head>"
         + "<body><h1>Folder root</h1>"
         + "</body></html>";
     writer.start(new DocId("root"), "");
     writer.finish();
     writer.close();
     assertEquals(golden, output.toString());
   }

   @Test
   public void testRelativizeBaseNoScheme() throws Exception {
     URI base = new URI(null, "example.com", "/", null, null);
     URI uri = new URI("http://example.com/some/where/file");
     assertEquals(uri, HtmlResponseWriter.relativize(base, uri));
   }

   @Test
   public void testRelativizeBaseNoAuthority() throws Exception {
     URI base = new URI("http", null, "/", null, null);
     URI uri = new URI("http://example.com/some/where/file");
     assertEquals(uri, HtmlResponseWriter.relativize(base, uri));
   }

   @Test
   public void testRelativizeBaseDifferentScheme() throws Exception {
     URI base = new URI("file", "example.com", "/", null, null);
     URI uri = new URI("http://example.com/some/where/file");
     assertEquals(uri, HtmlResponseWriter.relativize(base, uri));
   }

   @Test
   public void testRelativizeBaseDifferentAuthority() throws Exception {
     URI base = new URI("http", "google.com", "/", null, null);
     URI uri = new URI("http://example.com/some/where/file");
     assertEquals(uri, HtmlResponseWriter.relativize(base, uri));
   }

   @Test
   public void testRelativizeBaseMatches() throws Exception {
     URI base = new URI("http", "example.com", null, null, null);
     URI uri = new URI("http://example.com/some/where/file");
     URI golden = new URI(null, null, "/some/where/file", null, null);
     assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
   }

   @Test
   public void testRelativizeSimplePathChild() throws Exception {
     URI base = new URI("http", "example.com", "/some/where", null, null);
     URI uri = new URI("http://example.com/some/where/file");
     URI golden = new URI(null, null, "where/file", null, null);
     assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
   }

   @Test
   public void testRelativizeSimplePathParent() throws Exception {
     URI base = new URI("http", "example.com", "/some/where/file", null, null);
     URI uri = new URI("http://example.com/some/where");
     // Note that "." or "./" would not suffice, since we need "where" to not end
     // with a slash.
     URI golden = new URI(null, null, "../where", null, null);
     assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
   }

   @Test
   public void testRelativizeComplexPathFileBaseFolder()
       throws Exception {
     URI base = new URI("http", "example.com", "/some/there/", null, null);
     URI uri = new URI("http://example.com/some/where/file");
     URI golden = new URI(null, null, "../where/file", null, null);
     assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
   }

   @Test
   public void testRelativizeComplexPathFileBaseFile() throws Exception {
     URI base = new URI("http", "example.com", "/some/there/file", null, null);
     URI uri = new URI("http://example.com/some/where/file");
     URI golden = new URI(null, null, "../where/file", null, null);
     assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
   }

   @Test
   public void testRelativizeComplexPathFolderBaseFolder()
       throws Exception {
     URI base = new URI("http", "example.com", "/some/there/", null, null);
     URI uri = new URI("http://example.com/some/where/folder/");
     URI golden = new URI(null, null, "../where/folder/", null, null);
     assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
   }

   @Test
   public void testRelativizeComplexPathFolderBaseFile() throws Exception {
     URI base = new URI("http", "example.com", "/some/there/file", null, null);
     URI uri = new URI("http://example.com/some/where/folder/");
     URI golden = new URI(null, null, "../where/folder/", null, null);
     assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
   }

   @Test
   public void testRelativizeSame() throws Exception {
     URI uri = new URI("http://example.com/some/where/file");
     URI golden = new URI(null, null, null, null, "");
     assertEquals(golden, HtmlResponseWriter.relativize(uri, uri));
   }

   @Test
   public void testRelativizeNoMistakenScheme() throws Exception {
     URI base = new URI("http", "example.com", "/", null, null);
     URI uri = new URI("http://example.com/http://google.com/");
     URI golden = new URI(null, null, "./http://google.com/", null, null);
     assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
   }

   @Test
   public void testRelativizeNoMistakenSchemeNoSlash() throws Exception {
     URI base = new URI("http", "example.com", "/", null, null);
     URI uri = new URI("http://example.com/http:");
     URI golden = new URI(null, null, "./http:", null, null);
     assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
   }

   @Test
   public void testRelativizeNoMistakenSchemeSlashFirst() throws Exception {
     URI base = new URI("http", "example.com", "/", null, null);
     URI uri = new URI("http://example.com/http/:");
     URI golden = new URI(null, null, "http/:", null, null);
     assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
   }

   @Test
   public void testRelativizeSchemeImpliesAbsoluteUriAssumption()
       throws Exception {
     thrown.expect(URISyntaxException.class);
     new URI("http", "example.com", "doesnt/start/with/a/slash", null, null);
   }
 }
	// Copyright 2012 Google Inc. All Rights Reserved.
	//
	// Licensed under the Apache License, Version 2.0 (the "License");
	// you may not use this file except in compliance with the License.
	// You may obtain a copy of the License at
	//
	// http://www.apache.org/licenses/LICENSE-2.0
	//
	// Unless required by applicable law or agreed to in writing, software
	// distributed under the License is distributed on an "AS IS" BASIS,
	// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	// See the License for the specific language governing permissions and
	// limitations under the License.

	package com.google.enterprise.adaptor.fs;

	import static org.junit.Assert.*;

	import com.google.enterprise.adaptor.DocId;
	import com.google.enterprise.adaptor.DocIdEncoder;

	import org.junit.*;
	import org.junit.rules.ExpectedException;

	import java.io.StringWriter;
	import java.net.URI;
	import java.net.URISyntaxException;
	import java.util.Locale;

	/**
	* Test cases for {@link HtmlResponseWriter}.
	*/
	public class HtmlResponseWriterTest {
	@Rule
	public ExpectedException thrown = ExpectedException.none();

	private final DocIdEncoder docIdEncoder = new DocIdEncoder() {
	@Override
	public URI encodeDocId(DocId docId) {
	URI base = URI.create("http://localhost/");
	URI resource;
	try {
	resource = new URI(null, null, "/" + docId.getUniqueId(), null);
	} catch (URISyntaxException ex) {
	throw new AssertionError();
	}
	return base.resolve(resource);
	}
	};

	private StringWriter output = new StringWriter(1024);
	private HtmlResponseWriter writer = new HtmlResponseWriter(
	output, docIdEncoder, Locale.ENGLISH);

	@Test
	public void testConstructorNullWriter() {
	thrown.expect(NullPointerException.class);
	new HtmlResponseWriter(null, docIdEncoder, Locale.ENGLISH);
	}

	@Test
	public void testConstructorNullDocIdEncoder() {
	thrown.expect(NullPointerException.class);
	new HtmlResponseWriter(output, null, Locale.ENGLISH);
	}

	@Test
	public void testConstructorNullLocale() {
	thrown.expect(NullPointerException.class);
	new HtmlResponseWriter(output, docIdEncoder, null);
	}

	@Test
	public void testBasicFlow() throws Exception {
	final String golden = "<!DOCTYPE html>\n"
	+ "<html><head><title>Folder root</title></head>"
	+ "<body><h1>Folder root</h1>"
	+ "<li><a href=\"root/foo\">foo</a></li>"
	+ "<li><a href=\"root/bar\">bar</a></li>"
	+ "</body></html>";
	writer.start(new DocId("root"), null);
	writer.addLink(new DocId("root/foo"), "foo");
	writer.addLink(new DocId("root/bar"), "bar");
	writer.finish();
	assertEquals(golden, output.toString());
	}

	@Test
	public void testStartTwice() throws Exception {
	writer.start(new DocId(""), null);
	thrown.expect(IllegalStateException.class);
	writer.start(new DocId(""), null);
	}

	@Test
	public void testAddLinkPremature() throws Exception {
	thrown.expect(IllegalStateException.class);
	writer.addLink(new DocId(""), null);
	}

	@Test
	public void testAddLinkNullDocId() throws Exception {
	writer.start(new DocId(""), null);
	thrown.expect(NullPointerException.class);
	writer.addLink(null, null);
	}

	@Test
	public void testFinishPremature() throws Exception {
	thrown.expect(IllegalStateException.class);
	writer.finish();
	}

	@Test
	public void testFinishNoSections() throws Exception {
	final String golden = "<!DOCTYPE html>\n"
	+ "<html><head><title>Folder root</title></head>"
	+ "<body><h1>Folder root</h1>"
	+ "</body></html>";
	writer.start(new DocId("root"), "");
	writer.finish();
	writer.close();
	assertEquals(golden, output.toString());
	}

	@Test
	public void testRelativizeBaseNoScheme() throws Exception {
	URI base = new URI(null, "example.com", "/", null, null);
	URI uri = new URI("http://example.com/some/where/file");
	assertEquals(uri, HtmlResponseWriter.relativize(base, uri));
	}

	@Test
	public void testRelativizeBaseNoAuthority() throws Exception {
	URI base = new URI("http", null, "/", null, null);
	URI uri = new URI("http://example.com/some/where/file");
	assertEquals(uri, HtmlResponseWriter.relativize(base, uri));
	}

	@Test
	public void testRelativizeBaseDifferentScheme() throws Exception {
	URI base = new URI("file", "example.com", "/", null, null);
	URI uri = new URI("http://example.com/some/where/file");
	assertEquals(uri, HtmlResponseWriter.relativize(base, uri));
	}

	@Test
	public void testRelativizeBaseDifferentAuthority() throws Exception {
	URI base = new URI("http", "google.com", "/", null, null);
	URI uri = new URI("http://example.com/some/where/file");
	assertEquals(uri, HtmlResponseWriter.relativize(base, uri));
	}

	@Test
	public void testRelativizeBaseMatches() throws Exception {
	URI base = new URI("http", "example.com", null, null, null);
	URI uri = new URI("http://example.com/some/where/file");
	URI golden = new URI(null, null, "/some/where/file", null, null);
	assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
	}

	@Test
	public void testRelativizeSimplePathChild() throws Exception {
	URI base = new URI("http", "example.com", "/some/where", null, null);
	URI uri = new URI("http://example.com/some/where/file");
	URI golden = new URI(null, null, "where/file", null, null);
	assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
	}

	@Test
	public void testRelativizeSimplePathParent() throws Exception {
	URI base = new URI("http", "example.com", "/some/where/file", null, null);
	URI uri = new URI("http://example.com/some/where");
	// Note that "." or "./" would not suffice, since we need "where" to not end
	// with a slash.
	URI golden = new URI(null, null, "../where", null, null);
	assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
	}

	@Test
	public void testRelativizeComplexPathFileBaseFolder()
	throws Exception {
	URI base = new URI("http", "example.com", "/some/there/", null, null);
	URI uri = new URI("http://example.com/some/where/file");
	URI golden = new URI(null, null, "../where/file", null, null);
	assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
	}

	@Test
	public void testRelativizeComplexPathFileBaseFile() throws Exception {
	URI base = new URI("http", "example.com", "/some/there/file", null, null);
	URI uri = new URI("http://example.com/some/where/file");
	URI golden = new URI(null, null, "../where/file", null, null);
	assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
	}

	@Test
	public void testRelativizeComplexPathFolderBaseFolder()
	throws Exception {
	URI base = new URI("http", "example.com", "/some/there/", null, null);
	URI uri = new URI("http://example.com/some/where/folder/");
	URI golden = new URI(null, null, "../where/folder/", null, null);
	assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
	}

	@Test
	public void testRelativizeComplexPathFolderBaseFile() throws Exception {
	URI base = new URI("http", "example.com", "/some/there/file", null, null);
	URI uri = new URI("http://example.com/some/where/folder/");
	URI golden = new URI(null, null, "../where/folder/", null, null);
	assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
	}

	@Test
	public void testRelativizeSame() throws Exception {
	URI uri = new URI("http://example.com/some/where/file");
	URI golden = new URI(null, null, null, null, "");
	assertEquals(golden, HtmlResponseWriter.relativize(uri, uri));
	}

	@Test
	public void testRelativizeNoMistakenScheme() throws Exception {
	URI base = new URI("http", "example.com", "/", null, null);
	URI uri = new URI("http://example.com/http://google.com/");
	URI golden = new URI(null, null, "./http://google.com/", null, null);
	assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
	}

	@Test
	public void testRelativizeNoMistakenSchemeNoSlash() throws Exception {
	URI base = new URI("http", "example.com", "/", null, null);
	URI uri = new URI("http://example.com/http:");
	URI golden = new URI(null, null, "./http:", null, null);
	assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
	}

	@Test
	public void testRelativizeNoMistakenSchemeSlashFirst() throws Exception {
	URI base = new URI("http", "example.com", "/", null, null);
	URI uri = new URI("http://example.com/http/:");
	URI golden = new URI(null, null, "http/:", null, null);
	assertEquals(golden, HtmlResponseWriter.relativize(base, uri));
	}

	@Test
	public void testRelativizeSchemeImpliesAbsoluteUriAssumption()
	throws Exception {
	thrown.expect(URISyntaxException.class);
	new URI("http", "example.com", "doesnt/start/with/a/slash", null, null);
	}
	}