mirror of https://github.com/AMT-Cheif/drift.git
62 lines
2.0 KiB
Dart
62 lines
2.0 KiB
Dart
import 'package:test/test.dart';
|
|
import 'package:tests/data/sample_data.dart' as people;
|
|
import 'package:tests/database/database.dart';
|
|
import 'package:tests/tests.dart';
|
|
|
|
import 'suite.dart';
|
|
|
|
void migrationTests(TestExecutor executor) {
|
|
test('creates users table when opening version 1', () async {
|
|
final database = Database(executor.createConnection(), schemaVersion: 1);
|
|
|
|
// we write 3 users when the database is created
|
|
final count = await database.userCount().getSingle();
|
|
expect(count, 3);
|
|
|
|
await database.close();
|
|
});
|
|
|
|
test('saves and restores database', () async {
|
|
var database = Database(executor.createConnection(), schemaVersion: 1);
|
|
await database.writeUser(people.florian);
|
|
await database.close();
|
|
|
|
database = Database(executor.createConnection(), schemaVersion: 2);
|
|
|
|
// the 3 initial users plus People.florian
|
|
final count = await database.userCount().getSingle();
|
|
expect(count, 4);
|
|
expect(database.schemaVersionChangedFrom, 1);
|
|
expect(database.schemaVersionChangedTo, 2);
|
|
|
|
await database.close();
|
|
});
|
|
|
|
test('can use destructive migration', () async {
|
|
final old = Database(executor.createConnection(), schemaVersion: 1);
|
|
await old.executor.ensureOpen(old);
|
|
await old.close();
|
|
|
|
final database = Database(executor.createConnection(), schemaVersion: 2);
|
|
database.overrideMigration = database.destructiveFallback;
|
|
|
|
// No users now, we deleted everything
|
|
final count = await database.userCount().getSingle();
|
|
expect(count, 0);
|
|
});
|
|
|
|
test('runs the migrator when downgrading', () async {
|
|
var database = Database(executor.createConnection(), schemaVersion: 2);
|
|
await database.executor.ensureOpen(database); // Create the database
|
|
await database.close();
|
|
|
|
database = Database(executor.createConnection(), schemaVersion: 1);
|
|
await database.executor.ensureOpen(database); // Let the migrator run
|
|
|
|
expect(database.schemaVersionChangedFrom, 2);
|
|
expect(database.schemaVersionChangedTo, 1);
|
|
|
|
await database.close();
|
|
});
|
|
}
|