Package org.cassandraunit.dataset

Examples of org.cassandraunit.dataset.DataSet


    dataSet.getKeyspace();
  }

  @Test(expected = ParseException.class)
  public void shouldNotGetAYamlDataSetStructureBecauseOfInvalidDefaultColumnValueType() {
    DataSet dataSet = new ClassPathYamlDataSet("yaml/dataSetBadInvalidDefaultColumnValueType.yaml");
    dataSet.getKeyspace();
  }
View Full Code Here


    dataSet.getKeyspace();
  }

  @Test
  public void shouldGetAYamlDataSetWithData() {
    DataSet dataSet = new ClassPathYamlDataSet("yaml/dataSetDefaultValues.yaml");
    assertThat(dataSet, notNullValue());
    assertThat(dataSet.getKeyspace(), notNullValue());

    assertThat(dataSet.getColumnFamilies(), notNullValue());
    assertThat(dataSet.getColumnFamilies().size(), is(1));
    assertThat(dataSet.getColumnFamilies().get(0), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(0).getName(), is("columnFamily1"));
    assertThat(dataSet.getColumnFamilies().get(0).getRows(), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(0).getRows().size(), is(1));
    assertThat(dataSet.getColumnFamilies().get(0).getRows().get(0), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(0).getRows().get(0).getKey().getValue(), is("01"));
    assertThat(dataSet.getColumnFamilies().get(0).getRows().get(0).getKey().getType(),
        is(GenericTypeEnum.BYTES_TYPE));
    assertThat(dataSet.getColumnFamilies().get(0).getRows().get(0).getColumns(), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(0).getRows().get(0).getColumns().size(), is(1));
    assertThat(dataSet.getColumnFamilies().get(0).getRows().get(0).getColumns().get(0), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(0).getRows().get(0).getColumns().get(0).getName().getValue(),
        is("02"));
    assertThat(dataSet.getColumnFamilies().get(0).getRows().get(0).getColumns().get(0).getName().getType(),
        is(GenericTypeEnum.BYTES_TYPE));
    assertThat(dataSet.getColumnFamilies().get(0).getRows().get(0).getColumns().get(0).getValue().getValue(),
        is("03"));
    assertThat(dataSet.getColumnFamilies().get(0).getRows().get(0).getColumns().get(0).getValue().getType(),
        is(GenericTypeEnum.BYTES_TYPE));

  }
