package org.apache.cassandra.thrift;
/*
*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF 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.
*
*/
import java.nio.ByteBuffer;
import org.junit.Test;
import org.apache.cassandra.SchemaLoader;
import org.apache.cassandra.config.CFMetaData;
import org.apache.cassandra.config.ConfigurationException;
import org.apache.cassandra.config.ColumnDefinition;
import org.apache.cassandra.config.KSMetaData;
import org.apache.cassandra.config.Schema;
import org.apache.cassandra.db.marshal.AsciiType;
import org.apache.cassandra.db.marshal.UTF8Type;
import org.apache.cassandra.locator.LocalStrategy;
import org.apache.cassandra.locator.NetworkTopologyStrategy;
import org.apache.cassandra.utils.FBUtilities;
public class ThriftValidationTest extends SchemaLoader
{
@Test(expected=InvalidRequestException.class)
public void testValidateCommutativeWithStandard() throws InvalidRequestException
{
ThriftValidation.validateColumnFamily("Keyspace1", "Standard1", true);
}
@Test
public void testValidateCommutativeWithCounter() throws InvalidRequestException
{
ThriftValidation.validateColumnFamily("Keyspace1", "Counter1", true);
}
@Test
public void testColumnValueSizeForIndexedColumn() throws ConfigurationException, InvalidRequestException
{
CfDef cfDef = Schema.instance.getCFMetaData("Keyspace1", "Standard1").toThrift();
ByteBuffer columnName = AsciiType.instance.fromString("indexed");
// add an indexed column definition
cfDef.addToColumn_metadata(new ColumnDef(columnName, UTF8Type.class.getCanonicalName())
.setIndex_type(IndexType.KEYS)
.setIndex_name("indexed_col"));
CFMetaData metaData = CFMetaData.fromThrift(cfDef);
Column column = new Column(columnName)
.setValue(new byte[FBUtilities.MAX_UNSIGNED_SHORT + 1])
.setTimestamp(System.currentTimeMillis());
boolean gotException = false;
try
{
// this run should throw an exception
ThriftValidation.validateColumnData(metaData, column, false);
}
catch (InvalidRequestException e)
{
gotException = true;
}
assert gotException : "expected InvalidRequestException but not received.";
// change value to be less than unsigned short size
column.setValue(new byte[12]);
gotException = false; // reset flag
try
{
// this run should run clean
ThriftValidation.validateColumnData(metaData, column, false);
}
catch (InvalidRequestException e)
{
gotException = true;
}
assert !gotException : "got unexpected InvalidRequestException";
}
@Test
public void testColumnNameEqualToKeyAlias()
{
CFMetaData metaData = Schema.instance.getCFMetaData("Keyspace1", "Standard1");
CFMetaData newMetadata = metaData.clone();
boolean gotException = false;
// add a key_alias = "id"
newMetadata.keyAlias(AsciiType.instance.decompose("id"));
// should not throw IRE here
try
{
newMetadata.validate();
}
catch (ConfigurationException e)
{
gotException = true;
}
assert !gotException : "got unexpected ConfigurationException";
// add a column with name = "id"
newMetadata.addColumnDefinition(ColumnDefinition.utf8("id", null));
gotException = false;
try
{
newMetadata.validate();
}
catch (ConfigurationException e)
{
gotException = true;
}
assert gotException : "expected ConfigurationException but not received.";
}
@Test
public void testValidateKsDef()
{
KsDef ks_def = new KsDef()
.setName("keyspaceValid")
.setStrategy_class(LocalStrategy.class.getSimpleName());
boolean gotException = false;
try
{
KSMetaData.fromThrift(ks_def).validate();
}
catch (ConfigurationException e)
{
gotException = true;
}
assert gotException : "expected ConfigurationException but not received.";
ks_def.setStrategy_class(LocalStrategy.class.getName());
gotException = false;
try
{
KSMetaData.fromThrift(ks_def).validate();
}
catch (ConfigurationException e)
{
gotException = true;
}
assert gotException : "expected ConfigurationException but not received.";
ks_def.setStrategy_class(NetworkTopologyStrategy.class.getName());
gotException = false;
try
{
KSMetaData.fromThrift(ks_def).validate();
}
catch (ConfigurationException e)
{
gotException = true;
}
assert !gotException : "got unexpected ConfigurationException";
}
}