76 lines
3.0 KiB
Dart
76 lines
3.0 KiB
Dart
// dart format width=80
|
|
// ignore_for_file: unused_local_variable, unused_import
|
|
import 'package:drift/drift.dart';
|
|
import 'package:drift_dev/api/migrations_native.dart';
|
|
import 'package:surface/database/database.dart';
|
|
import 'package:flutter_test/flutter_test.dart';
|
|
import 'generated/schema.dart';
|
|
|
|
import 'generated/schema_v1.dart' as v1;
|
|
import 'generated/schema_v2.dart' as v2;
|
|
|
|
void main() {
|
|
driftRuntimeOptions.dontWarnAboutMultipleDatabases = true;
|
|
late SchemaVerifier verifier;
|
|
|
|
setUpAll(() {
|
|
verifier = SchemaVerifier(GeneratedHelper());
|
|
});
|
|
|
|
group('simple database migrations', () {
|
|
// These simple tests verify all possible schema updates with a simple (no
|
|
// data) migration. This is a quick way to ensure that written database
|
|
// migrations properly alter the schema.
|
|
const versions = GeneratedHelper.versions;
|
|
for (final (i, fromVersion) in versions.indexed) {
|
|
group('from $fromVersion', () {
|
|
for (final toVersion in versions.skip(i + 1)) {
|
|
test('to $toVersion', () async {
|
|
final schema = await verifier.schemaAt(fromVersion);
|
|
final db = AppDatabase(schema.newConnection());
|
|
await verifier.migrateAndValidate(db, toVersion);
|
|
await db.close();
|
|
});
|
|
}
|
|
});
|
|
}
|
|
});
|
|
|
|
// The following template shows how to write tests ensuring your migrations
|
|
// preserve existing data.
|
|
// Testing this can be useful for migrations that change existing columns
|
|
// (e.g. by alterating their type or constraints). Migrations that only add
|
|
// tables or columns typically don't need these advanced tests. For more
|
|
// information, see https://drift.simonbinder.eu/migrations/tests/#verifying-data-integrity
|
|
// TODO: This generated template shows how these tests could be written. Adopt
|
|
// it to your own needs when testing migrations with data integrity.
|
|
test('migration from v1 to v2 does not corrupt data', () async {
|
|
// Add data to insert into the old database, and the expected rows after the
|
|
// migration.
|
|
// TODO: Fill these lists
|
|
final oldSnLocalChatChannelData = <v1.SnLocalChatChannelData>[];
|
|
final expectedNewSnLocalChatChannelData = <v2.SnLocalChatChannelData>[];
|
|
|
|
final oldSnLocalChatMessageData = <v1.SnLocalChatMessageData>[];
|
|
final expectedNewSnLocalChatMessageData = <v2.SnLocalChatMessageData>[];
|
|
|
|
await verifier.testWithDataIntegrity(
|
|
oldVersion: 1,
|
|
newVersion: 2,
|
|
createOld: v1.DatabaseAtV1.new,
|
|
createNew: v2.DatabaseAtV2.new,
|
|
openTestedDatabase: AppDatabase.new,
|
|
createItems: (batch, oldDb) {
|
|
batch.insertAll(oldDb.snLocalChatChannel, oldSnLocalChatChannelData);
|
|
batch.insertAll(oldDb.snLocalChatMessage, oldSnLocalChatMessageData);
|
|
},
|
|
validateItems: (newDb) async {
|
|
expect(expectedNewSnLocalChatChannelData,
|
|
await newDb.select(newDb.snLocalChatChannel).get());
|
|
expect(expectedNewSnLocalChatMessageData,
|
|
await newDb.select(newDb.snLocalChatMessage).get());
|
|
},
|
|
);
|
|
});
|
|
}
|