test_alter_statements.py

from simple_ddl_parser import DDLParser


def test_alter_table_initial_support():
    ddl = """CREATE TABLE "materials" (
        "id" int PRIMARY KEY,
        "title" varchar NOT NULL default "New title",
        "description" varchar,
        "link" varchar,
        "created_at" timestamp NOT NULL,
        "updated_at" timestamp NOT NULL
        );

        CREATE TABLE "material_attachments" (
        "material_id" int NOT NULL,
        "attachment_id" int NOT NULL
        );

        CREATE TABLE "attachments" (
        "id" int PRIMARY KEY,
        "title" varchar,
        "description" varchar,
        "created_at" timestamp NOT NULL,
        "updated_at" timestamp NOT NULL
        );
        ALTER TABLE "material_attachments" ADD FOREIGN KEY ("material_id") REFERENCES "materials" ("id");

        ALTER TABLE "material_attachments" ADD FOREIGN KEY ("attachment_id") REFERENCES "attachments" ("id");
    """
    expected = [
        {
            "columns": [
                {
                    "name": '"id"',
                    "type": "int",
                    "size": None,
                    "nullable": False,
                    "default": None,
                    "check": None,
                    "references": None,
                    "unique": False,
                },
                {
                    "name": '"title"',
                    "type": "varchar",
                    "size": None,
                    "nullable": False,
                    "default": '"New title"',
                    "check": None,
                    "references": None,
                    "unique": False,
                },
                {
                    "name": '"description"',
                    "type": "varchar",
                    "size": None,
                    "nullable": True,
                    "default": None,
                    "check": None,
                    "references": None,
                    "unique": False,
                },
                {
                    "name": '"link"',
                    "type": "varchar",
                    "size": None,
                    "nullable": True,
                    "default": None,
                    "check": None,
                    "references": None,
                    "unique": False,
                },
                {
                    "name": '"created_at"',
                    "type": "timestamp",
                    "size": None,
                    "nullable": False,
                    "default": None,
                    "check": None,
                    "references": None,
                    "unique": False,
                },
                {
                    "name": '"updated_at"',
                    "type": "timestamp",
                    "size": None,
                    "nullable": False,
                    "default": None,
                    "check": None,
                    "references": None,
                    "unique": False,
                },
            ],
            "primary_key": ['"id"'],
            "index": [],
            "partitioned_by": [],
            "alter": {},
            "checks": [],
            "table_name": '"materials"',
            "tablespace": None,
            "schema": None,
        },
        {
            "columns": [
                {
                    "name": '"material_id"',
                    "type": "int",
                    "size": None,
                    "nullable": False,
                    "default": None,
                    "check": None,
                    "references": None,
                    "unique": False,
                },
                {
                    "name": '"attachment_id"',
                    "type": "int",
                    "size": None,
                    "nullable": False,
                    "default": None,
                    "check": None,
                    "references": None,
                    "unique": False,
                },
            ],
            "primary_key": [],
            "index": [],
            "partitioned_by": [],
            "alter": {
                "columns": [
                    {
                        "name": '"material_id"',
                        "constraint_name": None,
                        "references": {
                            "column": '"id"',
                            "table": '"materials"',
                            "schema": None,
                            "on_update": None,
                            "on_delete": None,
                            "deferrable_initially": None,
                        },
                    },
                    {
                        "name": '"attachment_id"',
                        "constraint_name": None,
                        "references": {
                            "column": '"id"',
                            "table": '"attachments"',
                            "schema": None,
                            "on_update": None,
                            "on_delete": None,
                            "deferrable_initially": None,
                        },
                    },
                ]
            },
            "checks": [],
            "table_name": '"material_attachments"',
            "tablespace": None,
            "schema": None,
        },
        {
            "columns": [
                {
                    "name": '"id"',
                    "type": "int",
                    "size": None,
                    "nullable": False,
                    "default": None,
                    "check": None,
                    "references": None,
                    "unique": False,
                },
                {
                    "name": '"title"',
                    "type": "varchar",
                    "size": None,
                    "nullable": True,
                    "default": None,
                    "check": None,
                    "references": None,
                    "unique": False,
                },
                {
                    "name": '"description"',
                    "type": "varchar",
                    "size": None,
                    "nullable": True,
                    "default": None,
                    "check": None,
                    "references": None,
                    "unique": False,
                },
                {
                    "name": '"created_at"',
                    "type": "timestamp",
                    "size": None,
                    "nullable": False,
                    "default": None,
                    "check": None,
                    "references": None,
                    "unique": False,
                },
                {
                    "name": '"updated_at"',
                    "type": "timestamp",
                    "size": None,
                    "nullable": False,
                    "default": None,
                    "check": None,
                    "references": None,
                    "unique": False,
                },
            ],
            "primary_key": ['"id"'],
            "index": [],
            "partitioned_by": [],
            "alter": {},
            "checks": [],
            "table_name": '"attachments"',
            "tablespace": None,
            "schema": None,
        },
    ]
    parse_results = DDLParser(ddl).run()
    assert expected == parse_results


