test: add script to run yatt tests (#9599)

* test: add script to run yatt tests * add unit test * move sql testing tool to own module * move stubkafkaservice back * move stubkafkaservicetest back
confluentinc · Sep 30, 2022 · 72e3f8c · 72e3f8c
1 parent 018f3b8
commit 72e3f8c
Show file tree

Hide file tree

Showing 67 changed files with 976 additions and 96 deletions.
diff --git a/bin/run-ksql-test b/bin/run-ksql-test
@@ -0,0 +1,17 @@
+#!/bin/bash
+# (Copyright) [2019 - 2019] Confluent, Inc.
+
+ #
+# Use shellcheck to lint this file
+#
+set -ue
+
+ base_dir=$( cd "$( dirname "${BASH_SOURCE[0]}" )/.." && pwd )
+: "${KSQL_CONFIG_DIR:="$base_dir/config"}"
+
+ : "${KSQL_LOG4J_OPTS:=""}"
+if [ -z "$KSQL_LOG4J_OPTS" ] && [ -e "$KSQL_CONFIG_DIR/log4j-sql-test.properties" ]; then
+  export KSQL_LOG4J_OPTS="-Dlog4j.configuration=file:$KSQL_CONFIG_DIR/log4j-sql-test.properties"
+fi
+
+ exec "$base_dir"/bin/ksql-run-class io.confluent.ksql.tools.test.SqlTestingTool "$@"
diff --git a/config/log4j-sql-test.properties b/config/log4j-sql-test.properties
@@ -0,0 +1,10 @@
+# Root logger -- disable all non-sql-test logging
+log4j.rootLogger=OFF
+
+# SQL tester logger
+log4j.logger.io.confluent.ksql.tools.test=INFO, console
+
+log4j.appender.console=org.apache.log4j.ConsoleAppender
+log4j.appender.console.Target=System.out
+log4j.appender.console.layout=org.apache.log4j.PatternLayout
+log4j.appender.consoleAppender.layout.ConversionPattern=[%t] %-5p %c %x - %m%n
diff --git a/ksqldb-functional-tests/pom.xml b/ksqldb-functional-tests/pom.xml
@@ -139,6 +139,12 @@
       <scope>test</scope>
     </dependency>
 
+    <dependency>
+      <groupId>io.confluent.ksql</groupId>
+      <artifactId>ksqldb-testing-tool</artifactId>
+      <version>${io.confluent.ksql.version}</version>
+    </dependency>
+
     <dependency>
       <groupId>io.confluent.avro</groupId>
       <artifactId>avro-random-generator</artifactId>
@@ -156,11 +162,6 @@
       <artifactId>protobuf-java-util</artifactId>
     </dependency>
 
-    <dependency>
-      <groupId>commons-io</groupId>
-      <artifactId>commons-io</artifactId>
-    </dependency>
-
   </dependencies>
 
   <build>

diff --git a/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/loader/JsonTestLoader.java b/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/loader/JsonTestLoader.java
@@ -20,8 +20,9 @@
 import com.fasterxml.jackson.databind.ObjectMapper;
 import io.confluent.ksql.test.TestFrameworkException;
 import io.confluent.ksql.test.model.TestFileContext;
-import io.confluent.ksql.test.tools.Test;
 import io.confluent.ksql.test.tools.TestJsonMapper;
+import io.confluent.ksql.tools.test.TestLoader;
+import io.confluent.ksql.tools.test.model.Test;
 import java.io.BufferedReader;
 import java.io.IOException;
 import java.io.InputStream;

diff --git a/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/loader/TestFile.java b/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/loader/TestFile.java
@@ -16,7 +16,7 @@
 package io.confluent.ksql.test.loader;
 
 import io.confluent.ksql.test.model.TestFileContext;
-import io.confluent.ksql.test.tools.Test;
+import io.confluent.ksql.tools.test.model.Test;
 import java.util.stream.Stream;
 
 public interface TestFile<TestTypeT extends Test> {

diff --git a/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/model/PathLocation.java b/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/model/PathLocation.java
@@ -18,6 +18,7 @@
 import static java.util.Objects.requireNonNull;
 
 import com.google.common.base.Preconditions;
+import io.confluent.ksql.tools.test.model.TestLocation;
 import java.nio.file.Path;
 
 /**

diff --git a/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/model/SchemaReferencesNode.java b/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/model/SchemaReferencesNode.java
@@ -25,9 +25,9 @@
 import com.fasterxml.jackson.databind.node.TextNode;
 import io.confluent.kafka.schemaregistry.ParsedSchema;
 import io.confluent.ksql.serde.protobuf.ProtobufFormat;
-import io.confluent.ksql.test.tools.SchemaReference;
 import io.confluent.ksql.test.tools.TestJsonMapper;
 import io.confluent.ksql.test.utils.SerdeUtil;
+import io.confluent.ksql.tools.test.model.SchemaReference;
 
 @JsonInclude(JsonInclude.Include.NON_EMPTY)
 public final class SchemaReferencesNode {

diff --git a/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/model/TestFileContext.java b/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/model/TestFileContext.java
@@ -19,6 +19,8 @@
 
 import com.google.common.collect.ImmutableList;
 import io.confluent.ksql.test.TestFrameworkException;
+import io.confluent.ksql.tools.test.model.LocationWithinFile;
+import io.confluent.ksql.tools.test.model.TestLocation;
 import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.util.List;

diff --git a/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/model/TopicNode.java b/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/model/TopicNode.java
@@ -28,11 +28,11 @@
 import io.confluent.kafka.schemaregistry.ParsedSchema;
 import io.confluent.ksql.serde.SerdeFeatures;
 import io.confluent.ksql.serde.protobuf.ProtobufFormat;
-import io.confluent.ksql.test.tools.SchemaReference;
 import io.confluent.ksql.test.tools.TestJsonMapper;
-import io.confluent.ksql.test.tools.Topic;
 import io.confluent.ksql.test.tools.exceptions.InvalidFieldException;
 import io.confluent.ksql.test.utils.SerdeUtil;
+import io.confluent.ksql.tools.test.model.SchemaReference;
+import io.confluent.ksql.tools.test.model.Topic;
 import java.util.List;
 import java.util.Optional;
 import java.util.stream.Collectors;

diff --git a/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/tools/TestCase.java b/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/tools/TestCase.java
@@ -23,8 +23,9 @@
 import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
 import io.confluent.ksql.schema.query.QuerySchemas;
 import io.confluent.ksql.test.model.KsqlVersion;
-import io.confluent.ksql.test.model.TestLocation;
 import io.confluent.ksql.test.tools.conditions.PostConditions;
+import io.confluent.ksql.tools.test.model.TestLocation;
+import io.confluent.ksql.tools.test.model.Topic;
 import io.confluent.ksql.util.KsqlConfig;
 import java.nio.file.Path;
 import java.util.Collection;

diff --git a/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/tools/TestCaseBuilder.java b/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/tools/TestCaseBuilder.java
@@ -22,9 +22,10 @@
 import io.confluent.ksql.test.model.PostConditionsNode;
 import io.confluent.ksql.test.model.RecordNode;
 import io.confluent.ksql.test.model.TestCaseNode;
-import io.confluent.ksql.test.model.TestLocation;
 import io.confluent.ksql.test.model.TopicNode;
 import io.confluent.ksql.test.tools.conditions.PostConditions;
+import io.confluent.ksql.tools.test.model.TestLocation;
+import io.confluent.ksql.tools.test.model.Topic;
 import io.confluent.ksql.util.KsqlConfig;
 import java.nio.file.Path;
 import java.util.Collections;

diff --git a/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/tools/TestCaseBuilderUtil.java b/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/tools/TestCaseBuilderUtil.java
@@ -49,6 +49,7 @@
 import io.confluent.ksql.serde.SerdeFeaturesFactory;
 import io.confluent.ksql.statement.ConfiguredStatement;
 import io.confluent.ksql.statement.SourcePropertyInjector;
+import io.confluent.ksql.tools.test.model.Topic;
 import io.confluent.ksql.util.KsqlConfig;
 import java.nio.file.Path;
 import java.util.Collection;

diff --git a/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/tools/TestExecutor.java b/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/tools/TestExecutor.java
@@ -59,11 +59,13 @@
 import io.confluent.ksql.test.model.TestHeader;
 import io.confluent.ksql.test.model.WindowData;
 import io.confluent.ksql.test.tools.TopicInfoCache.TopicInfo;
-import io.confluent.ksql.test.tools.stubs.StubKafkaClientSupplier;
-import io.confluent.ksql.test.tools.stubs.StubKafkaConsumerGroupClient;
 import io.confluent.ksql.test.tools.stubs.StubKafkaService;
-import io.confluent.ksql.test.tools.stubs.StubKafkaTopicClient;
 import io.confluent.ksql.test.utils.TestUtils;
+import io.confluent.ksql.tools.test.TestFunctionRegistry;
+import io.confluent.ksql.tools.test.model.Topic;
+import io.confluent.ksql.tools.test.stubs.StubKafkaClientSupplier;
+import io.confluent.ksql.tools.test.stubs.StubKafkaConsumerGroupClient;
+import io.confluent.ksql.tools.test.stubs.StubKafkaTopicClient;
 import io.confluent.ksql.util.BytesUtils;
 import io.confluent.ksql.util.KsqlConfig;
 import io.confluent.ksql.util.KsqlConstants;

diff --git a/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/tools/TestExecutorUtil.java b/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/tools/TestExecutorUtil.java
@@ -60,6 +60,7 @@
 import io.confluent.ksql.statement.InjectorChain;
 import io.confluent.ksql.statement.SourcePropertyInjector;
 import io.confluent.ksql.test.tools.stubs.StubKafkaService;
+import io.confluent.ksql.tools.test.model.Topic;
 import io.confluent.ksql.util.KsqlConfig;
 import io.confluent.ksql.util.KsqlConstants;
 import io.confluent.ksql.util.KsqlException;

diff --git a/...ctional-tests/src/main/java/io/confluent/ksql/test/tools/TopologyTestDriverContainer.java b/...ctional-tests/src/main/java/io/confluent/ksql/test/tools/TopologyTestDriverContainer.java
@@ -17,6 +17,7 @@
 
 import static java.util.Objects.requireNonNull;
 
+import io.confluent.ksql.tools.test.model.Topic;
 import io.confluent.ksql.util.Pair;
 import java.util.HashMap;
 import java.util.List;

diff --git a/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/tools/VersionedTest.java b/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/tools/VersionedTest.java
@@ -16,6 +16,7 @@
 package io.confluent.ksql.test.tools;
 
 import io.confluent.ksql.test.model.KsqlVersion;
+import io.confluent.ksql.tools.test.model.Test;
 
 public interface VersionedTest extends Test {
 

diff --git a/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/tools/command/TestOptions.java b/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/tools/command/TestOptions.java
@@ -19,6 +19,7 @@
 import com.github.rvesse.airline.annotations.Option;
 import com.github.rvesse.airline.annotations.restrictions.Once;
 import com.github.rvesse.airline.annotations.restrictions.Required;
+import io.confluent.ksql.tools.test.command.TestOptionsParser;
 import java.io.IOException;
 import java.util.Optional;
 

diff --git a/...b-functional-tests/src/main/java/io/confluent/ksql/test/tools/stubs/StubKafkaService.java b/...b-functional-tests/src/main/java/io/confluent/ksql/test/tools/stubs/StubKafkaService.java
@@ -16,7 +16,7 @@
 package io.confluent.ksql.test.tools.stubs;
 
 import com.google.common.collect.ImmutableList;
-import io.confluent.ksql.test.tools.Topic;
+import io.confluent.ksql.tools.test.model.Topic;
 import io.confluent.ksql.util.KsqlException;
 import java.util.ArrayList;
 import java.util.Collection;

diff --git a/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/utils/SerdeUtil.java b/ksqldb-functional-tests/src/main/java/io/confluent/ksql/test/utils/SerdeUtil.java
@@ -50,8 +50,8 @@
 import io.confluent.ksql.test.serde.protobuf.ValueSpecProtobufNoSRSerdeSupplier;
 import io.confluent.ksql.test.serde.protobuf.ValueSpecProtobufSerdeSupplier;
 import io.confluent.ksql.test.serde.string.StringSerdeSupplier;
-import io.confluent.ksql.test.tools.SchemaReference;
 import io.confluent.ksql.test.tools.exceptions.InvalidFieldException;
+import io.confluent.ksql.tools.test.model.SchemaReference;
 import java.util.List;
 import java.util.Map;
 import java.util.Optional;

diff --git a/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/SchemaTranslationTest.java b/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/SchemaTranslationTest.java
@@ -17,15 +17,15 @@
 import io.confluent.ksql.test.loader.JsonTestLoader;
 import io.confluent.ksql.test.loader.TestFile;
 import io.confluent.ksql.test.model.TestFileContext;
-import io.confluent.ksql.test.model.TestLocation;
 import io.confluent.ksql.test.tools.Record;
 import io.confluent.ksql.test.tools.TestCase;
-import io.confluent.ksql.test.tools.Topic;
 import io.confluent.ksql.test.tools.VersionBounds;
 import io.confluent.ksql.test.tools.conditions.PostConditions;
 import io.confluent.ksql.test.tools.exceptions.InvalidFieldException;
 import io.confluent.ksql.test.tools.exceptions.MissingFieldException;
 import io.confluent.ksql.test.utils.SerdeUtil;
+import io.confluent.ksql.tools.test.model.TestLocation;
+import io.confluent.ksql.tools.test.model.Topic;
 import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.util.ArrayList;

diff --git a/...ctional-tests/src/test/java/io/confluent/ksql/test/SchemaTranslationWithSchemaIdTest.java b/...ctional-tests/src/test/java/io/confluent/ksql/test/SchemaTranslationWithSchemaIdTest.java
@@ -17,15 +17,15 @@
 import io.confluent.ksql.test.loader.JsonTestLoader;
 import io.confluent.ksql.test.loader.TestFile;
 import io.confluent.ksql.test.model.TestFileContext;
-import io.confluent.ksql.test.model.TestLocation;
 import io.confluent.ksql.test.tools.Record;
 import io.confluent.ksql.test.tools.TestCase;
-import io.confluent.ksql.test.tools.Topic;
 import io.confluent.ksql.test.tools.VersionBounds;
 import io.confluent.ksql.test.tools.conditions.PostConditions;
 import io.confluent.ksql.test.tools.exceptions.InvalidFieldException;
 import io.confluent.ksql.test.tools.exceptions.MissingFieldException;
 import io.confluent.ksql.test.utils.SerdeUtil;
+import io.confluent.ksql.tools.test.model.TestLocation;
+import io.confluent.ksql.tools.test.model.Topic;
 import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.util.ArrayList;

diff --git a/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/driver/KsqlTesterTest.java b/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/driver/KsqlTesterTest.java
@@ -15,9 +15,9 @@
 
 package io.confluent.ksql.test.driver;
 
-import io.confluent.ksql.test.parser.SqlTestLoader;
-import io.confluent.ksql.test.parser.SqlTestLoader.SqlTest;
-import io.confluent.ksql.test.tools.SqlTestExecutor;
+import io.confluent.ksql.tools.test.parser.SqlTestLoader;
+import io.confluent.ksql.tools.test.parser.SqlTestLoader.SqlTest;
+import io.confluent.ksql.tools.test.SqlTestExecutor;
 import io.confluent.ksql.test.util.KsqlTestFolder;
 import java.io.IOException;
 import java.nio.file.Path;

diff --git a/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/planned/PlannedTestLoader.java b/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/planned/PlannedTestLoader.java
@@ -18,8 +18,8 @@
 import static java.util.Objects.requireNonNull;
 
 import com.google.common.annotations.VisibleForTesting;
-import io.confluent.ksql.test.loader.TestLoader;
 import io.confluent.ksql.test.tools.TestCase;
+import io.confluent.ksql.tools.test.TestLoader;
 import java.nio.file.Path;
 import java.util.List;
 import java.util.function.Predicate;

diff --git a/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/planned/TestCasePlan.java b/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/planned/TestCasePlan.java
@@ -15,7 +15,7 @@
 
 package io.confluent.ksql.test.planned;
 
-import io.confluent.ksql.test.model.TestLocation;
+import io.confluent.ksql.tools.test.model.TestLocation;
 import java.util.Objects;
 
 public final class TestCasePlan {

diff --git a/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/planned/TestCasePlanLoader.java b/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/planned/TestCasePlanLoader.java
@@ -37,8 +37,8 @@
 import io.confluent.ksql.test.tools.TestCaseBuilderUtil;
 import io.confluent.ksql.test.tools.TestExecutionListener;
 import io.confluent.ksql.test.tools.TestExecutor;
-import io.confluent.ksql.test.tools.TestFunctionRegistry;
-import io.confluent.ksql.test.tools.Topic;
+import io.confluent.ksql.tools.test.TestFunctionRegistry;
+import io.confluent.ksql.tools.test.model.Topic;
 import io.confluent.ksql.util.KsqlConfig;
 import io.confluent.ksql.util.PersistentQueryMetadata;
 import java.io.File;

diff --git a/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/rest/RestTestCase.java b/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/rest/RestTestCase.java
@@ -21,11 +21,11 @@
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.ImmutableMap;
 import io.confluent.ksql.rest.client.RestResponse;
-import io.confluent.ksql.test.model.TestLocation;
 import io.confluent.ksql.test.rest.model.Response;
 import io.confluent.ksql.test.tools.Record;
-import io.confluent.ksql.test.tools.Test;
-import io.confluent.ksql.test.tools.Topic;
+import io.confluent.ksql.tools.test.model.Test;
+import io.confluent.ksql.tools.test.model.TestLocation;
+import io.confluent.ksql.tools.test.model.Topic;
 import java.util.Collection;
 import java.util.List;
 import java.util.Map;

diff --git a/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/rest/RestTestCaseBuilder.java b/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/rest/RestTestCaseBuilder.java
@@ -19,11 +19,11 @@
 import io.confluent.ksql.rest.client.RestResponse;
 import io.confluent.ksql.test.model.RecordNode;
 import io.confluent.ksql.test.model.TestFileContext;
-import io.confluent.ksql.test.model.TestLocation;
 import io.confluent.ksql.test.model.TopicNode;
 import io.confluent.ksql.test.tools.Record;
 import io.confluent.ksql.test.tools.TestCaseBuilderUtil;
-import io.confluent.ksql.test.tools.Topic;
+import io.confluent.ksql.tools.test.model.TestLocation;
+import io.confluent.ksql.tools.test.model.Topic;
 import java.util.List;
 import java.util.Optional;
 import java.util.stream.Collectors;

diff --git a/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/rest/RestTestExecutor.java b/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/rest/RestTestExecutor.java
@@ -29,7 +29,6 @@
 import com.fasterxml.jackson.databind.JsonNode;
 import com.fasterxml.jackson.databind.json.JsonMapper;
 import com.google.common.annotations.VisibleForTesting;
-import com.google.common.base.Preconditions;
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.ImmutableMap;
 import io.confluent.kafka.schemaregistry.client.SchemaRegistryClient;
@@ -52,10 +51,10 @@
 import io.confluent.ksql.test.tools.Record;
 import io.confluent.ksql.test.tools.TestCaseBuilderUtil;
 import io.confluent.ksql.test.tools.TestJsonMapper;
-import io.confluent.ksql.test.tools.Topic;
 import io.confluent.ksql.test.tools.TopicInfoCache;
 import io.confluent.ksql.test.tools.TopicInfoCache.TopicInfo;
 import io.confluent.ksql.test.util.EmbeddedSingleNodeKafkaCluster;
+import io.confluent.ksql.tools.test.model.Topic;
 import io.confluent.ksql.util.KsqlConfig;
 import io.confluent.ksql.util.KsqlConstants;
 import io.confluent.ksql.util.KsqlException;
@@ -69,18 +68,15 @@
 import java.math.BigDecimal;
 import java.net.URL;
 import java.time.Duration;
-import java.util.ArrayList;
 import java.util.Collection;
 import java.util.Collections;
 import java.util.HashMap;
-import java.util.HashSet;
 import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Map.Entry;
 import java.util.Objects;
 import java.util.Optional;
-import java.util.Set;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.Future;
 import java.util.concurrent.TimeUnit;
@@ -91,12 +87,6 @@
 import org.apache.kafka.clients.producer.KafkaProducer;
 import org.apache.kafka.clients.producer.ProducerRecord;
 import org.apache.kafka.clients.producer.RecordMetadata;
-import org.apache.kafka.common.IsolationLevel;
-import org.apache.kafka.common.TopicPartition;
-import org.apache.kafka.streams.KafkaStreams.State;
-import org.apache.kafka.streams.TopologyDescription;
-import org.apache.kafka.streams.TopologyDescription.Source;
-import org.apache.kafka.streams.TopologyDescription.Subtopology;
 import org.hamcrest.Matcher;
 import org.hamcrest.StringDescription;
 import org.json.JSONObject;

diff --git a/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/tools/StubKafkaServiceTest.java b/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/tools/StubKafkaServiceTest.java
@@ -22,6 +22,7 @@
 
 import io.confluent.kafka.schemaregistry.ParsedSchema;
 import io.confluent.ksql.test.tools.stubs.StubKafkaService;
+import io.confluent.ksql.tools.test.model.Topic;
 import java.util.Optional;
 import org.apache.kafka.clients.producer.ProducerRecord;
 import org.junit.Before;

diff --git a/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/tools/TestExecutorTest.java b/ksqldb-functional-tests/src/test/java/io/confluent/ksql/test/tools/TestExecutorTest.java
@@ -58,6 +58,7 @@
 import io.confluent.ksql.test.tools.TestExecutor.TopologyBuilder;
 import io.confluent.ksql.test.tools.conditions.PostConditions;
 import io.confluent.ksql.test.tools.stubs.StubKafkaService;
+import io.confluent.ksql.tools.test.model.Topic;
 import io.confluent.ksql.util.KsqlException;
 import java.util.HashMap;
 import java.util.List;