View Full Code Here

  }

  @Test
  public void shouldGetAYamlDataSetWithDefinedValuesAndData() {
    DataSet dataSet = new ClassPathYamlDataSet("yaml/dataSetDefinedValues.yaml");
    assertThat(dataSet, notNullValue());
    assertThat(dataSet.getKeyspace(), notNullValue());
    assertThat(dataSet.getKeyspace().getName(), is("beautifulDefinedKeyspaceName"));
    assertThat(dataSet.getKeyspace().getReplicationFactor(), is(1));
    assertThat(dataSet.getKeyspace().getStrategy(), is(StrategyModel.SIMPLE_STRATEGY));

    assertThat(dataSet.getColumnFamilies(), notNullValue());
    assertThat(dataSet.getColumnFamilies().size(), is(4));

        ColumnFamilyModel columnFamily1 = dataSet.getColumnFamilies().get(0);
        assertThat(columnFamily1, notNullValue());
    assertThat(columnFamily1.getName(), is("columnFamily1"));
    assertThat(columnFamily1.getKeyType().getTypeName(),
        is(ComparatorType.UTF8TYPE.getTypeName()));
    assertThat(columnFamily1.getComparatorType().getTypeName(),
        is(ComparatorType.UTF8TYPE.getTypeName()));
    assertThat(columnFamily1.getSubComparatorType(), is(nullValue()));
        assertThat(columnFamily1.getComment(),is("amazing comment"));
        assertThat(columnFamily1.getCompactionStrategy(),is("LeveledCompactionStrategy"));
        assertThat(columnFamily1.getCompactionStrategyOptions().get(0).getName(),is("sstable_size_in_mb"));
        assertThat(columnFamily1.getCompactionStrategyOptions().get(0).getValue(),is("10"));
        assertThat(columnFamily1.getGcGraceSeconds(),is(9999));
        assertThat(columnFamily1.getMaxCompactionThreshold(),is(31));
        assertThat(columnFamily1.getMinCompactionThreshold(),is(3));
        assertThat(columnFamily1.getReadRepairChance(),is(0.1d));
        assertThat(columnFamily1.getReplicationOnWrite(),is(Boolean.FALSE));

    assertThat(columnFamily1.getRows(), notNullValue());
    assertThat(columnFamily1.getRows().size(), is(1));
    assertThat(columnFamily1.getRows().get(0), notNullValue());
    assertThat(columnFamily1.getRows().get(0).getKey().getValue(), is("key01"));
    assertThat(columnFamily1.getRows().get(0).getKey().getType(),
        is(GenericTypeEnum.UTF_8_TYPE));
    assertThat(columnFamily1.getRows().get(0).getColumns(), notNullValue());
    assertThat(columnFamily1.getRows().get(0).getColumns().size(), is(1));
    assertThat(columnFamily1.getRows().get(0).getColumns().get(0), notNullValue());
    assertThat(columnFamily1.getRows().get(0).getColumns().get(0).getName().getValue(),
        is("columnName1"));
    assertThat(columnFamily1.getRows().get(0).getColumns().get(0).getName().getType(),
        is(GenericTypeEnum.UTF_8_TYPE));
    assertThat(columnFamily1.getRows().get(0).getColumns().get(0).getValue().getValue(),
        is("columnValue1"));
    assertThat(columnFamily1.getRows().get(0).getColumns().get(0).getValue().getType(),
        is(GenericTypeEnum.UTF_8_TYPE));

    assertThat(dataSet.getColumnFamilies().get(1), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(1).getName(), is("columnFamily2"));
    assertThat(dataSet.getColumnFamilies().get(1).getKeyType().getTypeName(),
        is(ComparatorType.UTF8TYPE.getTypeName()));
    assertThat(dataSet.getColumnFamilies().get(1).getComparatorType().getTypeName(),
        is(ComparatorType.UTF8TYPE.getTypeName()));
    assertThat(dataSet.getColumnFamilies().get(1).getSubComparatorType().getTypeName(),
        is(ComparatorType.UTF8TYPE.getTypeName()));

    assertThat(dataSet.getColumnFamilies().get(1).getRows(), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(1).getRows().size(), is(1));
    assertThat(dataSet.getColumnFamilies().get(1).getRows().get(0), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(1).getRows().get(0).getKey().getValue(), is("key02"));
    assertThat(dataSet.getColumnFamilies().get(1).getRows().get(0).getKey().getType(),
        is(GenericTypeEnum.UTF_8_TYPE));
    assertThat(dataSet.getColumnFamilies().get(1).getRows().get(0).getColumns(), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(1).getRows().get(0).getColumns().isEmpty(), is(true));
    assertThat(dataSet.getColumnFamilies().get(1).getRows().get(0).getSuperColumns(), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(1).getRows().get(0).getSuperColumns().size(), is(1));
    assertThat(dataSet.getColumnFamilies().get(1).getRows().get(0).getSuperColumns().get(0), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(1).getRows().get(0).getSuperColumns().get(0).getName().getValue(),
        is("superColumnName2"));
    assertThat(dataSet.getColumnFamilies().get(1).getRows().get(0).getSuperColumns().get(0).getName().getType(),
        is(GenericTypeEnum.UTF_8_TYPE));
    assertThat(dataSet.getColumnFamilies().get(1).getRows().get(0).getSuperColumns().get(0).getColumns(),
        notNullValue());
    assertThat(dataSet.getColumnFamilies().get(1).getRows().get(0).getSuperColumns().get(0).getColumns().size(),
        is(1));
    assertThat(dataSet.getColumnFamilies().get(1).getRows().get(0).getSuperColumns().get(0).getColumns().get(0),
        notNullValue());
    assertThat(dataSet.getColumnFamilies().get(1).getRows().get(0).getSuperColumns().get(0).getColumns().get(0)
        .getName().getValue(), is("columnName2"));
    assertThat(dataSet.getColumnFamilies().get(1).getRows().get(0).getSuperColumns().get(0).getColumns().get(0)
        .getName().getType(), is(GenericTypeEnum.UTF_8_TYPE));
    assertThat(dataSet.getColumnFamilies().get(1).getRows().get(0).getSuperColumns().get(0).getColumns().get(0)
        .getValue().getValue(), is("2"));
    assertThat(dataSet.getColumnFamilies().get(1).getRows().get(0).getSuperColumns().get(0).getColumns().get(0)
        .getValue().getType(), is(GenericTypeEnum.LONG_TYPE));

  }
