com.stratio.decision.service.SaveToCassandraOperationsServiceTest.java Source code

Introduction

Here is the source code for com.stratio.decision.service.SaveToCassandraOperationsServiceTest.java

Source

/**
 * Copyright (C) 2014 Stratio (http://stratio.com)
 *
 * 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.stratio.decision.service;

import com.datastax.driver.core.Session;
import com.datastax.driver.core.querybuilder.Insert;
import com.stratio.decision.commons.constants.ColumnType;
import com.stratio.decision.commons.constants.STREAMING;
import com.stratio.decision.commons.messages.ColumnNameTypeValue;
import org.apache.commons.collections.set.ListOrderedSet;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.ArrayList;
import java.util.List;
import java.util.Set;

import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.mock;

/**
 * Created by aitor on 9/23/15.
 * TODO:
 * Current SaveToCassandraOperationsService uses datastax Session object but doesn't return anything
 * in the method implementations. To build a better integration and create better test it would be
 * nice refactor that code to process the returned data and validate the information.
 */
public class SaveToCassandraOperationsServiceTest {

    private static final Logger LOGGER = LoggerFactory.getLogger(SaveToCassandraOperationsServiceTest.class);

    private SaveToCassandraOperationsService service;

    private final String KEYSPACE = "mykeyspace";

    private final String TABLE = "mytable";

    private final List<ColumnNameTypeValue> columns = new ArrayList<ColumnNameTypeValue>();
    {
        columns.add(new ColumnNameTypeValue("id", ColumnType.INTEGER, 1));
        columns.add(new ColumnNameTypeValue("name", ColumnType.STRING, "my name"));
        columns.add(new ColumnNameTypeValue("enabled", ColumnType.BOOLEAN, "true"));
        columns.add(new ColumnNameTypeValue("mydouble", ColumnType.DOUBLE, 1.0));
        columns.add(new ColumnNameTypeValue("myfloat", ColumnType.FLOAT, 1f));
        columns.add(new ColumnNameTypeValue("mylong", ColumnType.LONG, Long.parseLong("1")));
        columns.add(new ColumnNameTypeValue("timestamp", ColumnType.STRING, "1234567890"));
    }

    @Mock
    private Session mockedSession;

    @Before
    public void setUp() throws Exception {
        LOGGER.debug("Initializing required classes");
        mockedSession = mock(Session.class);
        service = new SaveToCassandraOperationsService(mockedSession);

    }

    @After
    public void tearDown() throws Exception {

    }

    @Test
    public void testCreateKeyspace() throws Exception {
        service.createKeyspace(KEYSPACE);
        Exception ex = null;
        try {
            service.createKeyspace(KEYSPACE);
        } catch (Exception e) {
            ex = e;
        }
        assertEquals("Expected null but exception found", null, ex);
    }

    @Test
    public void testCreateTable() throws Exception {
        Exception ex = null;
        try {
            service.createTable(TABLE, columns, "timestamp");
        } catch (Exception e) {
            ex = e;
        }
        assertEquals("Expected null but exception found", null, ex);
    }

    @Test
    public void testCreateInsertStatement() throws Exception {
        Insert insert = service.createInsertStatement(TABLE, columns, "timestamp");
        assertEquals("Expected keyspace not found", "\"" + STREAMING.STREAMING_KEYSPACE_NAME + "\"",
                insert.getKeyspace());

    }

    @Test
    public void testAlterTable() throws Exception {
        Exception ex = null;
        try {

            columns.add(new ColumnNameTypeValue("newfield", ColumnType.STRING, "my new value"));
            Set<String> oldColumnNamesnew = new ListOrderedSet();
            oldColumnNamesnew.add("id");
            oldColumnNamesnew.add("name");
            oldColumnNamesnew.add("enabled");
            oldColumnNamesnew.add("timestamp");

            service.alterTable(TABLE, oldColumnNamesnew, columns);

        } catch (Exception e) {
            ex = e;
        }
        assertEquals("Expected null but exception found", null, ex);

    }
}