org.argeo.cms/ext/test/org/argeo/cms/tabular/JcrTabularTest.java

   1 /*
   2  * Copyright (C) 2007-2012 Argeo GmbH
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *         http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package org.argeo.cms.tabular;
  17
  18 import java.io.InputStreamReader;
  19 import java.util.ArrayList;
  20 import java.util.List;
  21
  22 import javax.jcr.Node;
  23 import javax.jcr.PropertyType;
  24
  25 import org.apache.commons.logging.Log;
  26 import org.apache.commons.logging.LogFactory;
  27 import org.apache.jackrabbit.commons.cnd.CndImporter;
  28 import org.argeo.jackrabbit.unit.AbstractJackrabbitTestCase;
  29 import org.argeo.node.ArgeoNames;
  30 import org.argeo.node.ArgeoTypes;
  31 import org.argeo.node.tabular.TabularColumn;
  32 import org.argeo.node.tabular.TabularRow;
  33 import org.argeo.node.tabular.TabularRowIterator;
  34 import org.argeo.node.tabular.TabularWriter;
  35
  36 public class JcrTabularTest extends AbstractJackrabbitTestCase {
  37         private final static Log log = LogFactory.getLog(JcrTabularTest.class);
  38
  39         public void testWriteReadCsv() throws Exception {
  40                 session().setNamespacePrefix("argeo", ArgeoNames.ARGEO_NAMESPACE);
  41                 InputStreamReader reader = new InputStreamReader(getClass()
  42                                 .getResourceAsStream("/org/argeo/node/node.cnd"));
  43                 CndImporter.registerNodeTypes(reader, session());
  44                 reader.close();
  45
  46                 // write
  47                 Integer columnCount = 15;
  48                 Long rowCount = 1000l;
  49                 String stringValue = "test, \ntest";
  50
  51                 List<TabularColumn> header = new ArrayList<TabularColumn>();
  52                 for (int i = 0; i < columnCount; i++) {
  53                         header.add(new TabularColumn("col" + i, PropertyType.STRING));
  54                 }
  55                 Node tableNode = session().getRootNode().addNode("table",
  56                                 ArgeoTypes.ARGEO_TABLE);
  57                 TabularWriter writer = new JcrTabularWriter(tableNode, header,
  58                                 ArgeoTypes.ARGEO_CSV);
  59                 for (int i = 0; i < rowCount; i++) {
  60                         List<Object> objs = new ArrayList<Object>();
  61                         for (int j = 0; j < columnCount; j++) {
  62                                 objs.add(stringValue);
  63                         }
  64                         writer.appendRow(objs.toArray());
  65                 }
  66                 writer.close();
  67                 session().save();
  68
  69                 if (log.isDebugEnabled())
  70                         log.debug("Wrote tabular content " + rowCount + " rows, "
  71                                         + columnCount + " columns");
  72                 // read
  73                 TabularRowIterator rowIt = new JcrTabularRowIterator(tableNode);
  74                 Long count = 0l;
  75                 while (rowIt.hasNext()) {
  76                         TabularRow tr = rowIt.next();
  77                         assertEquals(header.size(), tr.size());
  78                         count++;
  79                 }
  80                 assertEquals(rowCount, count);
  81                 if (log.isDebugEnabled())
  82                         log.debug("Read tabular content " + rowCount + " rows, "
  83                                         + columnCount + " columns");
  84         }
  85 }