1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74
   75
   76
   77
   78
   79
   80
   81
   82
   83
   84
   85
   86
   87
   88
   89
   90
   91
   92
   93
   94
   95
   96
   97
   98
   99


android_webview / browser / metrics / aw_metrics_service_client_sample_rate_unittest.cc [blame]

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

#include "android_webview/browser/metrics/aw_metrics_service_client.h"

#include "base/metrics/user_metrics.h"
#include "base/test/task_environment.h"
#include "base/test/test_simple_task_runner.h"
#include "base/time/time.h"
#include "components/embedder_support/android/metrics/android_metrics_service_client.h"
#include "components/metrics/content/subprocess_metrics_provider.h"
#include "components/metrics/metrics_pref_names.h"
#include "components/metrics/metrics_switches.h"
#include "components/prefs/testing_pref_service.h"
#include "content/public/test/browser_task_environment.h"
#include "content/public/test/test_content_client_initializer.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace android_webview {

using InstallerPackageType =
    metrics::AndroidMetricsServiceClient::InstallerPackageType;

namespace {
class AwMetricsServiceClientSampleRateTestDelegate
    : public AwMetricsServiceClient::Delegate {
  void RegisterAdditionalMetricsProviders(
      metrics::MetricsService* service) override {}
  void AddWebViewAppStateObserver(WebViewAppStateObserver* observer) override {}
  bool HasAwContentsEverCreated() const override { return false; }
};

class AwMetricsServiceTestClientForSampling : public AwMetricsServiceClient {
 public:
  explicit AwMetricsServiceTestClientForSampling(
      std::unique_ptr<Delegate> delegate)
      : AwMetricsServiceClient(std::move(delegate)) {}
  using AwMetricsServiceClient::IsInSample;
  using AwMetricsServiceClient::RegisterPrefs;
  void SetSampleRatePerMille(int sample_rate_per_mille) {
    _sample_rate_per_mille = sample_rate_per_mille;
  }

  int GetSampleRatePerMille() const override { return _sample_rate_per_mille; }

 private:
  int _sample_rate_per_mille;
};

class AwMetricsServiceClientSampleRateTest : public testing::Test {
 protected:
  AwMetricsServiceClientSampleRateTest()
      : task_runner_(new base::TestSimpleTaskRunner) {
    base::SetRecordActionTaskRunner(task_runner_);
    // Needed because RegisterMetricsProvidersAndInitState() checks for this.
    metrics::SubprocessMetricsProvider::CreateInstance();
  }

 private:
  // Needed since starting metrics reporting triggers code that uses content::
  // objects.
  content::TestContentClientInitializer test_content_initializer_;
  // Needed since starting metrics reporting triggers code that expects to be
  // running on the browser UI thread. Also needed for its FastForwardBy method.
  content::BrowserTaskEnvironment task_environment_{
      base::test::TaskEnvironment::TimeSource::MOCK_TIME};
  scoped_refptr<base::TestSimpleTaskRunner> task_runner_;
};

}  // namespace

TEST_F(AwMetricsServiceClientSampleRateTest, TestShouldSampleByClientUUID) {
  struct {
    const char* client_uuid;
    int sampling_rate_per_mille;
    bool expected_in_sample;
  } test_cases[] = {
      {"a7a68d68-8ba3-486d-832b-a0cded65fea2", 990 /*CANARY*/, false},
      {"fa5f5bd4-aae7-4d94-ab84-69c8ca40f400", 990 /*CANARY*/, true},
      {"fa5f5bd4-aae7-4d94-ab84-69c8ca40f400", 20 /*STABLE*/, false},
      {"747dbd7a-7b48-4496-8043-88edf84e0ab3", 20 /*STABLE*/, true}};

  for (const auto& test : test_cases) {
    auto prefs = std::make_unique<TestingPrefServiceSimple>();
    AwMetricsServiceTestClientForSampling::RegisterMetricsPrefs(
        prefs->registry());
    prefs->SetString(metrics::prefs::kMetricsClientID, test.client_uuid);
    auto client = std::make_unique<AwMetricsServiceTestClientForSampling>(
        std::make_unique<AwMetricsServiceClientSampleRateTestDelegate>());
    client->SetHaveMetricsConsent(/*user_consent=*/true, /*app_consent=*/true);
    client->Initialize(prefs.get());
    client->SetSampleRatePerMille(test.sampling_rate_per_mille);

    EXPECT_EQ(client->IsInSample(), test.expected_in_sample);
  }
}

}  // namespace android_webview