/*
 * 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.facebook.presto.kafka;

import com.facebook.presto.kafka.util.EmbeddedKafka;
import com.facebook.presto.tests.AbstractTestDistributedQueries;
import io.airlift.tpch.TpchTable;
import org.testng.annotations.AfterClass;
import org.testng.annotations.Test;

import java.io.IOException;

import static com.facebook.presto.kafka.KafkaQueryRunner.createKafkaQueryRunner;
import static com.facebook.presto.kafka.util.EmbeddedKafka.createEmbeddedKafka;
import static io.airlift.testing.Closeables.closeAllRuntimeException;

@Test
public class TestKafkaDistributed
        extends AbstractTestDistributedQueries
{
    private final EmbeddedKafka embeddedKafka;

    public TestKafkaDistributed()
            throws Exception
    {
        this(createEmbeddedKafka());
    }

    public TestKafkaDistributed(EmbeddedKafka embeddedKafka)
            throws Exception
    {
        super(createKafkaQueryRunner(embeddedKafka, TpchTable.getTables()));
        this.embeddedKafka = embeddedKafka;
    }

    @AfterClass(alwaysRun = true)
    public void destroy()
            throws IOException
    {
        closeAllRuntimeException(queryRunner, embeddedKafka);
    }

    //
    // Kafka connector does not support table creation.
    //

    @Override
    public void testCreateTable()
            throws Exception
    {
    }

    @Override
    public void testCreateTableAsSelect()
            throws Exception
    {
    }

    @Override
    public void testSymbolAliasing()
            throws Exception
    {
    }

    //
    // Kafka connector does not support views.
    //

    @Override
    public void testView()
            throws Exception
    {
    }

    @Override
    public void testCompatibleTypeChangeForView()
            throws Exception
    {
        // Kafka connector currently does not support views
    }

    @Override
    public void testCompatibleTypeChangeForView2()
            throws Exception
    {
        // Kafka connector currently does not support views
    }

    @Override
    public void testViewMetadata()
            throws Exception
    {
    }

    @Test
    public void testViewCaseSensitivity()
            throws Exception
    {
        // Kafka connector currently does not support views
    }

    //
    // Kafka connector does not insert.
    //

    @Override
    public void testInsert()
            throws Exception
    {
    }

    //
    // Kafka connector does not delete.
    //

    @Override
    public void testDelete()
            throws Exception
    {
    }

    //
    // Kafka connector does not table rename.
    //

    @Override
    public void testRenameTable()
            throws Exception
    {
    }

    //
    // Kafka connector does not rename column.
    //

    @Override
    public void testRenameColumn()
            throws Exception
    {
    }

    //
    // Kafka connector does not add column.
    //

    @Override
    public void testAddColumn()
            throws Exception
    {
    }
}

@@ -685,35 +685,24 @@
 AbstractTest
-Distributed
 Queries;%0Aimp
@@ -698,16 +698,16 @@
 ueries;%0A
+
 import i
@@ -1140,27 +1140,16 @@
 ractTest
-Distributed
 Queries%0A
@@ -1624,32 +1624,32 @@
 Exception%0A    %7B%0A
+
         closeAll
@@ -1703,1813 +1703,7 @@
    %7D
-%0A%0A    //%0A    // Kafka connector does not support table creation.%0A    //%0A%0A    @Override%0A    public void testCreateTable()%0A            throws Exception%0A    %7B%0A    %7D%0A%0A    @Override%0A    public void testCreateTableAsSelect()%0A            throws Exception%0A    %7B%0A    %7D%0A%0A    @Override%0A    public void testSymbolAliasing()%0A            throws Exception%0A    %7B%0A    %7D%0A%0A    //%0A    // Kafka connector does not support views.%0A    //%0A%0A    @Override%0A    public void testView()%0A            throws Exception%0A    %7B%0A    %7D%0A%0A    @Override%0A    public void testCompatibleTypeChangeForView()%0A            throws Exception%0A    %7B%0A        // Kafka connector currently does not support views%0A    %7D%0A%0A    @Override%0A    public void testCompatibleTypeChangeForView2()%0A            throws Exception%0A    %7B%0A        // Kafka connector currently does not support views%0A    %7D%0A%0A    @Override%0A    public void testViewMetadata()%0A            throws Exception%0A    %7B%0A    %7D%0A%0A    @Test%0A    public void testViewCaseSensitivity()%0A            throws Exception%0A    %7B%0A        // Kafka connector currently does not support views%0A    %7D%0A%0A    //%0A    // Kafka connector does not insert.%0A    //%0A%0A    @Override%0A    public void testInsert()%0A            throws Exception%0A    %7B%0A    %7D%0A%0A    //%0A    // Kafka connector does not delete.%0A    //%0A%0A    @Override%0A    public void testDelete()%0A            throws Exception%0A    %7B%0A    %7D%0A%0A    //%0A    // Kafka connector does not table rename.%0A    //%0A%0A    @Override%0A    public void testRenameTable()%0A            throws Exception%0A    %7B%0A    %7D%0A%0A    //%0A    // Kafka connector does not rename column.%0A    //%0A%0A    @Override%0A    public void testRenameColumn()%0A            throws Exception%0A    %7B%0A    %7D%0A%0A    //%0A    // Kafka connector does not add column.%0A    //%0A%0A    @Override%0A    public void testAddColumn()%0A            throws Exception%0A    %7B%0A    %7D
 %0A%7D%0A

Fix Kafka distributed tests