drift/extras/integration_tests/tests/lib/suite/migrations.dart

import 'package:test/test.dart';
import 'package:tests/data/sample_data.dart';
import 'package:tests/database/database.dart';

import 'suite.dart';

void migrationTests(TestExecutor executor) {
  test('creates users table when opening version 1', () async {
    final database = Database(executor.createExecutor(), schemaVersion: 1);

    // we write 3 users when the database is created
    final count = await database.userCountQuery().getSingle();
    expect(count, 3);

    await database.close();
  });

  test('saves and restores database', () async {
    var database = Database(executor.createExecutor(), schemaVersion: 1);
    await database.writeUser(People.florian);
    await database.close();

    database = Database(executor.createExecutor(), schemaVersion: 2);

    // the 3 initial users plus People.florian
    final count = await database.userCountQuery().getSingle();
    expect(count, 4);

    await database.close();
  });
}