123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196 |
- /*
- Copyright (c) 2007 Health Market Science, Inc.
-
- This library is free software; you can redistribute it and/or
- modify it under the terms of the GNU Lesser General Public
- License as published by the Free Software Foundation; either
- version 2.1 of the License, or (at your option) any later version.
-
- This library is distributed in the hope that it will be useful,
- but WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- Lesser General Public License for more details.
-
- You should have received a copy of the GNU Lesser General Public
- License along with this library; if not, write to the Free Software
- Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
- USA
-
- You can contact Health Market Science at info@healthmarketscience.com
- or at the following address:
-
- Health Market Science
- 2700 Horizon Drive
- Suite 200
- King of Prussia, PA 19406
- */
-
- package com.healthmarketscience.jackcess;
-
- import java.io.IOException;
- import java.nio.ByteBuffer;
- import java.nio.charset.Charset;
- import java.util.ArrayList;
- import java.util.Arrays;
- import java.util.List;
- import java.util.TimeZone;
-
- import junit.framework.TestCase;
-
- /**
- * @author Tim McCune
- */
- public class TableTest extends TestCase {
-
- private final PageChannel _pageChannel = new PageChannel(true);
- private List<Column> _columns = new ArrayList<Column>();
- private Table _testTable;
-
- public TableTest(String name) {
- super(name);
- }
-
- public void testCreateRow() throws Exception {
- Column col = newTestColumn();
- col.setType(DataType.INT);
- _columns.add(col);
- col = newTestColumn();
- col.setType(DataType.TEXT);
- _columns.add(col);
- col = newTestColumn();
- col.setType(DataType.TEXT);
- _columns.add(col);
- newTestTable();
-
- int colCount = _columns.size();
- ByteBuffer buffer = createRow(9, "Tim", "McCune");
-
- assertEquals((short) colCount, buffer.getShort());
- assertEquals((short) 9, buffer.getShort());
- assertEquals((byte) 'T', buffer.get());
- assertEquals((short) 22, buffer.getShort(22));
- assertEquals((short) 10, buffer.getShort(24));
- assertEquals((short) 4, buffer.getShort(26));
- assertEquals((short) 2, buffer.getShort(28));
- assertEquals((byte) 7, buffer.get(30));
- }
-
- public void testUnicodeCompression() throws Exception {
- Column col = newTestColumn();
- col = newTestColumn();
- col.setType(DataType.TEXT);
- _columns.add(col);
- col = newTestColumn();
- col.setType(DataType.MEMO);
- _columns.add(col);
- newTestTable();
-
- String small = "this is a string";
- String smallNotAscii = "this is a string\0";
- String large = DatabaseTest.createString(30);
- String largeNotAscii = large + "\0";
-
- ByteBuffer[] buf1 = encodeColumns(small, large);
- ByteBuffer[] buf2 = encodeColumns(smallNotAscii, largeNotAscii);
-
- for(Column tmp : _columns) {
- tmp.setCompressedUnicode(true);
- }
-
- ByteBuffer[] bufCmp1 = encodeColumns(small, large);
- ByteBuffer[] bufCmp2 = encodeColumns(smallNotAscii, largeNotAscii);
-
- assertEquals(buf1[0].remaining(),
- (bufCmp1[0].remaining() + small.length() - 2));
- assertEquals(buf1[1].remaining(),
- (bufCmp1[1].remaining() + large.length() - 2));
-
- for(int i = 0; i < buf2.length; ++i) {
- assertTrue(Arrays.equals(toBytes(buf2[i]), toBytes(bufCmp2[i])));
- }
-
- assertEquals(Arrays.asList(small, large),
- Arrays.asList(decodeColumns(bufCmp1)));
- assertEquals(Arrays.asList(smallNotAscii, largeNotAscii),
- Arrays.asList(decodeColumns(bufCmp2)));
-
- }
-
- private ByteBuffer createRow(Object... row)
- throws IOException
- {
- return _testTable.createRow(
- row, _testTable.getPageChannel().createPageBuffer());
- }
-
- private ByteBuffer[] encodeColumns(Object... row)
- throws IOException
- {
- ByteBuffer[] result = new ByteBuffer[_columns.size()];
- for(int i = 0; i < _columns.size(); ++i) {
- Column col = _columns.get(i);
- result[i] = col.write(row[i], _testTable.getFormat().MAX_ROW_SIZE);
- }
- return result;
- }
-
- private Object[] decodeColumns(ByteBuffer[] buffers)
- throws IOException
- {
- Object[] result = new Object[_columns.size()];
- for(int i = 0; i < _columns.size(); ++i) {
- Column col = _columns.get(i);
- result[i] = col.read(toBytes(buffers[i]));
- }
- return result;
- }
-
- private static byte[] toBytes(ByteBuffer buffer) {
- buffer.rewind();
- byte[] b = new byte[buffer.remaining()];
- buffer.get(b);
- return b;
- }
-
- private Table newTestTable()
- throws Exception
- {
- _testTable = new Table(true, _columns) {
- @Override
- public PageChannel getPageChannel() {
- return _pageChannel;
- }
- @Override
- public JetFormat getFormat() {
- return JetFormat.VERSION_4;
- }
- };
- return _testTable;
- }
-
- private Column newTestColumn() {
- return new Column(true, null) {
- @Override
- public Table getTable() {
- return _testTable;
- }
- @Override
- public JetFormat getFormat() {
- return getTable().getFormat();
- }
- @Override
- public PageChannel getPageChannel() {
- return getTable().getPageChannel();
- }
- @Override
- protected Charset getCharset() {
- return getFormat().CHARSET;
- }
- @Override
- protected TimeZone getTimeZone() {
- return TimeZone.getDefault();
- }
- };
- }
-
- }
|