it/kafka/build.gradle

/*
 * 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.
 */
import org.apache.beam.gradle.IoPerformanceTestUtilities

plugins { id 'org.apache.beam.module' }
applyJavaNature(
        automaticModuleName: 'org.apache.beam.it.kafka',
)

description = "Apache Beam :: IT :: Kafka"
ext.summary = "Integration test utilities for Kafka."

dependencies {
    implementation project(path: ":it:testcontainers", configuration: "shadow")
    implementation library.java.testcontainers_kafka
    implementation library.java.kafka_clients

    // TODO: excluding Guava until Truth updates it to >32.1.x
    testImplementation(library.java.truth) {
        exclude group: 'com.google.guava', module: 'guava'
    }
    testImplementation library.java.mockito_core
    testRuntimeOnly library.java.slf4j_simple
    // TODO: Remove the below dependencies after KafkaIOLT has been moved to the right directory.
    testImplementation project(path: ":it:google-cloud-platform")
    testImplementation project(path: ":sdks:java:io:kafka")
    testImplementation project(path: ":sdks:java:io:synthetic")
    testImplementation project(path: ":runners:direct-java")
    testImplementation project(path: ":runners:google-cloud-dataflow-java")
    testImplementation project(path: ":sdks:java:testing:test-utils")
    testImplementation project(path: ":sdks:java:extensions:google-cloud-platform-core", configuration: "testRuntimeMigration")
}

tasks.register(
        "KafkaStressTestMedium", IoPerformanceTestUtilities.IoPerformanceTest, project, 'kafka', 'KafkaIOST',
        ['configuration':'medium','project':'apache-beam-testing', 'artifactBucket':'io-performance-temp']
        + System.properties
)
tasks.register(
        "KafkaStressTestLarge", IoPerformanceTestUtilities.IoPerformanceTest, project, 'kafka', 'KafkaIOST',
        ['configuration':'large','project':'apache-beam-testing', 'artifactBucket':'io-performance-temp']
        + System.properties
)