# 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.

#
# Test that adding a column does not mess things up.
# We expect that deletes where the data is only different in this column will
# be handled correctly.
#

$ postgres-execute connection=postgres://postgres:postgres@postgres
CREATE TABLE alter_add_column_with_delete (f1 INTEGER, f2 INTEGER PRIMARY KEY);
ALTER TABLE alter_add_column_with_delete REPLICA IDENTITY FULL;
INSERT INTO alter_add_column_with_delete VALUES (123, 0),(123, 1),(123, 2),(123, 3);

$ schema-registry-wait topic=postgres.public.alter_add_column_with_delete

> CREATE CONNECTION IF NOT EXISTS csr_conn TO CONFLUENT SCHEMA REGISTRY (
    URL '${testdrive.schema-registry-url}'
  );

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

> CREATE SOURCE alter_add_column_with_delete
  FROM KAFKA CONNECTION kafka_conn (TOPIC 'postgres.public.alter_add_column_with_delete');

> CREATE TABLE alter_add_column_with_delete_tbl FROM SOURCE alter_add_column_with_delete (REFERENCE "postgres.public.alter_add_column_with_delete")
  FORMAT AVRO USING CONFLUENT SCHEMA REGISTRY CONNECTION csr_conn
  ENVELOPE DEBEZIUM;

> SELECT f1 FROM alter_add_column_with_delete_tbl;
123
123
123
123

$ postgres-execute connection=postgres://postgres:postgres@postgres
ALTER TABLE alter_add_column_with_delete ADD COLUMN new_column INTEGER DEFAULT 1;
INSERT INTO alter_add_column_with_delete VALUES (123,4,2);
INSERT INTO alter_add_column_with_delete VALUES (123,5,2);
DELETE FROM alter_add_column_with_delete WHERE new_column = 2;

# Even though we do not have new_column in our source, we expect that the
# updates above have landed on the appropriate distinct rows

> SELECT f1 FROM alter_add_column_with_delete_tbl;
123
123
123
123

$ postgres-execute connection=postgres://postgres:postgres@postgres
DELETE FROM alter_add_column_with_delete WHERE new_column = 1;

> SELECT COUNT(*) FROM alter_add_column_with_delete_tbl;
0