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

import com.datastax.driver.core.exceptions.UnsupportedFeatureException;
import org.testng.SkipException;
import org.testng.annotations.Test;

import static com.datastax.driver.core.ProtocolVersion.V1;
import static org.assertj.core.api.Assertions.assertThat;

/**
 * Tests targeting protocol v1 specifically.
 */
public class ProtocolV1Test extends CCMTestsSupport {

    @Override
    public Cluster.Builder createClusterBuilder() {
        return super.createClusterBuilder()
                .withProtocolVersion(V1);
    }

    @Override
    public void beforeTestClass(Object testInstance) throws Exception {
        if (CCMBridge.getGlobalCassandraVersion().compareTo(VersionNumber.parse("3.0")) >= 0) {
            throw new SkipException("C* 3.0+ does not support Protocol V1");
        }
        super.beforeTestClass(testInstance);
    }

    /**
     * Validates that a simple query with no variables is correctly executed.
     * @jira_ticket JAVA-1132
     */
    @Test(groups = "short")
    public void should_execute_query_with_no_variables() throws Exception {
        session().execute("select * from system.local");
    }

    /**
     * Validates that a simple query with variables is not allowed with protocol V1.
     * (Values in protocol V1 are only allowed in prepared statements).
     * @jira_ticket JAVA-1132
     */
    @Test(groups = "short")
    public void should_not_execute_query_with_variables() throws Exception {
        try {
            session().execute(new SimpleStatement("select * from system.local where key=?", "local"));
        } catch (UnsupportedFeatureException e) {
            assertThat(e).hasMessageContaining("Unsupported feature with the native protocol V1 (which is currently in use): Binary values are not supported");
        }
    }

    /**
     * Validates that a prepared statement with no variables is correctly prepared and executed.
     * @jira_ticket JAVA-1132
     */
    @Test(groups = "short")
    public void should_execute_prepared_statement_with_no_variables() throws Exception {
        PreparedStatement ps = session().prepare("select * from system.local");
        session().execute(ps.bind());
    }

    /**
     * Validates that a prepared statement with variables is correctly prepared and executed.
     * @jira_ticket JAVA-1132
     */
    @Test(groups = "short")
    public void should_execute_prepared_statement_with_variables() throws Exception {
        PreparedStatement ps = session().prepare("select * from system.local where key=?");
        session().execute(ps.bind("local"));
    }

}
