# Copyright Materialize, Inc. and contributors. All rights reserved.
#
# Use of this software is governed by the Business Source License
# included in the LICENSE file at the root of this repository.
#
# As of the Change Date specified in that file, in accordance with
# the Business Source License, use of this software will be governed
# by the Apache License, Version 2.0.

$ set-arg-default default-storage-size=1

$ kafka-create-topic topic=input_csv partitions=1
$ kafka-create-topic topic=input_csv_partitioned partitions=2
$ kafka-create-topic topic=input_proto

> CREATE CONNECTION kafka_conn
  TO KAFKA (BROKER '${testdrive.kafka-addr}', SECURITY PROTOCOL PLAINTEXT);

> CREATE CLUSTER input_csv_cluster SIZE '${arg.default-storage-size}';
> CREATE SOURCE input_csv
  IN CLUSTER input_csv_cluster
  FROM KAFKA CONNECTION kafka_conn (TOPIC 'testdrive-input_csv-${testdrive.seed}');

> CREATE TABLE input_csv_tbl (first, second)
  FROM SOURCE input_csv (REFERENCE "testdrive-input_csv-${testdrive.seed}")
  FORMAT CSV WITH 2 COLUMNS;

$ kafka-ingest format=bytes topic=input_csv
1,2
2,3

> SELECT * from input_csv_tbl;
first second
-------------
1     2
2     3

$ kafka-ingest format=bytes topic=input_csv_partitioned partition=0
1,2

$ kafka-ingest format=bytes topic=input_csv_partitioned partition=1
2,3

> CREATE CLUSTER input_csv_partitioned_cluster SIZE '${arg.default-storage-size}';
> CREATE SOURCE input_csv_partitioned
  IN CLUSTER input_csv_partitioned_cluster
  FROM KAFKA CONNECTION kafka_conn (START OFFSET=[1,0], TOPIC 'testdrive-input_csv_partitioned-${testdrive.seed}');

> CREATE TABLE input_csv_partitioned_tbl (first, second)
  FROM SOURCE input_csv_partitioned (REFERENCE "testdrive-input_csv_partitioned-${testdrive.seed}")
  FORMAT CSV WITH 2 COLUMNS;

> SELECT * FROM input_csv_partitioned_tbl
first second
------------
2     3