View Full Code Here

  }

  @Test
  public void shouldGetAStandardCounterColumnFamily() {
    DataSet dataSet = new ClassPathYamlDataSet("yaml/dataSetDefinedValues.yaml");
    assertThat(dataSet.getColumnFamilies().get(2).getRows(), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(2).getRows().size(), is(1));
    assertThat(dataSet.getColumnFamilies().get(2).getRows().get(0), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(2).getRows().get(0).getKey().getValue(), is("key10"));
    assertThat(dataSet.getColumnFamilies().get(2).getRows().get(0).getKey().getType(),
        is(GenericTypeEnum.UTF_8_TYPE));
    assertThat(dataSet.getColumnFamilies().get(2).getRows().get(0).getColumns(), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(2).getRows().get(0).getColumns().size(), is(1));
    assertThat(dataSet.getColumnFamilies().get(2).getRows().get(0).getColumns().get(0), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(2).getRows().get(0).getColumns().get(0).getName().getValue(),
        is("columnName11"));
    assertThat(dataSet.getColumnFamilies().get(2).getRows().get(0).getColumns().get(0).getValue().getValue(),
        is("11"));
    assertThat(dataSet.getColumnFamilies().get(2).getRows().get(0).getColumns().get(0).getValue().getType(),
        is(GenericTypeEnum.COUNTER_TYPE));
  }
View Full Code Here

        is(GenericTypeEnum.COUNTER_TYPE));
  }

  @Test
  public void shouldGetASuperCounterColumnFamily() {
    DataSet dataSet = new ClassPathYamlDataSet("yaml/dataSetDefinedValues.yaml");
    assertThat(dataSet.getColumnFamilies().get(3), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(3).getName(), is("superCounterColumnFamily"));
    assertThat(dataSet.getColumnFamilies().get(3).getRows(), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(3).getRows().size(), is(1));
    assertThat(dataSet.getColumnFamilies().get(3).getRows().get(0), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(3).getRows().get(0).getKey().getValue(), is("key10"));
    assertThat(dataSet.getColumnFamilies().get(3).getRows().get(0).getKey().getType(),
        is(GenericTypeEnum.UTF_8_TYPE));
    assertThat(dataSet.getColumnFamilies().get(3).getRows().get(0).getSuperColumns(), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(3).getRows().get(0).getSuperColumns().size(), is(1));
    assertThat(dataSet.getColumnFamilies().get(3).getRows().get(0).getSuperColumns().get(0), notNullValue());
    assertThat(dataSet.getColumnFamilies().get(3).getRows().get(0).getSuperColumns().get(0).getName().getValue(),
        is("superColumnName11"));
    assertThat(dataSet.getColumnFamilies().get(3).getRows().get(0).getSuperColumns().get(0).getColumns(),
        notNullValue());
    assertThat(dataSet.getColumnFamilies().get(3).getRows().get(0).getSuperColumns().get(0).getColumns().size(),
        is(1));
    assertThat(dataSet.getColumnFamilies().get(3).getRows().get(0).getSuperColumns().get(0).getColumns().get(0),
        notNullValue());
    assertThat(dataSet.getColumnFamilies().get(3).getRows().get(0).getSuperColumns().get(0).getColumns().get(0)
        .getName().getValue(), is("columnName111"));
    assertThat(dataSet.getColumnFamilies().get(3).getRows().get(0).getSuperColumns().get(0).getColumns().get(0)
        .getValue().getValue(), is("111"));
    assertThat(dataSet.getColumnFamilies().get(3).getRows().get(0).getSuperColumns().get(0).getColumns().get(0)
        .getValue().getType(), is(GenericTypeEnum.COUNTER_TYPE));
  }
