perf(import): batch playlist metadata enrichment

2026-09-02 16:20:57 +02:00 · 2026-08-30 23:18:38 +07:00
parent aecba736e1
commit bd07d5c233
2 changed files with 249 additions and 90 deletions
@@ -0,0 +1,103 @@
+import 'package:flutter_test/flutter_test.dart';
+import 'package:spotiflac_android/models/track.dart';
+import 'package:spotiflac_android/services/csv_import_service.dart';
+
+void main() {
+  test(
+    'playlist enrichment uses generic lookup with bounded concurrency',
+    () async {
+      var active = 0;
+      var maxActive = 0;
+      final progress = <int>[];
+      final tracks = List.generate(
+        5,
+        (index) => Track(
+          id: 'original-$index',
+          name: 'Song $index',
+          artistName: 'Artist',
+          albumName: '',
+          isrc: 'TEST0000000$index',
+          duration: 0,
+        ),
+      );
+
+      final enriched = await CsvImportService.enrichTracksMetadata(
+        tracks,
+        concurrency: 2,
+        onProgress: (current, _) => progress.add(current),
+        lookup: (query, limit) async {
+          active++;
+          if (active > maxActive) maxActive = active;
+          await Future<void>.delayed(const Duration(milliseconds: 10));
+          active--;
+          final index = int.parse(query.substring(query.length - 1));
+          return [
+            {
+              'id': 'provider-$index',
+              'provider_id': 'generic-provider',
+              'name': 'Song $index',
+              'artists': 'Artist',
+              'album_name': 'Album',
+              'cover_url': 'https://example.test/$index.jpg',
+              'duration_ms': 180000,
+              'isrc': query,
+            },
+          ];
+        },
+      );
+
+      expect(maxActive, 2);
+      expect(progress, [1, 2, 3, 4, 5]);
+      expect(
+        enriched.map((track) => track.id),
+        tracks.map((track) => track.id),
+      );
+      expect(enriched.every((track) => track.duration == 180), isTrue);
+      expect(enriched[3].coverUrl, 'https://example.test/3.jpg');
+    },
+  );
+
+  test('ISRC enrichment rejects a provider mismatch', () async {
+    const track = Track(
+      id: 'original',
+      name: 'Wanted Song',
+      artistName: 'Wanted Artist',
+      albumName: '',
+      isrc: 'AAABC1200001',
+      duration: 0,
+    );
+    var calls = 0;
+
+    final enriched = await CsvImportService.enrichTracksMetadata(
+      const [track],
+      lookup: (query, limit) async {
+        calls++;
+        if (calls == 1) {
+          return [
+            {
+              'id': 'wrong',
+              'name': 'Other Song',
+              'artists': 'Other Artist',
+              'isrc': 'ZZZZZ9900000',
+              'duration_ms': 100000,
+            },
+          ];
+        }
+        return [
+          {
+            'id': 'right',
+            'name': 'Wanted Song',
+            'artists': 'Wanted Artist',
+            'cover_url': 'https://example.test/right.jpg',
+            'duration_ms': 210000,
+          },
+        ];
+      },
+    );
+
+    expect(calls, 2);
+    expect(enriched.single.id, 'original');
+    expect(enriched.single.duration, 210);
+    expect(enriched.single.coverUrl, 'https://example.test/right.jpg');
+  });
+}