| // Copyright 2015 The Chromium Authors |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef COMPONENTS_METRICS_TEST_TEST_METRICS_PROVIDER_H_ |
| #define COMPONENTS_METRICS_TEST_TEST_METRICS_PROVIDER_H_ |
| |
| #include "base/metrics/histogram_snapshot_manager.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() = default; |
| |
| TestMetricsProvider(const TestMetricsProvider&) = delete; |
| TestMetricsProvider& operator=(const TestMetricsProvider&) = delete; |
| |
| // MetricsProvider: |
| void Init() override; |
| void OnRecordingDisabled() override; |
| bool HasPreviousSessionData() override; |
| void ProvidePreviousSessionData( |
| ChromeUserMetricsExtension* uma_proto) override; |
| void ProvideCurrentSessionData( |
| ChromeUserMetricsExtension* uma_proto) override; |
| void ProvideSystemProfileMetrics( |
| SystemProfileProto* system_profile_proto) override; |
| void RecordInitialHistogramSnapshots( |
| base::HistogramSnapshotManager* snapshot_manager) override; |
| void RecordHistogramSnapshots( |
| base::HistogramSnapshotManager* snapshot_manager) override; |
| |
| bool init_called() { return init_called_; } |
| bool on_recording_disabled_called() { return on_recording_disabled_called_; } |
| bool has_initial_stability_metrics_called() { |
| return has_initial_stability_metrics_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_; |
| } |
| bool provide_system_profile_metrics_called() const { |
| return provide_system_profile_metrics_called_; |
| } |
| bool record_initial_histogram_snapshots_called() const { |
| return record_initial_histogram_snapshots_called_; |
| } |
| bool record_histogram_snapshots_called() const { |
| return record_histogram_snapshots_called_; |
| } |
| void set_record_histogram_snapshots_called(bool val) { |
| record_histogram_snapshots_called_ = val; |
| } |
| |
| private: |
| bool init_called_ = false; |
| bool on_recording_disabled_called_ = false; |
| bool has_initial_stability_metrics_ = false; |
| bool has_initial_stability_metrics_called_ = false; |
| bool provide_initial_stability_metrics_called_ = false; |
| bool provide_stability_metrics_called_ = false; |
| bool provide_system_profile_metrics_called_ = false; |
| bool record_initial_histogram_snapshots_called_ = false; |
| bool record_histogram_snapshots_called_ = false; |
| }; |
| |
| } // namespace metrics |
| |
| #endif // COMPONENTS_METRICS_TEST_TEST_METRICS_PROVIDER_H_ |