test/com/google/enterprise/adaptor/examples/MetaTaggerTransformTest.java - plexi - Git at Google

 // Copyright 2011 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.examples;

 import static org.junit.Assert.*;

 import com.google.enterprise.adaptor.Metadata;
 import com.google.enterprise.adaptor.TransformException;

 import org.junit.Test;

 import java.io.ByteArrayOutputStream;
 import java.io.IOException;
 import java.util.HashMap;
 import java.util.Map;

 /**
  * Tests for {@link MetaTaggerTransform}.
  */
 public class MetaTaggerTransformTest {
   private static final String TEST_DIR = "test/com/google/enterprise/adaptor/";

   @Test
   public void testNoInput() throws IOException, TransformException {
     MetaTaggerTransform transform = new MetaTaggerTransform();
     ByteArrayOutputStream contentIn = new ByteArrayOutputStream();
     ByteArrayOutputStream contentOut = new ByteArrayOutputStream();
     Metadata metadata = new Metadata();
     Map<String, String> params = new HashMap<String, String>();
     params.put("key1", "value1");

     String testString = "";
     contentIn.write(testString.getBytes());
     transform.transform(contentIn, contentOut, metadata, params);

     assertEquals(testString, contentIn.toString());
     assertEquals(testString, contentOut.toString());
     assertTrue(metadata.isEmpty());
     assertEquals("value1", params.get("key1"));
     assertEquals(1, params.keySet().size());
   }

   @Test
   public void testNoPattern() throws IOException, TransformException {
     MetaTaggerTransform transform = new MetaTaggerTransform();
     ByteArrayOutputStream contentIn = new ByteArrayOutputStream();
     ByteArrayOutputStream contentOut = new ByteArrayOutputStream();
     Metadata metadata = new Metadata();
     Map<String, String> params = new HashMap<String, String>();
     params.put("key1", "value1");

     String testString = "Here is some input";
     contentIn.write(testString.getBytes());
     transform.transform(contentIn, contentOut, metadata, params);

     assertEquals(testString, contentIn.toString());
     assertEquals(testString, contentOut.toString());
     assertTrue(metadata.isEmpty());
     assertEquals("value1", params.get("key1"));
     assertEquals(1, params.keySet().size());
   }

   @Test
   public void testSimple() throws IOException, TransformException {
     MetaTaggerTransform transform = new MetaTaggerTransform(TEST_DIR + "testPattern1.txt");
     ByteArrayOutputStream contentIn = new ByteArrayOutputStream();
     ByteArrayOutputStream contentOut = new ByteArrayOutputStream();
     Metadata metadata = new Metadata();
     Map<String, String> params = new HashMap<String, String>();
     params.put("key1", "value1");
     String content =
         "<HTML>\n" +
         "<HEAD></head>\n" +
         "<BODY>\n" +
         "Today, John Paul gave a speach at local animal shelter. Animal lovers rejoice.\n" +
         "</BODY>\n" +
         "</HTML>\n";
     String goldenContent =
         "<HTML>\n" +
         "<HEAD>\n" +
         "<meta name=\"pope\" content=\"John Paul 2nd\" />\n" +
         "<meta name=\"city\" content=\"Mountain View\" />\n" +
         "</HEAD>\n" +
         "<BODY>\n" +
         "Today, John Paul gave a speach at local animal shelter. Animal lovers rejoice.\n" +
         "</BODY>\n" +
         "</HTML>\n";
     contentIn.write(content.getBytes());
     transform.transform(contentIn, contentOut, metadata, params);

     assertEquals(goldenContent, contentOut.toString());
     assertTrue(metadata.isEmpty());
     assertEquals("value1", params.get("key1"));
     assertEquals(1, params.keySet().size());
   }

   @Test
   public void testNoHead() throws IOException, TransformException {
     MetaTaggerTransform transform = new MetaTaggerTransform(TEST_DIR + "testPattern1.txt");
     ByteArrayOutputStream contentIn = new ByteArrayOutputStream();
     ByteArrayOutputStream contentOut = new ByteArrayOutputStream();
     Metadata metadata = new Metadata();
     Map<String, String> params = new HashMap<String, String>();
     params.put("key1", "value1");
     String content =
         "This is a document with no head element.\n" +
         "If there were a HEAD element, then the\n" +
         "transform would be inserting metadata somewhere in this doc.\n" +
         "  We should end up with the same output as input.\n";
     contentIn.write(content.getBytes());
     transform.transform(contentIn, contentOut, metadata, params);

     assertEquals(content, contentOut.toString());
     assertTrue(metadata.isEmpty());
     assertEquals("value1", params.get("key1"));
     assertEquals(1, params.keySet().size());
   }

   @Test
   public void testDuplicatePatternsInPatternFile() throws IOException, TransformException {
     MetaTaggerTransform transform = new MetaTaggerTransform(TEST_DIR + "testPatternDup.txt");
     ByteArrayOutputStream contentIn = new ByteArrayOutputStream();
     ByteArrayOutputStream contentOut = new ByteArrayOutputStream();
     Metadata metadata = new Metadata();
     Map<String, String> params = new HashMap<String, String>();
     params.put("key1", "value1");

     String content =
         "<HTML>\n" +
         "<HEAD></head>\n" +
         "<BODY>\n" +
         "Today, John Paul gave a speach at local animal shelter. Animal lovers rejoice.\n" +
         "</BODY>\n" +
         "</HTML>\n";
     String goldenContent = "<HTML>\n" +
         "<HEAD>\n" +
         "<meta name=\"pope\" content=\"John Paul 2nd\" />\n" +
         "<meta name=\"pope\" content=\"John Paul 3rd\" />\n" +
         "<meta name=\"city\" content=\"Mountain View\" />\n" +
         "</HEAD>\n" +
         "<BODY>\n" +
         "Today, John Paul gave a speach at local animal shelter. Animal lovers rejoice.\n" +
         "</BODY>\n" +
         "</HTML>\n";
     contentIn.write(content.getBytes());
     transform.transform(contentIn, contentOut, metadata, params);

     assertEquals(goldenContent, contentOut.toString());
     assertTrue(metadata.isEmpty());
     assertEquals("value1", params.get("key1"));
     assertEquals(1, params.keySet().size());
   }
 }
	// Copyright 2011 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.examples;

	import static org.junit.Assert.*;

	import com.google.enterprise.adaptor.Metadata;
	import com.google.enterprise.adaptor.TransformException;

	import org.junit.Test;

	import java.io.ByteArrayOutputStream;
	import java.io.IOException;
	import java.util.HashMap;
	import java.util.Map;

	/**
	* Tests for {@link MetaTaggerTransform}.
	*/
	public class MetaTaggerTransformTest {
	private static final String TEST_DIR = "test/com/google/enterprise/adaptor/";

	@Test
	public void testNoInput() throws IOException, TransformException {
	MetaTaggerTransform transform = new MetaTaggerTransform();
	ByteArrayOutputStream contentIn = new ByteArrayOutputStream();
	ByteArrayOutputStream contentOut = new ByteArrayOutputStream();
	Metadata metadata = new Metadata();
	Map<String, String> params = new HashMap<String, String>();
	params.put("key1", "value1");

	String testString = "";
	contentIn.write(testString.getBytes());
	transform.transform(contentIn, contentOut, metadata, params);

	assertEquals(testString, contentIn.toString());
	assertEquals(testString, contentOut.toString());
	assertTrue(metadata.isEmpty());
	assertEquals("value1", params.get("key1"));
	assertEquals(1, params.keySet().size());
	}

	@Test
	public void testNoPattern() throws IOException, TransformException {
	MetaTaggerTransform transform = new MetaTaggerTransform();
	ByteArrayOutputStream contentIn = new ByteArrayOutputStream();
	ByteArrayOutputStream contentOut = new ByteArrayOutputStream();
	Metadata metadata = new Metadata();
	Map<String, String> params = new HashMap<String, String>();
	params.put("key1", "value1");

	String testString = "Here is some input";
	contentIn.write(testString.getBytes());
	transform.transform(contentIn, contentOut, metadata, params);

	assertEquals(testString, contentIn.toString());
	assertEquals(testString, contentOut.toString());
	assertTrue(metadata.isEmpty());
	assertEquals("value1", params.get("key1"));
	assertEquals(1, params.keySet().size());
	}

	@Test
	public void testSimple() throws IOException, TransformException {
	MetaTaggerTransform transform = new MetaTaggerTransform(TEST_DIR + "testPattern1.txt");
	ByteArrayOutputStream contentIn = new ByteArrayOutputStream();
	ByteArrayOutputStream contentOut = new ByteArrayOutputStream();
	Metadata metadata = new Metadata();
	Map<String, String> params = new HashMap<String, String>();
	params.put("key1", "value1");
	String content =
	"<HTML>\n" +
	"<HEAD></head>\n" +
	"<BODY>\n" +
	"Today, John Paul gave a speach at local animal shelter. Animal lovers rejoice.\n" +
	"</BODY>\n" +
	"</HTML>\n";
	String goldenContent =
	"<HTML>\n" +
	"<HEAD>\n" +
	"<meta name=\"pope\" content=\"John Paul 2nd\" />\n" +
	"<meta name=\"city\" content=\"Mountain View\" />\n" +
	"</HEAD>\n" +
	"<BODY>\n" +
	"Today, John Paul gave a speach at local animal shelter. Animal lovers rejoice.\n" +
	"</BODY>\n" +
	"</HTML>\n";
	contentIn.write(content.getBytes());
	transform.transform(contentIn, contentOut, metadata, params);

	assertEquals(goldenContent, contentOut.toString());
	assertTrue(metadata.isEmpty());
	assertEquals("value1", params.get("key1"));
	assertEquals(1, params.keySet().size());
	}

	@Test
	public void testNoHead() throws IOException, TransformException {
	MetaTaggerTransform transform = new MetaTaggerTransform(TEST_DIR + "testPattern1.txt");
	ByteArrayOutputStream contentIn = new ByteArrayOutputStream();
	ByteArrayOutputStream contentOut = new ByteArrayOutputStream();
	Metadata metadata = new Metadata();
	Map<String, String> params = new HashMap<String, String>();
	params.put("key1", "value1");
	String content =
	"This is a document with no head element.\n" +
	"If there were a HEAD element, then the\n" +
	"transform would be inserting metadata somewhere in this doc.\n" +
	" We should end up with the same output as input.\n";
	contentIn.write(content.getBytes());
	transform.transform(contentIn, contentOut, metadata, params);

	assertEquals(content, contentOut.toString());
	assertTrue(metadata.isEmpty());
	assertEquals("value1", params.get("key1"));
	assertEquals(1, params.keySet().size());
	}

	@Test
	public void testDuplicatePatternsInPatternFile() throws IOException, TransformException {
	MetaTaggerTransform transform = new MetaTaggerTransform(TEST_DIR + "testPatternDup.txt");
	ByteArrayOutputStream contentIn = new ByteArrayOutputStream();
	ByteArrayOutputStream contentOut = new ByteArrayOutputStream();
	Metadata metadata = new Metadata();
	Map<String, String> params = new HashMap<String, String>();
	params.put("key1", "value1");

	String content =
	"<HTML>\n" +
	"<HEAD></head>\n" +
	"<BODY>\n" +
	"Today, John Paul gave a speach at local animal shelter. Animal lovers rejoice.\n" +
	"</BODY>\n" +
	"</HTML>\n";
	String goldenContent = "<HTML>\n" +
	"<HEAD>\n" +
	"<meta name=\"pope\" content=\"John Paul 2nd\" />\n" +
	"<meta name=\"pope\" content=\"John Paul 3rd\" />\n" +
	"<meta name=\"city\" content=\"Mountain View\" />\n" +
	"</HEAD>\n" +
	"<BODY>\n" +
	"Today, John Paul gave a speach at local animal shelter. Animal lovers rejoice.\n" +
	"</BODY>\n" +
	"</HTML>\n";
	contentIn.write(content.getBytes());
	transform.transform(contentIn, contentOut, metadata, params);

	assertEquals(goldenContent, contentOut.toString());
	assertTrue(metadata.isEmpty());
	assertEquals("value1", params.get("key1"));
	assertEquals(1, params.keySet().size());
	}
	}