tools/blobstats/test/blobstats_test.dart

// Copyright 2020 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

// TODO(https://fxbug.dev/84961): Fix null safety and remove this language version.
// @dart=2.9

import 'dart:io';

import 'package:blobstats/blob.dart';
import 'package:blobstats/blobstats.dart';
import 'package:blobstats/package.dart';
import 'package:test/test.dart';
import 'package:path/path.dart' as p;

void main() {
  group('blobstats tests', () {
    test('generate valid blobs.csv', () async {
      var tmpDir = Directory.systemTemp.createTempSync();
      var stats = BlobStats(tmpDir, tmpDir, '');

      var blob1 = Blob()
        ..hash = '123abc'
        ..sourcePath = 'entry1'
        ..size = 123
        ..count = 1;
      var blob2 = Blob()
        ..hash = 'zyx456'
        ..sourcePath = 'entry2'
        ..size = 16
        ..count = 4;
      var blob3 = Blob()
        ..hash = '1x1x1x'
        ..sourcePath = '/path/file'
        ..size = 256
        ..count = 2;

      stats.blobsByHash[blob1.hash] = blob1;
      stats.blobsByHash[blob2.hash] = blob2;
      stats.blobsByHash[blob3.hash] = blob3;

      var blobsCsvPath = await stats.csvBlobs();
      List<String> lines = File(blobsCsvPath).readAsLinesSync();

      expect(lines, hasLength(4));
      expect(lines[0], equals('Size,Share,Prop,Saved,Path'));
      expect(lines[1], equals('256,2,128,256,/path/file'));
      expect(lines[2], equals('123,1,123,0,entry1'));
      expect(lines[3], equals('16,4,4,48,entry2'));
    });

    test('generate valid packages.csv', () async {
      var tmpDir = Directory.systemTemp.createTempSync();
      var stats = BlobStats(tmpDir, tmpDir, '');

      var package1 = Package()
        ..name = 'package1'
        ..proportional = 3
        ..size = 123
        ..private = 1;
      var package2 = Package()
        ..name = 'another_package'
        ..proportional = 1
        ..size = 16
        ..private = 4;
      var package3 = Package()
        ..name = 'ABC'
        ..proportional = 12
        ..size = 256
        ..private = 2;

      stats.packages.add(package1);
      stats.packages.add(package2);
      stats.packages.add(package3);

      var packagesCsvPath = await stats.csvPackages();
      List<String> lines = File(packagesCsvPath).readAsLinesSync();

      expect(lines, hasLength(4));
      expect(lines[0], equals('Size,Prop,Private,Name'));
      expect(lines[1], equals('256,12,2,ABC'));
      expect(lines[2], equals('123,3,1,package1'));
      expect(lines[3], equals('16,1,4,another_package'));
    });
    test('correctly parse image_assembly.json', () async {
      var tmpDir = Directory.systemTemp.createTempSync();
      var imageAssemblyFilePath = p.join(tmpDir.path, 'abc.json');
      await File(imageAssemblyFilePath).writeAsString('''
{
  "system": [
    "path/to/foo"
  ],
  "ignored": [
    "xzwv"
  ],
  "base": [
    "obj/build/test",
    "a/b/c"
  ],
  "cache": [
    "qwerty"
  ]
}
''');
      var stats = BlobStats(tmpDir, tmpDir, '');
      await stats.addImageAssembly(imageAssemblyFilePath);
      stats.pendingPackages.sort();
      // All items under "system", "base", and "cache" should be included.
      // Blobstats also automatically adds this entry: obj/build/images/fuchsia/fuchsia/base/package_manifest.json
      expect(stats.pendingPackages.length, 5);
      expect(stats.pendingPackages[0], 'a/b/c');
      expect(stats.pendingPackages[1],
          'obj/build/images/fuchsia/fuchsia/base/package_manifest.json');
      expect(stats.pendingPackages[2], 'obj/build/test');
      expect(stats.pendingPackages[3], 'path/to/foo');
      expect(stats.pendingPackages[4], 'qwerty');
    });
  });
}