/*
 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License").
 * You may not use this file except in compliance with the License.
 * A copy of the License is located at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * or in the "license" file accompanying this file. This file 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 software.aws.performance.implementations.tests;

import lombok.SneakyThrows;
import org.junit.jupiter.api.Disabled;
import software.aws.performance.DataTypePerformance;
import software.aws.performance.PerformanceTestExecutor;
import software.aws.performance.implementations.executors.SqlGremlinJDBCExecutor;

import static software.aws.performance.implementations.PerformanceTestConstants.LIMIT_COUNT;

@Disabled
public class SqlGremlinJDBCTest extends DataTypePerformance {

    @Override
    @SneakyThrows
    protected PerformanceTestExecutor getPerformanceTestExecutor() {
        return new SqlGremlinJDBCExecutor();
    }

    @Override
    protected String getAllDataQuery() {
        return "SELECT * FROM `airport`";
    }

    @Override
    protected String getNumberOfResultsQuery() {
        return String.format("%s LIMIT %d", getAllDataQuery(), LIMIT_COUNT);
    }

    @Override
    // Need to grab specific properties of certain nodes.
    protected String getTransformNumberOfIntegersQuery() {
        return "SELECT `airport`.`elev` AS `elev` FROM `airport`";
    }

    @Override
    // Need to grab specific properties of certain nodes.
    protected String getTransformNumberOfStringsQuery() {
        return "SELECT `airport`.`code` AS `code` FROM `airport`";
    }

    @Override
    protected String getBaseTestName() {
        return "SqlGremlinJDBC";
    }
}