/**
 * Test KILL QUERY functionality
 */

#include "testconnections.h"

int main(int argc, char* argv[])
{
    TestConnections test(argc, argv);

    for (int i = 0; i < 3; i++)
    {
        auto a = test.maxscales->rwsplit();
        auto b = test.maxscales->rwsplit();
        test.expect(a.connect() && b.connect(), "Connections should work");

        auto id = a.thread_id();

        std::thread thr(
            [&]() {
                const char* query =
                    "BEGIN NOT ATOMIC "
                    "  DECLARE v1 INT DEFAULT 5; "
                    "  CREATE OR REPLACE TABLE test.t1(id INT); "
                    "  WHILE v1 <> 0 DO "
                    "    INSERT INTO test.t1 VALUES (1); "
                    "    SET v1 = (SELECT COUNT(*) FROM test.t1); "
                    "  END WHILE;"
                    "END";

                    // The ALTER should take longer than 15 seconds to complete so that a KILL is required to
                // interrupt it.
                test.expect(!a.query(query), "Query should fail");

                const char* expected = "Query execution was interrupted";
                test.expect(strstr(a.error(), expected),
                            "Alter should fail with '%s' but it failed with '%s'",
                            expected, a.error());
            });

        // Wait for a few seconds to make sure the other thread has started executing the query before killing it.
        sleep(5);
        test.expect(b.query("KILL QUERY " + std::to_string(id)), "KILL QUERY failed: %s", b.error());

        test.set_timeout(15);
        thr.join();
        test.stop_timeout();
    }

    auto conn = test.maxscales->rwsplit();
    conn.connect();
    conn.query("DROP TABLE test.t1");

    return test.global_result;
}