drift/sqlparser/test/analysis/regression_test.dart

import 'package:sqlparser/sqlparser.dart';
import 'package:test/test.dart';

import 'data.dart';

void main() {
  test('regression test for #917', () {
    // Test for https://github.com/simolus3/drift/issues/917
    final engine = SqlEngine()
      ..registerTableFromSql('CREATE TABLE Shops (id INTEGER);')
      ..registerTableFromSql(
          'CREATE TABLE Sales (shop_id INTEGER, date INTEGER, amount REAL);');

    final result = engine.analyze('''
      SELECT
          Shops.id,
          SalesYesterday.amount AS amount_yesterday,
          SalesToday.amount AS amount_today
      FROM Sales AS SalesYesterday,
      (
          SELECT SalesToday.amount AS amount, SalesToday.shop_id AS shop_id
          FROM Sales AS SalesToday
          LEFT JOIN Shops ON SalesToday.shop_id = Shops.id
          WHERE SalesToday.date = 8
      ) AS SalesToday
      INNER JOIN Shops
          ON (SalesYesterday.shop_id = Shops.id)
          AND (SalesToday.shop_id = Shops.id)
      WHERE SalesYesterday.date = 9
      ORDER BY Shops.id;
      ''');

    expect(result.errors, isEmpty);
  });

  test('regression test for #1188', () {
    // Test for https://github.com/simolus3/drift/issues/1188
    final engine = SqlEngine(EngineOptions(useDriftExtensions: true))
      ..registerTableFromSql('''
        CREATE TABLE IF NOT EXISTS "employees" (
     	    "id" INTEGER NOT NULL PRIMARY KEY,
     	    "name" TEXT NOT NULL UNIQUE,
     	    "manager_id" INTEGER
     	  );
      ''')
      ..registerTableFromSql('''
        CREATE TABLE IF NOT EXISTS "employee_notes" (
          "employee_id"	INTEGER NOT NULL PRIMARY KEY,
          "note" TEXT
        );
      ''');

    final result = engine.analyze('''
      WITH RECURSIVE
      employeeHierarchy(id, name, manager_id) AS (
        SELECT id,
               name,
               manager_id
          FROM employees
          WHERE manager_id IS NULL
          UNION ALL
        SELECT e.id,
               e.name,
               e.manager_id
          FROM employees e
          JOIN employeeHierarchy ON e.manager_id = employeeHierarchy.id
      )
      SELECT e.id,
            e.name,
            e.manager_id,
            n.note
        FROM employeeHierarchy e
        LEFT OUTER JOIN
            employee_notes n ON e.id = n.employee_id
      ORDER BY e.id;
    ''');

    final select = result.root as SelectStatement;
    final columns = select.resolvedColumns!;

    expect(columns.map((e) => e.name), ['id', 'name', 'manager_id', 'note']);
    expect(columns.map((e) => result.typeOf(e).nullable),
        [false, false, true, true]);
  });

  test('regression test for #1234', () {
    // https://github.com/simolus3/drift/issues/1234#issuecomment-853270925
    final engine = SqlEngine(EngineOptions(useDriftExtensions: true))
      ..registerTableFromSql('''
        CREATE TABLE inboxes (
          id TEXT PRIMARY KEY NOT NULL,
          group_id TEXT NOT NULL
        );
      ''')
      ..registerTableFromSql('''
        CREATE TABLE assignable_users (
          user_id TEXT NOT NULL,
          inbox_id TEXT NOT NULL
        );
      ''');

    final result = engine.analyze('''
      SELECT * FROM inboxes as i
        LEFT JOIN assignable_users as au ON au.inbox_id = i.id
      WHERE group_id = :group_id;
    ''');

    final select = result.root as SelectStatement;
    final columns = select.resolvedColumns!;

    expect(
        columns.map((e) => e.name), ['id', 'group_id', 'user_id', 'inbox_id']);
    expect(columns.map((e) => result.typeOf(e).nullable),
        [false, false, true, true]);
  });

  test('regression test for #1096', () {
    // https://github.com/simolus3/drift/issues/1096#issuecomment-931378474
    final engine = SqlEngine(
        EngineOptions(useDriftExtensions: true, version: SqliteVersion.v3_35))
      ..registerTableFromSql('''
CREATE TABLE downloads (
    id INT NOT NULL PRIMARY KEY AUTOINCREMENT,
    uri TEXT NOT NULL,
    last_modified_time INT,
    attempts_started_count INT NOT NULL DEFAULT 0,
    priority INT NOT NULL DEFAULT 0,
    last_attempt_timestamp INT,
    status TEXT NOT NULL,
    handed_off bool NOT NULL DEFAULT false,
    sync_session INT NOT NULL,

    UNIQUE (sync_session, uri)
);
      ''');

    final result = engine.analyze('''
    UPDATE downloads
    SET status = 'CHECKED_OUT'
    FROM (
        SELECT * FROM downloads
        ) as w
    WHERE downloads.id = w.id
    RETURNING downloads.id;
    ''');

    expect(result.errors, isEmpty);

    final select = result.root as UpdateStatement;
    final columns = select.returnedResultSet!.resolvedColumns!;

    expect(columns, hasLength(1));
    expect(
        result.typeOf(columns.single).type,
        isA<ResolvedType>()
            .having((e) => e.type, 'type', BasicType.int)
            .having((e) => e.nullable, 'nullable', isFalse));
  });

  test('regression test for #1858', () {
    // https://github.com/simolus3/drift/issues/1858
    final engine = SqlEngine(
        EngineOptions(useDriftExtensions: true, version: SqliteVersion.v3_38));

    engine.registerTableFromSql('''
CREATE TABLE IF NOT EXISTS contract_has_add_fees
(
   position_id INTEGER NOT NULL,
   position_db INTEGER NOT NULL,
   addfee      TEXT    NOT NULL,
   amount      REAL    NOT NULL,
   changed     DATETIME,
   removed     BOOLEAN NOT NULL DEFAULT FALSE,
   PRIMARY KEY (position_id, position_db, addfee)
) WITHOUT ROWID;
''');

    final result = engine.analyze(r'''
DELETE
FROM contract_has_add_fees
WHERE EXISTS(SELECT *
             FROM json_each(:json) AS j
             WHERE position_id = json_extract(j.value, '$.positionid')
               AND position_db = json_extract(j.value, '$.positiondb')
               AND addfee = json_extract(j.value, '$.addfee')
         );
''');

    expect(result.errors, isEmpty);
  });

  test('regression test for #2010', () {
    // https://github.com/simolus3/drift/issues/2010
    final engine = SqlEngine()
      ..registerTableFromSql('CREATE TABLE place_hierarchy (feature_id INT);')
      ..registerTableFromSql('CREATE TABLE place_name (feature_id INT);')
      ..registerTableFromSql('CREATE TABLE place (feature_id INT);');

    final result = engine.analyze('''
  SELECT
    (
      SELECT
        true
      FROM (
        SELECT
          'test' AS name
        FROM
          place_hierarchy ph
        where
          ph.feature_id = p.feature_id
      ) second_select
    ) first_select
  FROM place_name matches
  INNER JOIN place p ON p.feature_id = matches.feature_id;
''');

    expect(result.errors, isEmpty);
  });
}