2023鐵人賽_D8_5

my_database.dart

import 'package:drift/drift.dart';

part 'my_database.g.dart';

class Products extends Table {
  IntColumn get id => integer().autoIncrement()();
  TextColumn get type => text()();
}

@DriftDatabase(tables: [Products])
class MyDatabase extends _$MyDatabase {
  MyDatabase(QueryExecutor e) : super(e);

  @override
  int get schemaVersion => 1;

  Future<int> createProduct(String type) {
    return into(products).insert(ProductsCompanion.insert(type: type));
  }

  Future<List<Product>> getAll() async {
    return await select(products).get();
  }
}

my_favorites.dart

import 'my_database.dart';

class MyFavorites {
  final MyDatabase database;

  MyFavorites(this.database);

  Future<void> add(Product product) async {
    await database.createProduct(product.type);
  }

  Future<List<Product>> getAll() async {
    return await database.getAll();
  }
}

my_favorites_test.dart

import 'package:drift/native.dart';
import 'package:flutter_test/flutter_test.dart';
import 'my_database.dart';
import 'my_favorites.dart';

main() {
  test("add favorite", () async {
    MyDatabase database = MyDatabase(NativeDatabase.memory());

    var myFavorites = MyFavorites(database);

    await myFavorites.add(const Product(id: 1, type: "book"));

    expect(await myFavorites.getAll(), [const Product(id: 1, type: "book")]);

    await database.close();
  });
}