OLD | NEW |
1 // Copyright 2017 The Chromium Authors. All rights reserved. | 1 // Copyright 2017 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef COMPONENTS_UKM_TEST_UKM_SERVICE_H_ | 5 #ifndef COMPONENTS_UKM_TEST_UKM_SERVICE_H_ |
6 #define COMPONENTS_UKM_TEST_UKM_SERVICE_H_ | 6 #define COMPONENTS_UKM_TEST_UKM_SERVICE_H_ |
7 | 7 |
8 #include <stddef.h> | 8 #include <stddef.h> |
9 #include <memory> | 9 #include <memory> |
10 | 10 |
11 #include "components/metrics/test_metrics_service_client.h" | 11 #include "components/metrics/test_metrics_service_client.h" |
12 #include "components/prefs/testing_pref_service.h" | 12 #include "components/prefs/testing_pref_service.h" |
13 #include "components/ukm/ukm_pref_names.h" | 13 #include "components/ukm/ukm_pref_names.h" |
14 #include "components/ukm/ukm_service.h" | 14 #include "components/ukm/ukm_service.h" |
15 | 15 |
16 namespace ukm { | 16 namespace ukm { |
17 | 17 |
18 // Wraps an UkmService with additional accessors used for testing. | 18 // Wraps an UkmService with additional accessors used for testing. |
19 class TestUkmService : public UkmService { | 19 class TestUkmService : public UkmService { |
20 public: | 20 public: |
21 explicit TestUkmService(PrefService* pref_service); | 21 explicit TestUkmService(PrefService* pref_service); |
22 ~TestUkmService() override; | 22 ~TestUkmService() override; |
23 | 23 |
24 size_t sources_count() const { return sources_for_testing().size(); } | 24 size_t sources_count() const { return sources_for_testing().size(); } |
25 const UkmSource* GetSource(size_t source_num) const; | 25 const UkmSource* GetSource(size_t source_num) const; |
26 | 26 |
| 27 size_t entries_count() const { return entries_for_testing().size(); } |
| 28 const UkmEntry* GetEntry(size_t entry_num) const; |
| 29 |
27 private: | 30 private: |
28 metrics::TestMetricsServiceClient test_metrics_service_client_; | 31 metrics::TestMetricsServiceClient test_metrics_service_client_; |
29 | 32 |
30 DISALLOW_COPY_AND_ASSIGN(TestUkmService); | 33 DISALLOW_COPY_AND_ASSIGN(TestUkmService); |
31 }; | 34 }; |
32 | 35 |
33 // Convenience harness used for testing; creates a TestUkmService and | 36 // Convenience harness used for testing; creates a TestUkmService and |
34 // supplies it with a prefs service with a longer lifespan. | 37 // supplies it with a prefs service with a longer lifespan. |
35 class UkmServiceTestingHarness { | 38 class UkmServiceTestingHarness { |
36 public: | 39 public: |
37 UkmServiceTestingHarness(); | 40 UkmServiceTestingHarness(); |
38 virtual ~UkmServiceTestingHarness(); | 41 virtual ~UkmServiceTestingHarness(); |
39 | 42 |
40 TestUkmService* test_ukm_service() { return test_ukm_service_.get(); } | 43 TestUkmService* test_ukm_service() { return test_ukm_service_.get(); } |
41 | 44 |
42 private: | 45 private: |
43 TestingPrefServiceSimple test_prefs_; | 46 TestingPrefServiceSimple test_prefs_; |
44 std::unique_ptr<TestUkmService> test_ukm_service_; | 47 std::unique_ptr<TestUkmService> test_ukm_service_; |
45 | 48 |
46 DISALLOW_COPY_AND_ASSIGN(UkmServiceTestingHarness); | 49 DISALLOW_COPY_AND_ASSIGN(UkmServiceTestingHarness); |
47 }; | 50 }; |
48 | 51 |
49 } // namespace ukm | 52 } // namespace ukm |
50 | 53 |
51 #endif // COMPONENTS_UKM_TEST_UKM_SERVICE_H_ | 54 #endif // COMPONENTS_UKM_TEST_UKM_SERVICE_H_ |
OLD | NEW |