twonly-app/test/drift/stickers_dao_test.dart

import 'dart:typed_data';

import 'package:crypto/crypto.dart';
import 'package:drift/native.dart';
import 'package:flutter_test/flutter_test.dart';
import 'package:twonly/src/database/twonly.db.dart';

void main() {
  late TwonlyDB database;

  setUp(() => database = TwonlyDB(NativeDatabase.memory()));
  tearDown(() => database.close());

  test('identical stickers are deduplicated without resetting usage', () async {
    final bytes = Uint8List.fromList([1, 2, 3]);
    final hash = sha256.convert(bytes).toString();
    expect(
      await database.stickersDao.save(webp: bytes, width: 30, height: 20),
      isTrue,
    );
    await database.stickersDao.recordUse(hash);
    expect(
      await database.stickersDao.save(webp: bytes, width: 30, height: 20),
      isFalse,
    );
    final stickers = await database.stickersDao.watchAll().first;
    expect(stickers, hasLength(1));
    expect(stickers.single.usageCount, 1);
  });

  test(
    'most used stickers come first and removal only removes the selected one',
    () async {
      final first = Uint8List.fromList([1]);
      final second = Uint8List.fromList([2]);
      final firstHash = sha256.convert(first).toString();
      final secondHash = sha256.convert(second).toString();
      await database.stickersDao.save(webp: first, width: 30, height: 20);
      await database.stickersDao.save(webp: second, width: 30, height: 20);
      await database.stickersDao.recordUse(secondHash);
      await database.stickersDao.recordUse(firstHash);
      await database.stickersDao.recordUse(firstHash);
      final stickers = await database.stickersDao.watchAll().first;
      expect(stickers.map((sticker) => sticker.contentHash), [
        firstHash,
        secondHash,
      ]);
      expect(stickers.map((sticker) => sticker.usageCount), [2, 1]);
      await database.stickersDao.remove(firstHash);
      expect(await database.stickersDao.getByHash(firstHash), isNull);
      expect((await database.stickersDao.getByHash(secondHash))?.usageCount, 1);
    },
  );

  test(
    'false digests and invalid bounds cannot enter the collection',
    () async {
      final bytes = Uint8List.fromList([1]);
      await expectLater(
        database.stickersDao.save(
          webp: bytes,
          width: 30,
          height: 20,
          expectedHash: 'incorrect',
        ),
        throwsArgumentError,
      );
      await expectLater(
        database.stickersDao.save(webp: bytes, width: 301, height: 20),
        throwsArgumentError,
      );
      expect(await database.stickersDao.watchAll().first, isEmpty);
    },
  );
}