Function workflow_default

test/sqlsmith/mzcompose.py:88–233

(c: Composition, parser: WorkflowArgumentParser)

def workflow_default(c: Composition, parser: WorkflowArgumentParser) -> None:
    parser.add_argument("--num-sqlsmith", default=len(MZ_SERVERS), type=int)
    # parser.add_argument("--queries", default=10000, type=int)
    parser.add_argument("--runtime", default=600, type=int)
    parser.add_argument("--max-joins", default=5, type=int)
    parser.add_argument("--explain-only", action="store_true")
    parser.add_argument("--exclude-catalog", default=False, type=bool)
    parser.add_argument("--seed", default=None, type=int)
    args = parser.parse_args()

    c.up(*MZ_SERVERS)

    for mz_server in MZ_SERVERS:
        # Very simple data for our workload
        c.sql(
            """
            CREATE TABLE t1 (a int2, b int4, c int8, d uint2, e uint4, f uint8, g text);
            INSERT INTO t1 VALUES (1, 2, 3, 4, 5, 6, '7'), (3, 4, 5, 6, 7, 8, '9'), (5, 6, 7, 8, 9, 10, '11'), (7, 8, 9, 10, 11, 12, '13'), (9, 10, 11, 12, 13, 14, '15'), (11, 12, 13, 14, 15, 16, '17'), (13, 14, 15, 16, 17, 18, '19'), (15, 16, 17, 18, 19, 20, '21');
            CREATE MATERIALIZED VIEW mv AS SELECT a + b AS col1, c + d AS col2, e + f AS col3, g AS col4 FROM t1;
            CREATE MATERIALIZED VIEW mv2 AS SELECT count(*) FROM mv;
            CREATE DEFAULT INDEX ON mv;
            CREATE TABLE t2 (
                a_bool        BOOL,
                b_float4      FLOAT4,
                c_float8      FLOAT8,
                d_numeric     NUMERIC,
                e_char        CHAR(5),
                f_varchar     VARCHAR(10),
                g_bytes       BYTES,
                h_date        DATE,
                i_time        TIME,
                k_timestamp   TIMESTAMP,
                l_timestamptz TIMESTAMPTZ,
                m_interval    INTERVAL,
                n_jsonb       JSONB,
                o_uuid        UUID
            );

            INSERT INTO t2 VALUES
            (
                TRUE,
                1.23,
                4.56,
                7.89,
                'abc',
                'hello',
                '\x68656c6c6f',
                DATE '2023-01-01',
                TIME '12:34:56',
                TIMESTAMP '2023-01-01 12:34:56',
                TIMESTAMPTZ '2023-01-01 12:34:56+00',
                INTERVAL '1 day 2 hours',
                '{"key": "value"}',
                '550e8400-e29b-41d4-a716-446655440000'
            );

            CREATE MATERIALIZED VIEW mv3 AS SELECT * FROM t2;
            """,