#!/usr/bin/env python3
# Tags: no-parallel

import os
import sys

CURDIR = os.path.dirname(os.path.realpath(__file__))
sys.path.insert(0, os.path.join(CURDIR, "helpers"))

from client import client, end_of_block, prompt

log = None
# uncomment the line below for debugging
# log=sys.stdout

with client(name="client1>", log=log) as client1, client(
    name="client2>", log=log
) as client2, client(name="client3>", log=log) as client3:
    client1.expect(prompt)
    client2.expect(prompt)
    client3.expect(prompt)

    client1.send("SET enable_analyzer = 0")
    client1.expect(prompt)
    client1.send("SET allow_experimental_window_view = 1")
    client1.expect(prompt)
    client1.send("SET window_view_heartbeat_interval = 1")
    client1.expect(prompt)
    client2.send("SET allow_experimental_window_view = 1")
    client2.expect(prompt)
    client2.send("SET enable_analyzer = 0")
    client2.expect(prompt)
    client3.send("SET allow_experimental_window_view = 1")
    client3.expect(prompt)
    client3.send("SET window_view_heartbeat_interval = 1")
    client3.expect(prompt)
    client3.send("SET enable_analyzer = 0")
    client3.expect(prompt)

    client1.send("CREATE DATABASE IF NOT EXISTS 01078_window_view_alter_query_watch")
    client1.expect(prompt)
    client1.send("DROP TABLE IF EXISTS 01078_window_view_alter_query_watch.mt SYNC")
    client1.expect(prompt)
    client1.send("DROP TABLE IF EXISTS 01078_window_view_alter_query_watch.wv SYNC")
    client1.expect(prompt)

    client1.send(
        "CREATE TABLE 01078_window_view_alter_query_watch.mt(a Int32, timestamp DateTime) ENGINE=MergeTree ORDER BY tuple()"
    )
    client1.expect(prompt)
    client1.send(
        "CREATE WINDOW VIEW 01078_window_view_alter_query_watch.wv ENGINE Memory WATERMARK=ASCENDING AS SELECT count(a) AS count, hopEnd(wid) AS w_end FROM 01078_window_view_alter_query_watch.mt GROUP BY hop(timestamp, INTERVAL '2' SECOND, INTERVAL '3' SECOND, 'US/Samoa') AS wid"
    )
    client1.expect(prompt)

    client1.send("WATCH 01078_window_view_alter_query_watch.wv")
    client1.expect("Query id" + end_of_block)
    client1.expect("Progress: 0.00 rows.*\\)")
    client2.send(
        "INSERT INTO 01078_window_view_alter_query_watch.mt VALUES (1, toDateTime('1990/01/01 12:00:00', 'US/Samoa'));"
    )
    client2.expect("Ok.")
    client2.send(
        "INSERT INTO 01078_window_view_alter_query_watch.mt VALUES (1, toDateTime('1990/01/01 12:00:05', 'US/Samoa'));"
    )
    client2.expect("Ok.")
    client1.expect("1" + end_of_block)
    client1.expect("Progress: 1.00 rows.*\\)")
    client2.send(
        "ALTER TABLE 01078_window_view_alter_query_watch.wv MODIFY QUERY SELECT count(a) * 2 AS count, hopEnd(wid) AS w_end FROM 01078_window_view_alter_query_watch.mt GROUP BY hop(timestamp, INTERVAL '2' SECOND, INTERVAL '3' SECOND, 'US/Samoa') AS wid"
    )
    client2.expect("Ok.")
    client2.expect(prompt)
    client1.expect("1 row" + end_of_block)
    client1.expect(prompt)
    client3.send("WATCH 01078_window_view_alter_query_watch.wv")
    client3.expect("Query id" + end_of_block)
    client3.expect("Progress: 0.00 rows.*\\)")
    client2.send(
        "INSERT INTO 01078_window_view_alter_query_watch.mt VALUES (1, toDateTime('1990/01/01 12:00:06', 'US/Samoa'));"
    )
    client2.expect("Ok.")
    client2.send(
        "INSERT INTO 01078_window_view_alter_query_watch.mt VALUES (1, toDateTime('1990/01/01 12:00:10', 'US/Samoa'));"
    )
    client2.expect("Ok.")
    client3.expect("2" + end_of_block)
    client3.expect("Progress: 1.00 rows.*\\)")

    # send Ctrl-C
    client3.send("\x03", eol="")
    match = client3.expect("(%s)|([#\\$] )" % prompt)
    if match.groups()[1]:
        client3.send(client3.command)
        client3.expect(prompt)
    client3.send("DROP TABLE 01078_window_view_alter_query_watch.wv SYNC;")
    client3.expect(prompt)
    client3.send("DROP TABLE 01078_window_view_alter_query_watch.mt;")
    client3.expect(prompt)
    client3.send("DROP DATABASE IF EXISTS 01078_window_view_alter_query_watch;")
    client3.expect(prompt)