summaryrefslogtreecommitdiffstats
path: root/components/metrics/test_metrics_provider.h
blob: 4d34b2438d45bb4b488852b0c235634ad5af4ee4 (plain)
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
// Copyright 2015 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.

#ifndef COMPONENTS_METRICS_TEST_METRICS_PROVIDER_H_
#define COMPONENTS_METRICS_TEST_METRICS_PROVIDER_H_

#include "base/macros.h"
#include "components/metrics/metrics_provider.h"

namespace metrics {

// A simple implementation of MetricsProvider that checks that its providing
// functions are called, for use in tests.
class TestMetricsProvider : public MetricsProvider {
 public:
  TestMetricsProvider()
      : on_recording_disabled_called_(false),
        has_initial_stability_metrics_(false),
        provide_initial_stability_metrics_called_(false),
        provide_stability_metrics_called_(false) {}

  // MetricsProvider:
  void OnRecordingDisabled() override;
  bool HasInitialStabilityMetrics() override;
  void ProvideInitialStabilityMetrics(
      SystemProfileProto* system_profile_proto) override;
  void ProvideStabilityMetrics(
      SystemProfileProto* system_profile_proto) override;

  bool on_recording_disabled_called() { return on_recording_disabled_called_; }
  void set_has_initial_stability_metrics(bool has_initial_stability_metrics) {
    has_initial_stability_metrics_ = has_initial_stability_metrics;
  }
  bool provide_initial_stability_metrics_called() const {
    return provide_initial_stability_metrics_called_;
  }
  bool provide_stability_metrics_called() const {
    return provide_stability_metrics_called_;
  }

 private:
  bool on_recording_disabled_called_;
  bool has_initial_stability_metrics_;
  bool provide_initial_stability_metrics_called_;
  bool provide_stability_metrics_called_;

  DISALLOW_COPY_AND_ASSIGN(TestMetricsProvider);
};

}  // namespace metrics

#endif  // COMPONENTS_METRICS_TEST_METRICS_PROVIDER_H_