components/safe_browsing/content/browser/download/download_stats_unittest.cc

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

#include "components/safe_browsing/content/browser/download/download_stats.h"

#include "base/files/file_path.h"
#include "base/test/metrics/histogram_tester.h"
#include "components/download/public/common/download_stats.h"
#include "components/safe_browsing/content/common/file_type_policies.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/abseil-cpp/absl/types/optional.h"

namespace {

// The below constants are based on download_file_types.asciipb.
const int kExeFileTypeUmaValue = 0;
const int kApkFileTypeUmaValue = 20;

}  // namespace

namespace safe_browsing {

TEST(SafeBrowsingDownloadStatsTest, RecordDangerousDownloadWarningShown) {
  base::HistogramTester histogram_tester;

  RecordDangerousDownloadWarningShown(
      download::DownloadDangerType::DOWNLOAD_DANGER_TYPE_DANGEROUS_CONTENT,
      base::FilePath(FILE_PATH_LITERAL("file.apk")),
      /*is_https=*/true, /*has_user_gesture=*/true);
  histogram_tester.ExpectUniqueSample(
      "SBClientDownload.Warning.FileType.Malicious.Shown",
      /*sample=*/kApkFileTypeUmaValue, /*expected_bucket_count=*/1);
  histogram_tester.ExpectUniqueSample(
      "SBClientDownload.Warning.DownloadIsHttps.Malicious.Shown",
      /*sample=*/1, /*expected_bucket_count=*/1);
  histogram_tester.ExpectUniqueSample(
      "SBClientDownload.Warning.DownloadHasUserGesture.Malicious.Shown",
      /*sample=*/1, /*expected_bucket_count=*/1);

  RecordDangerousDownloadWarningShown(
      download::DownloadDangerType::DOWNLOAD_DANGER_TYPE_UNCOMMON_CONTENT,
      base::FilePath(FILE_PATH_LITERAL("file.exe")),
      /*is_https=*/false, /*has_user_gesture=*/false);
  histogram_tester.ExpectBucketCount(
      "SBClientDownload.Warning.FileType.Uncommon.Shown",
      /*sample=*/kExeFileTypeUmaValue,
      /*expected_count=*/1);
  histogram_tester.ExpectBucketCount(
      "SBClientDownload.Warning.DownloadIsHttps.Uncommon.Shown",
      /*sample=*/0,
      /*expected_count=*/1);
  histogram_tester.ExpectBucketCount(
      "SBClientDownload.Warning.DownloadHasUserGesture.Uncommon.Shown",
      /*sample=*/0,
      /*expected_count=*/1);
}

TEST(SafeBrowsingDownloadStatsTest, RecordDangerousDownloadWarningBypassed) {
  base::HistogramTester histogram_tester;

  RecordDangerousDownloadWarningBypassed(
      download::DownloadDangerType::DOWNLOAD_DANGER_TYPE_DANGEROUS_FILE,
      base::FilePath(FILE_PATH_LITERAL("file.apk")),
      /*is_https=*/true, /*has_user_gesture=*/false);
  histogram_tester.ExpectUniqueSample(
      "SBClientDownload.Warning.FileType.DangerousFileType.Bypassed",
      /*sample=*/kApkFileTypeUmaValue, /*expected_bucket_count=*/1);
  histogram_tester.ExpectUniqueSample(
      "SBClientDownload.Warning.DownloadIsHttps.DangerousFileType.Bypassed",
      /*sample=*/1, /*expected_bucket_count=*/1);
  histogram_tester.ExpectUniqueSample(
      "SBClientDownload.Warning.DownloadHasUserGesture.DangerousFileType."
      "Bypassed",
      /*sample=*/0, /*expected_bucket_count=*/1);
}

TEST(SafeBrowsingDownloadStatsTest, RecordDownloadOpened) {
  base::HistogramTester histogram_tester;

  base::Time download_end_time = base::Time::Now();
  download::DownloadContent fake_content =
      download::DownloadContent::SPREADSHEET;
  // Not logged for dangerous downloads.
  RecordDownloadOpened(
      download::DownloadDangerType::DOWNLOAD_DANGER_TYPE_DANGEROUS_CONTENT,
      fake_content, download_end_time + base::Days(1), download_end_time,
      /*show_download_in_folder=*/false);
  histogram_tester.ExpectTotalCount(
      "SBClientDownload.SafeDownloadOpenedLatency2.OpenDirectly", 0);

  RecordDownloadOpened(
      download::DownloadDangerType::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS,
      fake_content, download_end_time + base::Days(1), download_end_time,
      /*show_download_in_folder=*/false);
  histogram_tester.ExpectTimeBucketCount(
      "SBClientDownload.SafeDownloadOpenedLatency2.OpenDirectly",
      /*sample=*/base::Days(1),
      /*count=*/1);

  RecordDownloadOpened(
      download::DownloadDangerType::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS,
      fake_content, download_end_time + base::Hours(5), download_end_time,
      /*show_download_in_folder=*/true);
  histogram_tester.ExpectTimeBucketCount(
      "SBClientDownload.SafeDownloadOpenedLatency2.ShowInFolder",
      /*sample=*/base::Hours(5),
      /*count=*/1);
}

TEST(SafeBrowsingDownloadStatsTest, RecordDownloadOpenedFileType) {
  base::HistogramTester histogram_tester;

  base::Time download_end_time = base::Time::Now();

  RecordDownloadOpenedFileType(download::DownloadContent::SPREADSHEET,
                               download_end_time + base::Days(1),
                               download_end_time);
  histogram_tester.ExpectTimeBucketCount(
      "SBClientDownload.SafeDownloadOpenedLatencyByContentType.SPREADSHEET",
      /*sample=*/base::Days(1),
      /*count=*/1);

  RecordDownloadOpenedFileType(download::DownloadContent::PRESENTATION,
                               download_end_time + base::Hours(5),
                               download_end_time);
  histogram_tester.ExpectTimeBucketCount(
      "SBClientDownload.SafeDownloadOpenedLatencyByContentType.PRESENTATION",
      /*sample=*/base::Hours(5),
      /*count=*/1);

  RecordDownloadOpenedFileType(download::DownloadContent::ARCHIVE,
                               download_end_time + base::Days(1),
                               download_end_time);
  histogram_tester.ExpectTimeBucketCount(
      "SBClientDownload.SafeDownloadOpenedLatencyByContentType.ARCHIVE",
      /*sample=*/base::Days(1),
      /*count=*/1);
}

TEST(SafeBrowsingDownloadStatsTest, RecordDownloadFileTypeAttributes) {
  {
    base::HistogramTester histogram_tester;
    RecordDownloadFileTypeAttributes(DownloadFileType::ALLOW_ON_USER_GESTURE,
                                     /*has_user_gesture=*/false,
                                     /*visited_referrer_before=*/false,
                                     /*latest_bypass_time=*/absl::nullopt);
    histogram_tester.ExpectUniqueSample(
        "SBClientDownload.UserGestureFileType.Attributes",
        /*sample=*/UserGestureFileTypeAttributes::TOTAL_TYPE_CHECKED,
        /*expected_bucket_count=*/1);
    histogram_tester.ExpectTotalCount(
        "SBClientDownload.UserGestureFileType.LastBypassDownloadInterval",
        /*count=*/0);
  }
  {
    base::HistogramTester histogram_tester;
    RecordDownloadFileTypeAttributes(
        DownloadFileType::ALLOW_ON_USER_GESTURE,
        /*has_user_gesture=*/true,
        /*visited_referrer_before=*/true,
        /*latest_bypass_time=*/base::Time::Now() - base::Hours(1));
    histogram_tester.ExpectBucketCount(
        "SBClientDownload.UserGestureFileType.Attributes",
        /*sample=*/UserGestureFileTypeAttributes::TOTAL_TYPE_CHECKED,
        /*expected_count=*/1);
    histogram_tester.ExpectBucketCount(
        "SBClientDownload.UserGestureFileType.Attributes",
        /*sample=*/UserGestureFileTypeAttributes::HAS_USER_GESTURE,
        /*expected_count=*/1);
    histogram_tester.ExpectBucketCount(
        "SBClientDownload.UserGestureFileType.Attributes",
        /*sample=*/UserGestureFileTypeAttributes::HAS_REFERRER_VISIT,
        /*expected_count=*/1);
    histogram_tester.ExpectBucketCount(
        "SBClientDownload.UserGestureFileType.Attributes",
        /*sample=*/
        UserGestureFileTypeAttributes::HAS_BOTH_USER_GESTURE_AND_REFERRER_VISIT,
        /*expected_count=*/1);
    histogram_tester.ExpectUniqueTimeSample(
        "SBClientDownload.UserGestureFileType.LastBypassDownloadInterval",
        /*sample=*/base::Hours(1),
        /*expected_bucket_count=*/1);
  }
}

}  // namespace safe_browsing