/*
 * Copyright DataStax, Inc.
 *
 * 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.datastax.driver.core.schemabuilder;

import org.testng.annotations.Test;

import static com.datastax.driver.core.schemabuilder.SchemaBuilder.*;
import static org.assertj.core.api.Assertions.assertThat;

public class DropTest {

    @Test(groups = "unit")
    public void should_drop_table() throws Exception {
        //When
        SchemaStatement statement = dropTable("test");

        //Then
        assertThat(statement.getQueryString()).isEqualTo("DROP TABLE test");
    }

    @Test(groups = "unit")
    public void should_drop_table_with_keyspace() throws Exception {
        //When
        SchemaStatement statement = dropTable("ks", "test");

        //Then
        assertThat(statement.getQueryString()).isEqualTo("DROP TABLE ks.test");
    }

    @Test(groups = "unit")
    public void should_drop_table_with_keyspace_if_exists() throws Exception {
        //When
        SchemaStatement statement = dropTable("ks", "test").ifExists();

        //Then
        assertThat(statement.getQueryString()).isEqualTo("DROP TABLE IF EXISTS ks.test");
    }

    @Test(groups = "unit")
    public void should_drop_type() throws Exception {
        //When
        SchemaStatement statement = dropType("test");

        //Then
        assertThat(statement.getQueryString()).isEqualTo("DROP TYPE test");
    }

    @Test(groups = "unit")
    public void should_drop_type_with_keyspace() throws Exception {
        //When
        SchemaStatement statement = dropType("ks", "test");

        //Then
        assertThat(statement.getQueryString()).isEqualTo("DROP TYPE ks.test");
    }

    @Test(groups = "unit")
    public void should_drop_type_with_keyspace_if_exists() throws Exception {
        //When
        SchemaStatement statement = dropType("ks", "test").ifExists();

        //Then
        assertThat(statement.getQueryString()).isEqualTo("DROP TYPE IF EXISTS ks.test");
    }

    @Test(groups = "unit")
    public void should_drop_index() throws Exception {
        //When
        SchemaStatement statement = dropIndex("test");

        //Then
        assertThat(statement.getQueryString()).isEqualTo("DROP INDEX test");
    }

    @Test(groups = "unit")
    public void should_drop_index_with_keyspace() throws Exception {
        //When
        SchemaStatement statement = dropIndex("ks", "test");

        //Then
        assertThat(statement.getQueryString()).isEqualTo("DROP INDEX ks.test");
    }

    @Test(groups = "unit")
    public void should_drop_index_with_keyspace_if_exists() throws Exception {
        //When
        SchemaStatement statement = dropIndex("ks", "test").ifExists();

        //Then
        assertThat(statement.getQueryString()).isEqualTo("DROP INDEX IF EXISTS ks.test");
    }

    @Test(groups = "unit", expectedExceptions = IllegalArgumentException.class,
            expectedExceptionsMessageRegExp = "The keyspace name 'add' is not allowed because it is a reserved keyword")
    public void should_fail_if_keyspace_name_is_a_reserved_keyword() throws Exception {
        dropTable("add", "test").getQueryString();
    }

    @Test(groups = "unit", expectedExceptions = IllegalArgumentException.class,
            expectedExceptionsMessageRegExp = "The table name 'add' is not allowed because it is a reserved keyword")
    public void should_fail_if_table_name_is_a_reserved_keyword() throws Exception {
        dropTable("add").getQueryString();
    }
}
