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.cms.ArgeoTypes;
  29 import org.argeo.jackrabbit.unit.AbstractJackrabbitTestCase;
  30 import org.argeo.node.tabular.TabularColumn;
  31 import org.argeo.node.tabular.TabularRow;
  32 import org.argeo.node.tabular.TabularRowIterator;
  33 import org.argeo.node.tabular.TabularWriter;
  34
  35 public class JcrTabularTest extends AbstractJackrabbitTestCase {
  36         private final static Log log = LogFactory.getLog(JcrTabularTest.class);
  37
  38         public void testWriteReadCsv() throws Exception {
  39                 // session().setNamespacePrefix("argeo", ArgeoNames.ARGEO_NAMESPACE);
  40                 InputStreamReader reader = new InputStreamReader(getClass().getResourceAsStream("/org/argeo/node/node.cnd"));
  41                 CndImporter.registerNodeTypes(reader, session());
  42                 reader.close();
  43                 reader = new InputStreamReader(getClass().getResourceAsStream("/org/argeo/cms/argeo.cnd"));
  44                 CndImporter.registerNodeTypes(reader, session());
  45                 reader.close();
  46 //              reader = new InputStreamReader(getClass().getResourceAsStream("/org/argeo/cms/cms.cnd"));
  47 //              CndImporter.registerNodeTypes(reader, session());
  48 //              reader.close();
  49
  50                 // write
  51                 Integer columnCount = 15;
  52                 Long rowCount = 1000l;
  53                 String stringValue = "test, \ntest";
  54
  55                 List<TabularColumn> header = new ArrayList<TabularColumn>();
  56                 for (int i = 0; i < columnCount; i++) {
  57                         header.add(new TabularColumn("col" + i, PropertyType.STRING));
  58                 }
  59                 Node tableNode = session().getRootNode().addNode("table", ArgeoTypes.ARGEO_TABLE);
  60                 TabularWriter writer = new JcrTabularWriter(tableNode, header, ArgeoTypes.ARGEO_CSV);
  61                 for (int i = 0; i < rowCount; i++) {
  62                         List<Object> objs = new ArrayList<Object>();
  63                         for (int j = 0; j < columnCount; j++) {
  64                                 objs.add(stringValue);
  65                         }
  66                         writer.appendRow(objs.toArray());
  67                 }
  68                 writer.close();
  69                 session().save();
  70
  71                 if (log.isDebugEnabled())
  72                         log.debug("Wrote tabular content " + rowCount + " rows, " + columnCount + " columns");
  73                 // read
  74                 TabularRowIterator rowIt = new JcrTabularRowIterator(tableNode);
  75                 Long count = 0l;
  76                 while (rowIt.hasNext()) {
  77                         TabularRow tr = rowIt.next();
  78                         assertEquals(header.size(), tr.size());
  79                         count++;
  80                 }
  81                 assertEquals(rowCount, count);
  82                 if (log.isDebugEnabled())
  83                         log.debug("Read tabular content " + rowCount + " rows, " + columnCount + " columns");
  84         }
  85 }