integration/spark/src/test/scala/org/apache/carbondata/spark/testsuite/index/TestIndexCommand.scala

/*
 * 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.
 */

package org.apache.carbondata.spark.testsuite.index

import org.apache.spark.sql.Row
import org.apache.spark.sql.test.util.QueryTest
import org.scalatest.BeforeAndAfterAll

import org.apache.carbondata.common.exceptions.MetadataProcessException
import org.apache.carbondata.common.exceptions.sql.MalformedIndexCommandException
import org.apache.carbondata.core.constants.CarbonCommonConstants
import org.apache.carbondata.core.util.CarbonProperties

class TestIndexCommand extends QueryTest with BeforeAndAfterAll {

  val testData = s"$resourcesPath/sample.csv"

  override def beforeAll {
    sql("drop table if exists indextest")
    sql("drop table if exists indexshowtest")
    sql("drop table if exists uniqdata")
    sql("create table indextest (a string, b string, c string) STORED AS carbondata")
  }

  val newClass = "org.apache.spark.sql.CarbonSource"

  test("test index create: don't support using non-exist class") {
    intercept[MetadataProcessException] {
      sql(s"CREATE INDEX index1 ON indextest (a) AS '$newClass'")
    }
  }

  test("test index create with properties: don't support using non-exist class") {
    intercept[MetadataProcessException] {
      sql(s"CREATE INDEX index2 ON indextest (a) AS '$newClass' PROPERTIES('key'='value')")
    }
  }

  test("test index create with existing name: don't support using non-exist class") {
    intercept[MetadataProcessException] {
      sql(
        s"CREATE INDEX index2 ON indextest (a) AS '$newClass' PROPERTIES('key'='value')")
    }
  }

  test("test show indexes with no index") {
    sql("drop table if exists indexshowtest")
    sql("create table indexshowtest (a string, b string, c string) STORED AS carbondata")
    assert(sql("show indexes on indexshowtest").collect().length == 0)
  }

  test("test show indexes: show index property related information") {
    val tableName = "indexshowtest"
    val indexName = "bloomindex"
    val indexName2 = "bloomindex2"
    val indexName3 = "bloomindex3"
    sql(s"drop table if exists $tableName")
    // for index
    sql(s"create table $tableName (a string, b string, c string) STORED AS carbondata")
    sql(
      s"""
         | create index $indexName
         | on $tableName (a)
         | as 'bloomfilter'
         | PROPERTIES ('bloom_size'='32000', 'bloom_fpp'='0.001')
       """.stripMargin)
    sql(
      s"""
         | CREATE INDEX $indexName2
         | on table $tableName (b)
         | as 'bloomfilter'
       """.stripMargin)
    sql(
      s"""
         | CREATE INDEX $indexName3
         | on table $tableName (c)
         | as 'bloomfilter'
       """.stripMargin)
    var result = sql(s"show indexes on $tableName").cache()
    checkAnswer(sql(s"show indexes on $tableName"),
      Seq(Row(indexName, "bloomfilter", "a", "'INDEX_COLUMNS'='a','bloom_fpp'='0.001','bloom_size'='32000'", "ENABLED", "NA"),
        Row(indexName2, "bloomfilter", "b", "'INDEX_COLUMNS'='b'", "ENABLED", "NA"),
        Row(indexName3, "bloomfilter", "c", "'INDEX_COLUMNS'='c'", "ENABLED", "NA")))
    result.unpersist()
    sql(s"drop table if exists $tableName")

  }

  test("test don't support lucene on binary data type") {
    val tableName = "indexshowtest20"
    sql(s"drop table if exists $tableName")

    sql(s"CREATE TABLE $tableName(id int, name string, city string, age string, image binary)" +
        s" STORED AS carbondata")

    sql(s"insert into $tableName  values(1,'a3','b3','c1','image2')")
    sql(s"insert into $tableName  values(2,'a3','b2','c2','image2')")
    sql(s"insert into $tableName  values(3,'a1','b2','c1','image3')")
    sql(
      s"""
         | CREATE INDEX agg10
         | ON $tableName (name)
         | AS 'lucene'
         | """.stripMargin)

    checkAnswer(sql(s"show indexes on $tableName"),
      Seq(Row("agg10", "lucene", "name", "'INDEX_COLUMNS'='name'", "ENABLED", "NA")))

    val e = intercept[MalformedIndexCommandException] {
      sql(
        s"""
           | CREATE INDEX agg1
           | ON $tableName (image)
           | AS 'lucene'
           | """.stripMargin)
    }
    assert(e.getMessage.contains("Only String column is supported, column 'image' is BINARY type."))
    checkAnswer(sql(s"show indexes on table $tableName"),
      Seq(Row("agg10", "lucene", "name", "'INDEX_COLUMNS'='name'", "ENABLED", "NA")))

    val pre = sql(
      s"""
         | select name,image, id
         | from $tableName
         | where name = 'a3'
             """.stripMargin)

    assert(2 == pre.collect().length)
    pre.collect().foreach { each =>
      assert(3 == each.length)
      assert("a3".equals(each.get(0)))
      assert("image2".equals(new String(each.getAs[Array[Byte]](1))))
      assert(2 == each.get(2) || 1 == each.get(2))
    }

    sql(s"drop table if exists $tableName")
  }

  override def afterAll {
    sql("DROP TABLE IF EXISTS maintable")
    sql("drop table if exists uniqdata")
    CarbonProperties.getInstance().addProperty(CarbonCommonConstants.ENABLE_HIVE_SCHEMA_META_STORE,
      CarbonCommonConstants.ENABLE_HIVE_SCHEMA_META_STORE_DEFAULT)
    sql("drop table if exists indextest")
    sql("drop table if exists indexshowtest")
  }
}