UtilsTest.java

/*
 * Copyright 2014, Stratio.
 *
 *  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.deep.commons.utils;

import static com.stratio.deep.commons.utils.Utils.castNumberType;
import static com.stratio.deep.commons.utils.Utils.getExtractorInstance;
import static com.stratio.deep.commons.utils.Utils.removeAddressPort;
import static com.stratio.deep.commons.utils.Utils.splitListByComma;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;

import java.math.BigDecimal;
import java.math.BigInteger;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicLong;

import org.apache.spark.Partition;
import org.junit.Test;

import com.stratio.deep.commons.config.BaseConfig;
import com.stratio.deep.commons.entity.Cells;
import com.stratio.deep.commons.querybuilder.UpdateQueryBuilder;
import com.stratio.deep.commons.rdd.IExtractor;

/**
 * The type Utils test.
 * <p/>
 * Created by rcrespo on 12/11/14.
 */
public class UtilsTest {

    /**
     * Test new type instance.
     *
     * @throws Exception the exception
     */
    @Test
    public void testNewTypeInstance() throws Exception {

    }

    /**
     * Test new type instance 1.
     *
     * @throws Exception the exception
     */
    @Test
    public void testNewTypeInstance1() throws Exception {

    }

    /**
     * Test quote.
     *
     * @throws Exception the exception
     */
    @Test
    public void testQuote() throws Exception {

    }

    /**
     * Test single quote.
     *
     * @throws Exception the exception
     */
    @Test
    public void testSingleQuote() throws Exception {

    }

    /**
     * Test batch query generator.
     *
     * @throws Exception the exception
     */
    @Test
    public void testBatchQueryGenerator() throws Exception {

    }

    /**
     * Test prepare tuple 4 cql driver.
     *
     * @throws Exception the exception
     */
    @Test
    public void testPrepareTuple4CqlDriver() throws Exception {

    }

    /**
     * Test find setter.
     *
     * @throws Exception the exception
     */
    @Test
    public void testFindSetter() throws Exception {

    }

    /**
     * Test find getter.
     *
     * @throws Exception the exception
     */
    @Test
    public void testFindGetter() throws Exception {

    }

    /**
     * Test inet address from location.
     *
     * @throws Exception the exception
     */
    @Test
    public void testInetAddressFromLocation() throws Exception {

    }

    /**
     * Test get all fields.
     *
     * @throws Exception the exception
     */
    @Test
    public void testGetAllFields() throws Exception {

    }

    /**
     * Test remove address port.
     *
     * @throws Exception the exception
     */
    @Test
    public void testRemoveAddressPort() throws Exception {
        List<String> expected = new ArrayList();
        expected.add("stratio1");
        expected.add("stratio2");
        expected.add("stratio3");
        expected.add("stratio4");

        List<String> test = new ArrayList();
        test.add("stratio1:2010");
        test.add("stratio2:2030");
        test.add("stratio3:2040");
        test.add("stratio4");

        List<String> stringList = removeAddressPort(test);

        assertEquals(stringList, expected);

    }

    /**
     * Test split list by comma.
     *
     * @throws Exception the exception
     */
    @Test
    public void testSplitListByComma() throws Exception {

        List<String> hostList = new ArrayList<>();
        hostList.add("host1");
        hostList.add("host2");
        hostList.add("host3");

        String hosts = splitListByComma(hostList);

        assertEquals("host1,host2,host3", hosts);
    }

    /**
     * Test get extractor instance.
     *
     * @throws Exception the exception
     */
    @Test
    public void testGetExtractorInstance() throws Exception {





        BaseConfig<Cells, BaseConfig> baseConfig = new BaseConfig<>();

        baseConfig.setEntityClass(Cells.class);
        baseConfig.setExtractorImplClass(testExtractor.class);

        IExtractor extractorInstance1 = getExtractorInstance(baseConfig);


        assertNotNull(extractorInstance1);

    }

    /**
     * Test cast number type.
     *
     * @throws Exception the exception
     */
    @Test
    public void testCastNumberType() throws Exception {
        Object aDouble = 10D;



//        AtomicInteger, AtomicLong, BigDecimal, BigInteger, Byte, Double, Float, Integer, Long, Short

        Object o ;

        o = castNumberType(aDouble, AtomicInteger.class);

        assertEquals(AtomicInteger.class, o.getClass());


        o = castNumberType(aDouble, AtomicLong.class);

        assertEquals(AtomicLong.class, o.getClass());


        o = castNumberType(aDouble, BigDecimal.class);

        assertEquals(BigDecimal.class, o.getClass());


        o = castNumberType(aDouble, BigInteger.class);

        assertEquals(BigInteger.class, o.getClass());


        o = castNumberType(aDouble, Byte.class);

        assertEquals(Byte.class, o.getClass());

        o = castNumberType(aDouble, Double.class);
        assertEquals(Double.class, o.getClass());


        o = castNumberType(aDouble, Float.class);

        assertEquals(Float.class, o.getClass());


        o = castNumberType(aDouble, Integer.class);

        assertEquals(Integer.class, o.getClass());

        o = castNumberType(aDouble, Long.class);

        assertEquals(Long.class, o.getClass());

        o = castNumberType(aDouble, Short.class);

        assertEquals(Short.class, o.getClass());

    }



    static class testExtractor<T> implements IExtractor<T, BaseConfig> {

        public testExtractor(){

        }

        public testExtractor(Class<T> tClass){
            super();
        }

        @Override
        public Partition[] getPartitions(BaseConfig config) {
            return new Partition[0];
        }

        @Override
        public boolean hasNext() {
            return false;
        }

        @Override
        public T next() {
            return null;
        }

        @Override
        public void close() {

        }

        @Override
        public void initIterator(Partition dp, BaseConfig config) {

        }

        @Override
        public void saveRDD(Object o) {

        }

        @Override
        public List<String> getPreferredLocations(Partition split) {
            return null;
        }

        @Override
        public void initSave(BaseConfig config, Object first, UpdateQueryBuilder queryBuilder) {

        }
    }

}