svn commit: r823451 - /lucene/solr/trunk/contrib/dataimporthandler/src/test/java/org/apache/solr/handler/dataimport/TestXPathEntityProcessor.java

[noble]

Author: noble
Date: Fri Oct  9 08:23:35 2009
New Revision: 823451

URL: http://svn.apache.org/viewvc?rev=823451&view=rev
Log:
testcase for flatten=true

Modified:
    
lucene/solr/trunk/contrib/dataimporthandler/src/test/java/org/apache/solr/handler/dataimport/TestXPathEntityProcessor.java

Modified: 
lucene/solr/trunk/contrib/dataimporthandler/src/test/java/org/apache/solr/handler/dataimport/TestXPathEntityProcessor.java
URL: 
http://svn.apache.org/viewvc/lucene/solr/trunk/contrib/dataimporthandler/src/test/java/org/apache/solr/handler/dataimport/TestXPathEntityProcessor.java?rev=823451&r1=823450&r2=823451&view=diff
==============================================================================
--- 
lucene/solr/trunk/contrib/dataimporthandler/src/test/java/org/apache/solr/handler/dataimport/TestXPathEntityProcessor.java
 (original)
+++ 
lucene/solr/trunk/contrib/dataimporthandler/src/test/java/org/apache/solr/handler/dataimport/TestXPathEntityProcessor.java
 Fri Oct  9 08:23:35 2009
@@ -89,6 +89,26 @@
   }
 
   @Test
+  public void testMultiValuedFlatten() throws Exception  {
+    Map entityAttrs = createMap("name", "e", "url", "testdata.xml",
+            XPathEntityProcessor.FOR_EACH, "/root");
+    List fields = new ArrayList();
+    fields.add(createMap("column", "a", "xpath", "/root/a" ,"flatten","true"));
+    Context c = AbstractDataImportHandlerTest.getContext(null,
+            new VariableResolverImpl(), getDataSource(testXmlFlatten), 
Context.FULL_DUMP, fields, entityAttrs);
+    XPathEntityProcessor xPathEntityProcessor = new XPathEntityProcessor();
+    xPathEntityProcessor.init(c);
+    Map<String, Object> result = null;
+    while (true) {
+      Map<String, Object> row = xPathEntityProcessor.nextRow();
+      if (row == null)
+        break;
+      result = row;
+    }
+    Assert.assertEquals("1B2", result.get("a"));
+  }
+
+  @Test
   public void withFieldsAndXpathStream() throws Exception {
     Map entityAttrs = createMap("name", "e", "url", "cd.xml",
         XPathEntityProcessor.FOR_EACH, "/catalog/cd", "stream", "true", 
"batchSize","1");
@@ -204,4 +224,6 @@
           + "\t\t<year>1982</year>\n" + "\t</cd>\n" + "</catalog>\t";
 
   private static final String testXml = "<?xml version=\"1.0\" 
encoding=\"UTF-8\"?><root><a>1</a><a>2</a></root>";
+
+  private static final String testXmlFlatten = "<?xml version=\"1.0\" 
encoding=\"UTF-8\"?><root><a>1<b>B</b>2</a></root>";
 }