com.impetus.kundera.client.cassandra.composite.DSCQLTranslatorTest.java Source code

Introduction

Here is the source code for com.impetus.kundera.client.cassandra.composite.DSCQLTranslatorTest.java

Source

/*******************************************************************************
 * * Copyright 2013 Impetus Infotech.
 *  *
 *  * Licensed 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.impetus.kundera.client.cassandra.composite;

import java.util.Date;
import java.util.Map;
import java.util.UUID;

import javax.persistence.EntityManagerFactory;
import javax.persistence.Persistence;

import junit.framework.Assert;

import org.apache.commons.lang.StringUtils;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.impetus.client.cassandra.common.CassandraConstants;
import com.impetus.client.cassandra.thrift.CQLTranslator;
import com.impetus.client.cassandra.thrift.CQLTranslator.TranslationType;
import com.impetus.kundera.client.cassandra.persistence.CassandraCli;
import com.impetus.kundera.metadata.KunderaMetadataManager;
import com.impetus.kundera.metadata.model.EntityMetadata;
import com.impetus.kundera.persistence.EntityManagerFactoryImpl;

/**
 * JUnit for CQL translator test
 * 
 * @author vivek.mishra
 * 
 */
public class DSCQLTranslatorTest {
    private EntityManagerFactory emf;

    private static final Logger logger = LoggerFactory.getLogger(DSCompositeTypeTest.class);

    private static final String KEYSPACE = "KunderaExamples";

    private static final String _PU = "ds_composite_pu";

    /**
     * @throws java.lang.Exception
     */
    @Before
    public void setUp() throws Exception {
        CassandraCli.dropKeySpace(KEYSPACE);
        CassandraCli.cassandraSetUp();
        emf = Persistence.createEntityManagerFactory(_PU);
    }

    @Test
    public void testPrepareColumns() {
        logger.info("On prepare columns.");
        CQLTranslator translator = new CQLTranslator();
        UUID timeLineId = UUID.randomUUID();
        Date currentDate = new Date();
        UserTimeLine key = new UserTimeLine("mevivs", 1, timeLineId);
        PrimeUser user = new PrimeUser(key);
        user.setTweetBody("my first tweet");
        user.setTweetDate(currentDate);
        EntityMetadata entityMetadata = KunderaMetadataManager
                .getEntityMetadata(((EntityManagerFactoryImpl) emf).getKunderaMetadataInstance(), PrimeUser.class);
        Map<String, StringBuilder> translatedSql = translator.prepareColumnOrColumnValues(user, entityMetadata,
                TranslationType.VALUE, null, ((EntityManagerFactoryImpl) emf).getKunderaMetadataInstance())
                .get(TranslationType.VALUE);
        String columnAsCsv = "'mevivs',1,"
                + timeLineId /*+ ",'my first tweet','" + currentDate.getTime() + */ /*+ "'"*/;

        Assert.assertTrue(
                StringUtils.contains(translatedSql.get(entityMetadata.getTableName()).toString(), columnAsCsv));
        //        Assert.assertEquals(columnAsCsv, translatedSql);
    }

    @Test
    public void testGetKeyword() {
        CQLTranslator translator = new CQLTranslator();
        Assert.assertEquals("read_repair_chance", translator.getKeyword(CassandraConstants.READ_REPAIR_CHANCE));
        Assert.assertEquals("dclocal_read_repair_chance",
                translator.getKeyword(CassandraConstants.DCLOCAL_READ_REPAIR_CHANCE));
        Assert.assertEquals("bloom_filter_fp_chance",
                translator.getKeyword(CassandraConstants.BLOOM_FILTER_FP_CHANCE));
        Assert.assertEquals("compaction_strategy_class",
                translator.getKeyword(CassandraConstants.COMPACTION_STRATEGY));
        Assert.assertEquals("bloom_filter_fp_chance",
                translator.getKeyword(CassandraConstants.BLOOM_FILTER_FP_CHANCE));
        Assert.assertEquals("replicate_on_write", translator.getKeyword(CassandraConstants.REPLICATE_ON_WRITE));
        Assert.assertEquals("caching", translator.getKeyword(CassandraConstants.CACHING));
        Assert.assertEquals("comment", translator.getKeyword(CassandraConstants.COMMENT));
        Assert.assertEquals("gc_grace_seconds", translator.getKeyword(CassandraConstants.GC_GRACE_SECONDS));

    }

    @After
    public void tearDown() {
        CassandraCli.dropKeySpace(KEYSPACE);
        emf.close();
    }
}