/*
* Licensed to STRATIO (C) under one or more contributor license agreements.
* See the NOTICE file distributed with this work for additional information
* regarding copyright ownership. The STRATIO (C) licenses this file
* to you 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.stratio.cassandra.lucene.column;
import org.apache.cassandra.db.marshal.UTF8Type;
import org.junit.Test;
import java.util.ArrayList;
import java.util.List;
import static org.junit.Assert.assertEquals;
/**
* @author Andres de la Pena {@literal <adelapena@stratio.com>}
*/
public class ColumnsTest {
@Test
public void testBuild() {
Columns columns = new Columns();
assertEquals(0, columns.size());
}
@Test
public void testAdd() {
Columns columns = new Columns();
columns.add(Column.builder("field1").buildWithComposed("value1", UTF8Type.instance));
columns.add(Column.builder("field2").buildWithComposed("value2", UTF8Type.instance));
assertEquals("Columns size is wrong", 2, columns.size());
}
@Test
public void testAddAll() {
Columns columns = new Columns().add(Column.builder("field1").buildWithComposed("value1", UTF8Type.instance))
.add(Column.builder("field2").buildWithComposed("value2", UTF8Type.instance));
assertEquals("Columns size is wrong", 2, columns.size());
}
@Test
public void testIterator() {
Columns columns = new Columns().add(Column.builder("field1").buildWithComposed("value1", UTF8Type.instance))
.add(Column.builder("field2").buildWithComposed("value2", UTF8Type.instance));
List<Column<?>> columnList = new ArrayList<>();
for (Column<?> column : columns) {
columnList.add(column);
}
assertEquals("Columns size is wrong", 2, columnList.size());
}
@Test
public void testGetColumnsByCellName() {
Columns columns = new Columns();
columns.add(Column.builder("field1").buildWithComposed("value", UTF8Type.instance));
columns.add(Column.builder("field1").withUDTName("1").buildWithComposed("value", UTF8Type.instance));
columns.add(Column.builder("field1").withMapName("1").buildWithComposed("value", UTF8Type.instance));
columns.add(Column.builder("field1")
.withUDTName("1")
.withMapName("1")
.buildWithComposed("value", UTF8Type.instance));
assertEquals("Columns size is wrong", 4, columns.getColumnsByCellName("field1").size());
assertEquals("Columns size is wrong", 4, columns.getColumnsByCellName("field1.1").size());
assertEquals("Columns size is wrong", 4, columns.getColumnsByCellName("field1$1").size());
assertEquals("Columns size is wrong", 4, columns.getColumnsByCellName("field1.1$1$2").size());
assertEquals("Columns size is wrong", 4, columns.getColumnsByCellName("field1.2").size());
assertEquals("Columns size is wrong", 0, columns.getColumnsByCellName("field2").size());
}
@Test
public void testGetColumnsByFullName() {
Columns columns = new Columns();
columns.add(Column.builder("field1").buildWithComposed("value", UTF8Type.instance));
columns.add(Column.builder("field1").withUDTName("1").buildWithComposed("value", UTF8Type.instance));
columns.add(Column.builder("field1").withMapName("1").buildWithComposed("value", UTF8Type.instance));
columns.add(Column.builder("field1")
.withUDTName("1")
.withMapName("1")
.buildWithComposed("value", UTF8Type.instance));
columns.add(Column.builder("field1")
.withUDTName("1")
.withUDTName("1")
.buildWithComposed("value", UTF8Type.instance));
columns.add(Column.builder("field1")
.withMapName("1")
.withMapName("1")
.buildWithComposed("value", UTF8Type.instance));
assertEquals("Columns size is wrong", 1, columns.getColumnsByFullName("field1").size());
assertEquals("Columns size is wrong", 1, columns.getColumnsByFullName("field1.1").size());
assertEquals("Columns size is wrong", 1, columns.getColumnsByFullName("field1.1.1").size());
assertEquals("Columns size is wrong", 1, columns.getColumnsByFullName("field1$1").size());
assertEquals("Columns size is wrong", 1, columns.getColumnsByFullName("field1.1$1").size());
assertEquals("Columns size is wrong", 1, columns.getColumnsByFullName("field1$1$1").size());
assertEquals("Columns size is wrong", 0, columns.getColumnsByFullName("field2").size());
}
@Test
public void testGetColumnsByMapperName() {
Columns columns = new Columns();
columns.add(Column.builder("field1").buildWithComposed("value", UTF8Type.instance));
columns.add(Column.builder("field1").withUDTName("1").buildWithComposed("value", UTF8Type.instance));
columns.add(Column.builder("field1").withMapName("1").buildWithComposed("value", UTF8Type.instance));
columns.add(Column.builder("field1")
.withUDTName("1")
.withMapName("1")
.buildWithComposed("value", UTF8Type.instance));
columns.add(Column.builder("field1")
.withUDTName("1")
.withUDTName("1")
.buildWithComposed("value", UTF8Type.instance));
columns.add(Column.builder("field1")
.withMapName("1")
.withMapName("1")
.buildWithComposed("value", UTF8Type.instance));
assertEquals("Columns size is wrong", 3, columns.getColumnsByMapperName("field1").size());
assertEquals("Columns size is wrong", 2, columns.getColumnsByMapperName("field1.1").size());
assertEquals("Columns size is wrong", 3, columns.getColumnsByMapperName("field1$1").size());
assertEquals("Columns size is wrong", 2, columns.getColumnsByMapperName("field1.1$2").size());
assertEquals("Columns size is wrong", 1, columns.getColumnsByMapperName("field1.1.1").size());
assertEquals("Columns size is wrong", 0, columns.getColumnsByMapperName("field2").size());
}
@Test
public void testToString() {
assertEquals("Method #toString is wrong", "Columns{}", new Columns().toString());
}
@Test
public void testToStringWithColumns() {
Columns columns = new Columns();
columns.add(Column.builder("field1").buildWithComposed("value", UTF8Type.instance));
columns.add(Column.builder("field1").withUDTName("1").buildWithComposed("value", UTF8Type.instance));
columns.add(Column.builder("field1").withMapName("1").buildWithComposed("value", UTF8Type.instance));
columns.add(Column.builder("field1")
.withUDTName("1")
.withMapName("1")
.buildWithComposed("value", UTF8Type.instance));
columns.add(Column.builder("field1")
.withUDTName("1")
.withUDTName("1")
.buildWithComposed("value", UTF8Type.instance));
columns.add(Column.builder("field1")
.withMapName("1")
.withMapName("1")
.buildWithComposed("value", UTF8Type.instance));
assertEquals("Method #toString is wrong",
"Columns{field1=value, field1.1=value, field1$1=value, " +
"field1.1$1=value, field1.1.1=value, field1$1$1=value}",
columns.toString());
}
}