def test_alter_check():
    ddl = """CREATE TABLE Persons (
        ID int NOT NULL,
        LastName varchar(255) NOT NULL,
        FirstName varchar(255),
        Age int,
        City varchar(255),
        );

        ALTER TABLE Persons
        ADD CHECK (Age>=18);
    """
    expected = [
        {
            "columns": [
                {
                    "name": "ID",
                    "type": "int",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": False,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "LastName",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": False,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "FirstName",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "Age",
                    "type": "int",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "City",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
            ],
            "primary_key": [],
            "index": [],
            "alter": {"checks": [{"constraint_name": None, "statement": "Age>=18"}]},
            "checks": [],
            "table_name": "Persons",
            "tablespace": None,
            "schema": None,
            "partitioned_by": [],
        }
    ]

    assert DDLParser(ddl).run() == expected


def test_alter_check_combine_all_variants():

    ddl = """
    CREATE TABLE employees (
    id SERIAL PRIMARY KEY,
    first_name VARCHAR (50) default 'User Name',
    last_name VARCHAR (50) default 'User Last Name',
    birth_date DATE CHECK (birth_date > '1900-01-01'),
    joined_date DATE CHECK (joined_date > birth_date),
    salary numeric CHECK(salary > 0)
    );
    CREATE TABLE Persons (
    ID int NOT NULL,
    LastName varchar(255) NOT NULL,
    FirstName varchar(255),
    Age int,
    City varchar(255),
    CONSTRAINT CHK_Person CHECK (Age>=18 AND City='Sandnes')
    );

    ALTER TABLE Persons
    ADD CHECK (Age>=18);
    """
    expected = [
        {
            "columns": [
                {
                    "name": "id",
                    "type": "SERIAL",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": False,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "first_name",
                    "type": "VARCHAR",
                    "size": 50,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": "'User Name'",
                    "check": None,
                },
                {
                    "name": "last_name",
                    "type": "VARCHAR",
                    "size": 50,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": "'User Last Name'",
                    "check": None,
                },
                {
                    "name": "birth_date",
                    "type": "DATE",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": "birth_date > '1900-01-01'",
                },
                {
                    "name": "joined_date",
                    "type": "DATE",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": "joined_date > birth_date",
                },
                {
                    "check": "salary > 0",
                    "default": None,
                    "name": "salary",
                    "nullable": True,
                    "references": None,
                    "size": None,
                    "type": "numeric",
                    "unique": False,
                },
            ],
            "primary_key": ["id"],
            "index": [],
            "alter": {},
            "checks": [],
            "table_name": "employees",
            "tablespace": None,
            "schema": None,
            "partitioned_by": [],
        },
        {
            "columns": [
                {
                    "name": "ID",
                    "type": "int",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": False,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "LastName",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": False,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "FirstName",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "Age",
                    "type": "int",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "City",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
            ],
            "primary_key": [],
            "index": [],
            "alter": {"checks": [{"constraint_name": None, "statement": "Age>=18"}]},
            "checks": [
                {
                    "constraint_name": "CHK_Person",
                    "statement": "Age>=18 AND City= 'Sandnes'",
                }
            ],
            "constraints": {
                "checks": [
                    {
                        "constraint_name": "CHK_Person",
                        "statement": "Age>=18 AND City= 'Sandnes'",
                    }
                ]
            },
            "table_name": "Persons",
            "tablespace": None,
            "schema": None,
            "partitioned_by": [],
        },
    ]

    assert expected == DDLParser(ddl).run()


def test_alter_check_with_constraint():
    parse_results = DDLParser(
        """

        CREATE TABLE Persons (
        ID int NOT NULL,
        LastName varchar(255) NOT NULL,
        FirstName varchar(255),
        Age int,
        City varchar(255),

        );
        Alter Table Persons ADD CONSTRAINT CHK_PersonAge CHECK (Age>=18 AND City='Sandnes');
"""
    ).run()
    expected = [
        {
            "columns": [
                {
                    "name": "ID",
                    "type": "int",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": False,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "LastName",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": False,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "FirstName",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "Age",
                    "type": "int",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "City",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
            ],
            "primary_key": [],
            "index": [],
            "alter": {
                "checks": [
                    {
                        "constraint_name": "CHK_PersonAge",
                        "statement": "Age>=18 AND City= 'Sandnes'",
                    }
                ]
            },
            "checks": [],
            "table_name": "Persons",
            "tablespace": None,
            "schema": None,
            "partitioned_by": [],
        }
    ]
    assert expected == parse_results


def test_alter_foreiggn_with_constraint():
    parse_results = DDLParser(
        """
        CREATE TABLE Persons (
        ID int NOT NULL,
        LastName varchar(255) NOT NULL,
        FirstName varchar(255),
        Age int,
        City varchar(255),
        CONSTRAINT CHK_Person CHECK (Age>=18 AND City='Sandnes')
        );

        Alter Table Persons ADD CONSTRAINT fk_group FOREIGN KEY (id) REFERENCES employees (id);
        """
    ).run()
    expected = [
        {
            "columns": [
                {
                    "name": "ID",
                    "type": "int",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": False,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "LastName",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": False,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "FirstName",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "Age",
                    "type": "int",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "City",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
            ],
            "primary_key": [],
            "index": [],
            "alter": {
                "columns": [
                    {
                        "name": "id",
                        "constraint_name": "fk_group",
                        "references": {
                            "column": "id",
                            "table": "employees",
                            "schema": None,
                            "on_update": None,
                            "on_delete": None,
                            "deferrable_initially": None,
                        },
                    }
                ]
            },
            "constraints": {
                "checks": [
                    {
                        "constraint_name": "CHK_Person",
                        "statement": "Age>=18 AND City= 'Sandnes'",
                    }
                ]
            },
            "checks": [
                {
                    "constraint_name": "CHK_Person",
                    "statement": "Age>=18 AND City= 'Sandnes'",
                }
            ],
            "table_name": "Persons",
            "tablespace": None,
            "schema": None,
            "partitioned_by": [],
        }
    ]
    assert parse_results == expected


def test_alter_without_constraint_and_constraint_in_table():
    parse_results = DDLParser(
        """
        CREATE TABLE Persons (
        ID int NOT NULL,
        LastName varchar(255) NOT NULL,
        FirstName varchar(255),
        Age int,
        City varchar(255),
        CONSTRAINT CHK_Person CHECK (Age>=18 AND City= 'Sandnes')
        );

        ALTER TABLE Persons ADD CHECK (Age>=18 AND City= 'Sandnes');

        """
    ).run()
    expected = [
        {
            "columns": [
                {
                    "name": "ID",
                    "type": "int",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": False,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "LastName",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": False,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "FirstName",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "Age",
                    "type": "int",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "City",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
            ],
            "primary_key": [],
            "index": [],
            "alter": {
                "checks": [
                    {
                        "constraint_name": None,
                        "statement": "Age>=18 AND City= 'Sandnes'",
                    }
                ]
            },
            "checks": [
                {
                    "constraint_name": "CHK_Person",
                    "statement": "Age>=18 AND City= 'Sandnes'",
                }
            ],
            "constraints": {
                "checks": [
                    {
                        "constraint_name": "CHK_Person",
                        "statement": "Age>=18 AND City= 'Sandnes'",
                    }
                ]
            },
            "table_name": "Persons",
            "tablespace": None,
            "schema": None,
            "partitioned_by": [],
        }
    ]
    assert expected == parse_results


def test_combo_with_alter_and_table_constraints():
    parse_results = DDLParser(
        """
    CREATE TABLE employees (
        id SERIAL PRIMARY KEY,
        first_name VARCHAR (50),
        last_name VARCHAR (50),
        birth_date DATE CHECK (birth_date > '1900-01-01'),
        joined_date DATE CHECK (joined_date > birth_date),
        salary numeric CHECK(salary > 0)
    );
    CREATE TABLE Persons (
        ID int NOT NULL,
        LastName varchar(255) NOT NULL,
        FirstName varchar(255),
        Age int,
        City varchar(255),
        CONSTRAINT CHK_Person CHECK (Age>=18 AND City='Sandnes')
    );
    ALTER TABLE Persons ADD CHECK (Age>=18 AND City='Sandnes');
    ALTER TABLE Persons Add CONSTRAINT ck_person  CHECK (Age>=18 AND City='Sandnes');
    Alter Table Persons ADD CONSTRAINT fk_group FOREIGN KEY (id) REFERENCES employees (id);"""
    ).run()

    expected = [
        {
            "columns": [
                {
                    "name": "id",
                    "type": "SERIAL",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": False,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "first_name",
                    "type": "VARCHAR",
                    "size": 50,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "last_name",
                    "type": "VARCHAR",
                    "size": 50,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "birth_date",
                    "type": "DATE",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": "birth_date > '1900-01-01'",
                },
                {
                    "name": "joined_date",
                    "type": "DATE",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": "joined_date > birth_date",
                },
                {
                    "check": "salary > 0",
                    "default": None,
                    "name": "salary",
                    "nullable": True,
                    "references": None,
                    "size": None,
                    "type": "numeric",
                    "unique": False,
                },
            ],
            "primary_key": ["id"],
            "index": [],
            "alter": {},
            "checks": [],
            "table_name": "employees",
            "tablespace": None,
            "schema": None,
            "partitioned_by": [],
        },
        {
            "columns": [
                {
                    "name": "ID",
                    "type": "int",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": False,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "LastName",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": False,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "FirstName",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "Age",
                    "type": "int",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "City",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
            ],
            "primary_key": [],
            "index": [],
            "alter": {
                "checks": [
                    {
                        "constraint_name": None,
                        "statement": "Age>=18 AND City= 'Sandnes'",
                    },
                    {
                        "constraint_name": "ck_person",
                        "statement": "Age>=18 AND City= 'Sandnes'",
                    },
                ],
                "columns": [
                    {
                        "name": "id",
                        "constraint_name": "fk_group",
                        "references": {
                            "column": "id",
                            "table": "employees",
                            "schema": None,
                            "on_update": None,
                            "on_delete": None,
                            "deferrable_initially": None,
                        },
                    }
                ],
            },
            "checks": [
                {
                    "constraint_name": "CHK_Person",
                    "statement": "Age>=18 AND City= 'Sandnes'",
                }
            ],
            "constraints": {
                "checks": [
                    {
                        "constraint_name": "CHK_Person",
                        "statement": "Age>=18 AND City= 'Sandnes'",
                    }
                ]
            },
            "table_name": "Persons",
            "tablespace": None,
            "schema": None,
            "partitioned_by": [],
        },
    ]
    assert expected == parse_results


def test_alter_foreign_with_multiple_ids():
    parse_results = DDLParser(
        """
CREATE TABLE employees (
            id SERIAL PRIMARY KEY,
            first_name VARCHAR (50),
            last_name VARCHAR (50),
            birth_date DATE CHECK (birth_date > '1900-01-01'),
            joined_date DATE CHECK (joined_date > birth_date),
            salary numeric CHECK(salary > 0)
            Age int,
        );
        CREATE TABLE Persons (
            ID int NOT NULL,
            LastName varchar(255) NOT NULL,
            FirstName varchar(255),
            Age int,
            City varchar(255),
            birth_date DATE CHECK (birth_date > '1900-01-01'),
            CONSTRAINT CHK_Person CHECK (Age>=18 AND City='Sandnes')
        );
        Alter Table Persons ADD CONSTRAINT fk_group FOREIGN KEY (id, Age, birth_date) REFERENCES employees (id, Age, birth_date);
"""  # noqa E501
    ).run()
    expected = [
        {
            "columns": [
                {
                    "name": "id",
                    "type": "SERIAL",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": False,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "first_name",
                    "type": "VARCHAR",
                    "size": 50,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "last_name",
                    "type": "VARCHAR",
                    "size": 50,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "birth_date",
                    "type": "DATE",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": "birth_date > '1900-01-01'",
                },
                {
                    "name": "joined_date",
                    "type": "DATE",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": "joined_date > birth_date",
                },
                {
                    "name": "salary",
                    "type": "numeric",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": "salary > 0 Age int",
                },
            ],
            "primary_key": ["id"],
            "alter": {},
            "checks": [],
            "index": [],
            "schema": None,
            "partitioned_by": [],
            "table_name": "employees",
            "tablespace": None,
        },
        {
            "columns": [
                {
                    "name": "ID",
                    "type": "int",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": False,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "LastName",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": False,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "FirstName",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "Age",
                    "type": "int",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "City",
                    "type": "varchar",
                    "size": 255,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": None,
                },
                {
                    "name": "birth_date",
                    "type": "DATE",
                    "size": None,
                    "references": None,
                    "unique": False,
                    "nullable": True,
                    "default": None,
                    "check": "birth_date > '1900-01-01'",
                },
            ],
            "primary_key": [],
            "alter": {
                "columns": [
                    {
                        "name": "id",
                        "constraint_name": "fk_group",
                        "references": {
                            "column": "id",
                            "table": "employees",
                            "schema": None,
                            "on_update": None,
                            "on_delete": None,
                            "deferrable_initially": None,
                        },
                    },
                    {
                        "name": "Age",
                        "constraint_name": "fk_group",
                        "references": {
                            "column": "Age",
                            "table": "employees",
                            "schema": None,
                            "on_update": None,
                            "on_delete": None,
                            "deferrable_initially": None,
                        },
                    },
                    {
                        "name": "birth_date",
                        "constraint_name": "fk_group",
                        "references": {
                            "column": "birth_date",
                            "table": "employees",
                            "schema": None,
                            "on_update": None,
                            "on_delete": None,
                            "deferrable_initially": None,
                        },
                    },
                ]
            },
            "checks": [
                {
                    "constraint_name": "CHK_Person",
                    "statement": "Age>=18 AND City= 'Sandnes'",
                }
            ],
            "constraints": {
                "checks": [
                    {
                        "constraint_name": "CHK_Person",
                        "statement": "Age>=18 AND City= 'Sandnes'",
                    }
                ]
            },
            "index": [],
            "schema": None,
            "partitioned_by": [],
            "table_name": "Persons",
            "tablespace": None,
        },
    ]
    assert expected == parse_results


def test_several_alter_fk_for_same_table():
    parse_results = DDLParser(
        """
CREATE TABLE employees (
            id SERIAL PRIMARY KEY,
            first_name VARCHAR (50),
            last_name VARCHAR (50),
            birth_date DATE CHECK (birth_date > '1900-01-01'),
            joined_date DATE CHECK (joined_date > birth_date),
            salary numeric CHECK(salary > 0)
            Age int,
        );
        CREATE TABLE Persons (
            ID int NOT NULL,
            LastName varchar(255) NOT NULL,
            FirstName varchar(255),
            Age int,
            City varchar(255),
            birth_date DATE CHECK (birth_date > '1900-01-01'),
            CONSTRAINT CHK_Person CHECK (Age>=18 AND City='Sandnes')
        );
        Alter Table Persons ADD CONSTRAINT fk_group FOREIGN KEY (id, Age) REFERENCES employees (id, Age, birth_date);
        Alter Table Persons ADD CONSTRAINT new_fk FOREIGN KEY (birth_date) REFERENCES employees (birth_date);"""
    ).run()
    expected = [
        {
            "alter": {},
            "checks": [],
            "columns": [
                {
                    "check": None,
                    "default": None,
                    "name": "id",
                    "nullable": False,
                    "references": None,
                    "size": None,
                    "type": "SERIAL",
                    "unique": False,
                },
                {
                    "check": None,
                    "default": None,
                    "name": "first_name",
                    "nullable": True,
                    "references": None,
                    "size": 50,
                    "type": "VARCHAR",
                    "unique": False,
                },
                {
                    "check": None,
                    "default": None,
                    "name": "last_name",
                    "nullable": True,
                    "references": None,
                    "size": 50,
                    "type": "VARCHAR",
                    "unique": False,
                },
                {
                    "check": "birth_date > '1900-01-01'",
                    "default": None,
                    "name": "birth_date",
                    "nullable": True,
                    "references": None,
                    "size": None,
                    "type": "DATE",
                    "unique": False,
                },
                {
                    "check": "joined_date > birth_date",
                    "default": None,
                    "name": "joined_date",
                    "nullable": True,
                    "references": None,
                    "size": None,
                    "type": "DATE",
                    "unique": False,
                },
                {
                    "check": "salary > 0 Age int",
                    "default": None,
                    "name": "salary",
                    "nullable": True,
                    "references": None,
                    "size": None,
                    "type": "numeric",
                    "unique": False,
                },
            ],
            "index": [],
            "partitioned_by": [],
            "primary_key": ["id"],
            "schema": None,
            "table_name": "employees",
            "tablespace": None,
        },
        {
            "alter": {
                "columns": [
                    {
                        "constraint_name": "fk_group",
                        "name": "id",
                        "references": {
                            "column": "id",
                            "deferrable_initially": None,
                            "on_delete": None,
                            "on_update": None,
                            "schema": None,
                            "table": "employees",
                        },
                    },
                    {
                        "constraint_name": "fk_group",
                        "name": "Age",
                        "references": {
                            "column": "Age",
                            "deferrable_initially": None,
                            "on_delete": None,
                            "on_update": None,
                            "schema": None,
                            "table": "employees",
                        },
                    },
                    {
                        "constraint_name": "new_fk",
                        "name": "birth_date",
                        "references": {
                            "column": "birth_date",
                            "deferrable_initially": None,
                            "on_delete": None,
                            "on_update": None,
                            "schema": None,
                            "table": "employees",
                        },
                    },
                ]
            },
            "checks": [
                {
                    "constraint_name": "CHK_Person",
                    "statement": "Age>=18 AND City= 'Sandnes'",
                }
            ],
            "columns": [
                {
                    "check": None,
                    "default": None,
                    "name": "ID",
                    "nullable": False,
                    "references": None,
                    "size": None,
                    "type": "int",
                    "unique": False,
                },
                {
                    "check": None,
                    "default": None,
                    "name": "LastName",
                    "nullable": False,
                    "references": None,
                    "size": 255,
                    "type": "varchar",
                    "unique": False,
                },
                {
                    "check": None,
                    "default": None,
                    "name": "FirstName",
                    "nullable": True,
                    "references": None,
                    "size": 255,
                    "type": "varchar",
                    "unique": False,
                },
                {
                    "check": None,
                    "default": None,
                    "name": "Age",
                    "nullable": True,
                    "references": None,
                    "size": None,
                    "type": "int",
                    "unique": False,
                },
                {
                    "check": None,
                    "default": None,
                    "name": "City",
                    "nullable": True,
                    "references": None,
                    "size": 255,
                    "type": "varchar",
                    "unique": False,
                },
                {
                    "check": "birth_date > '1900-01-01'",
                    "default": None,
                    "name": "birth_date",
                    "nullable": True,
                    "references": None,
                    "size": None,
                    "type": "DATE",
                    "unique": False,
                },
            ],
            "index": [],
            "partitioned_by": [],
            "primary_key": [],
            "constraints": {
                "checks": [
                    {
                        "constraint_name": "CHK_Person",
                        "statement": "Age>=18 AND City= 'Sandnes'",
                    }
                ]
            },
            "schema": None,
            "table_name": "Persons",
            "tablespace": None,
        },
    ]
    assert expected == parse_results


def test_alter_table_only():
    ddl = """
    CREATE TABLE public.accounts (
        user_id integer NOT NULL,
        username character varying(50) NOT NULL,
        password character varying(50) NOT NULL,
        email character varying(255) NOT NULL,
        created_on timestamp without time zone NOT NULL,
        last_login timestamp without time zone
    );
    ALTER TABLE ONLY public.accounts
        ADD CONSTRAINT accounts_username_key UNIQUE (username);
        """
    result = DDLParser(ddl).run(group_by_type=True, output_mode="bigquery")
    expected = {
        "ddl_properties": [],
        "domains": [],
        "schemas": [],
        "sequences": [],
        "tables": [
            {
                "alter": {
                    "uniques": [
                        {
                            "columns": ["username"],
                            "constraint_name": "accounts_username_key",
                        }
                    ]
                },
                "checks": [],
                "columns": [
                    {
                        "check": None,
                        "default": None,
                        "name": "user_id",
                        "nullable": False,
                        "references": None,
                        "size": None,
                        "type": "integer",
                        "unique": False,
                    },
                    {
                        "check": None,
                        "default": None,
                        "name": "username",
                        "nullable": False,
                        "references": None,
                        "size": 50,
                        "type": "character varying",
                        "unique": True,
                    },
                    {
                        "check": None,
                        "default": None,
                        "name": "password",
                        "nullable": False,
                        "references": None,
                        "size": 50,
                        "type": "character varying",
                        "unique": False,
                    },
                    {
                        "check": None,
                        "default": None,
                        "name": "email",
                        "nullable": False,
                        "references": None,
                        "size": 255,
                        "type": "character varying",
                        "unique": False,
                    },
                    {
                        "check": None,
                        "default": None,
                        "name": "created_on",
                        "nullable": False,
                        "references": None,
                        "size": None,
                        "type": "timestamp without time zone",
                        "unique": False,
                    },
                    {
                        "check": None,
                        "default": None,
                        "name": "last_login",
                        "nullable": True,
                        "references": None,
                        "size": None,
                        "type": "timestamp without time zone",
                        "unique": False,
                    },
                ],
                "dataset": "public",
                "index": [],
                "partitioned_by": [],
                "primary_key": [],
                "table_name": "accounts",
                "tablespace": None,
            }
        ],
        "types": [],
    }
    assert expected == result


def test_alter_table_if_exists():
    ddl = """
    CREATE TABLE public.accounts (
        user_id integer NOT NULL,
        username character varying(50) NOT NULL,
        password character varying(50) NOT NULL,
        email character varying(255) NOT NULL,
        created_on timestamp without time zone NOT NULL,
        last_login timestamp without time zone
    );
    ALTER TABLE IF EXISTS public.accounts
        ADD CONSTRAINT accounts_username_key UNIQUE (username);
        """
    result = DDLParser(ddl).run(group_by_type=True, output_mode="bigquery")
    expected = {
        "ddl_properties": [],
        "domains": [],
        "schemas": [],
        "sequences": [],
        "tables": [
            {
                "alter": {
                    "uniques": [
                        {
                            "columns": ["username"],
                            "constraint_name": "accounts_username_key",
                        }
                    ]
                },
                "checks": [],
                "columns": [
                    {
                        "check": None,
                        "default": None,
                        "name": "user_id",
                        "nullable": False,
                        "references": None,
                        "size": None,
                        "type": "integer",
                        "unique": False,
                    },
                    {
                        "check": None,
                        "default": None,
                        "name": "username",
                        "nullable": False,
                        "references": None,
                        "size": 50,
                        "type": "character varying",
                        "unique": True,
                    },
                    {
                        "check": None,
                        "default": None,
                        "name": "password",
                        "nullable": False,
                        "references": None,
                        "size": 50,
                        "type": "character varying",
                        "unique": False,
                    },
                    {
                        "check": None,
                        "default": None,
                        "name": "email",
                        "nullable": False,
                        "references": None,
                        "size": 255,
                        "type": "character varying",
                        "unique": False,
                    },
                    {
                        "check": None,
                        "default": None,
                        "name": "created_on",
                        "nullable": False,
                        "references": None,
                        "size": None,
                        "type": "timestamp without time zone",
                        "unique": False,
                    },
                    {
                        "check": None,
                        "default": None,
                        "name": "last_login",
                        "nullable": True,
                        "references": None,
                        "size": None,
                        "type": "timestamp without time zone",
                        "unique": False,
                    },
                ],
                "dataset": "public",
                "index": [],
                "partitioned_by": [],
                "primary_key": [],
                "table_name": "accounts",
                "tablespace": None,
            }
        ],
        "types": [],
    }
    assert expected == result


def test_alter_table_primary_key():
    expected = {
        "ddl_properties": [],
        "domains": [],
        "schemas": [],
        "sequences": [],
        "tables": [
            {
                "alter": {
                    "primary_keys": [
                        {
                            "columns": ["DATETIME", "REGIONID"],
                            "constraint_name": "PK_MY_TABLE",
                        }
                    ]
                },
                "checks": [],
                "columns": [
                    {
                        "check": None,
                        "default": None,
                        "name": "DATETIME",
                        "nullable": True,
                        "references": None,
                        "size": None,
                        "type": "datetime",
                        "unique": False,
                    },
                    {
                        "check": None,
                        "default": None,
                        "name": "REGIONID",
                        "nullable": True,
                        "references": None,
                        "size": None,
                        "type": "varchar",
                        "unique": False,
                    },
                ],
                "index": [],
                "partitioned_by": [],
                "primary_key": [],
                "schema": None,
                "table_name": "MY_TABLE",
                "tablespace": None,
            }
        ],
        "types": [],
    }

    ddl = """
    CREATE TABLE MY_TABLE (
        DATETIME datetime,
        REGIONID varchar
    )
    alter table MY_TABLE
    add constraint PK_MY_TABLE primary key (DATETIME, REGIONID);"""
    result = DDLParser(ddl, silent=False, normalize_names=True).run(group_by_type=True)
    assert expected == result


def test_alter_using():

    ddl = """
    CREATE TABLE MY_TABLE (
        DATETIME datetime,
        REGIONID varchar
    );
    alter table MY_TABLE
    add constraint PK_MY_TABLE primary key (DATETIME, REGIONID)
        using index tablespace SOME_TABLE_SPACE"""
    result = DDLParser(ddl, silent=False, normalize_names=True).run(group_by_type=True)
    expected = {
        "ddl_properties": [],
        "domains": [],
        "schemas": [],
        "sequences": [],
        "tables": [
            {
                "alter": {
                    "primary_keys": [
                        {
                            "columns": ["DATETIME", "REGIONID"],
                            "constraint_name": "PK_MY_TABLE",
                            "using": {
                                "index": True,
                                "tablespace": {
                                    "properties": None,
                                    "tablespace_name": "SOME_TABLE_SPACE",
                                    "temporary": False,
                                    "type": None,
                                },
                            },
                        }
                    ]
                },
                "checks": [],
                "columns": [
                    {
                        "check": None,
                        "default": None,
                        "name": "DATETIME",
                        "nullable": True,
                        "references": None,
                        "size": None,
                        "type": "datetime",
                        "unique": False,
                    },
                    {
                        "check": None,
                        "default": None,
                        "name": "REGIONID",
                        "nullable": True,
                        "references": None,
                        "size": None,
                        "type": "varchar",
                        "unique": False,
                    },
                ],
                "index": [],
                "partitioned_by": [],
                "primary_key": [],
                "schema": None,
                "table_name": "MY_TABLE",
                "tablespace": None,
            }
        ],
        "types": [],
    }
    assert expected == result