View Full Code Here

        .getValue().getType(), is(GenericTypeEnum.COUNTER_TYPE));
  }

  @Test(expected = ParseException.class)
  public void shouldNotGetCounterColumnFamilyBecauseThereIsFunctionOverridingDefaultValueType() {
    DataSet dataSet = new ClassPathYamlDataSet("yaml/dataSetBadCounterColumnFamilyWithFunction.yaml");
    dataSet.getKeyspace();
  }
View Full Code Here

    dataSet.getKeyspace();
  }

  @Test
  public void shouldGetAColumnFamilyWithSecondaryIndex() {
    DataSet dataSet = new ClassPathYamlDataSet("yaml/dataSetWithSecondaryIndex.yaml");

    assertThat(dataSet.getColumnFamilies().get(0).getColumnsMetadata().get(0).getColumnName().getValue(),
        is("columnWithIndexAndUTF8ValidationClass"));
    assertThat(dataSet.getColumnFamilies().get(0).getColumnsMetadata().get(0).getColumnIndexType(),
        is(ColumnIndexType.KEYS));
    assertThat(dataSet.getColumnFamilies().get(0).getColumnsMetadata().get(0).getValidationClass(),
        is(ComparatorType.UTF8TYPE));

        assertThat(dataSet.getColumnFamilies().get(0).getColumnsMetadata().get(1).getColumnName().getValue(),
                is("columnWithIndexAndIndexNameAndUTF8ValidationClass"));
        assertThat(dataSet.getColumnFamilies().get(0).getColumnsMetadata().get(1).getColumnIndexType(),
                is(ColumnIndexType.KEYS));
        assertThat(dataSet.getColumnFamilies().get(0).getColumnsMetadata().get(1).getValidationClass(),
                is(ComparatorType.UTF8TYPE));
        assertThat(dataSet.getColumnFamilies().get(0).getColumnsMetadata().get(1).getIndexName(),is("indexNameOfTheIndex"));

    assertThat(dataSet.getColumnFamilies().get(0).getColumnsMetadata().get(2).getColumnName().getValue(),
        is("columnWithUTF8ValidationClass"));
    assertThat(dataSet.getColumnFamilies().get(0).getColumnsMetadata().get(2).getColumnIndexType(), nullValue());
    assertThat(dataSet.getColumnFamilies().get(0).getColumnsMetadata().get(2).getValidationClass(),
        is(ComparatorType.UTF8TYPE));
  }
View Full Code Here

        is(ComparatorType.UTF8TYPE));
  }

  @Test
  public void shouldGetAColumnFamilyWithCompositeType() throws Exception {
    DataSet dataSet = new ClassPathYamlDataSet("yaml/dataSetWithCompositeType.yaml");
    assertThatKeyspaceModelWithCompositeTypeIsOk(dataSet);
  }
View Full Code Here

    assertThatKeyspaceModelWithCompositeTypeIsOk(dataSet);
  }

  @Test(expected = ParseException.class)
  public void shouldNotGetAColumnFamilyWithCompositeType() throws Exception {
    DataSet dataSet = new ClassPathYamlDataSet("yaml/dataSetWithBadCompositeType.yaml");
    dataSet.getKeyspace();
  }
View Full Code Here

    dataSet.getKeyspace();
  }

    @Test
    public void shouldGetAColumnFamilyWithNullColumnValue() {
        DataSet dataSet = new ClassPathYamlDataSet("yaml/dataSetWithNullColumnValue.yaml");
        ColumnFamilyModel columnFamilyModel = dataSet.getColumnFamilies().get(0);
        assertThat(columnFamilyModel.getName(), is("columnFamilyWithNullColumnValue"));
        ColumnModel columnModel = columnFamilyModel.getRows().get(0).getColumns().get(0);
        assertThat(columnModel.getName().getValue(), is("columnWithNullColumnValue"));
        assertThat(columnModel.getValue(), nullValue());
    }
View Full Code Here

TOP

Related Classes of org.cassandraunit.dataset.DataSet

Copyright © 2018 www.massapicom. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.