| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 CHROMEOS_SYSTEM_FAKE_STATISTICS_PROVIDER_H_ | 5 #ifndef CHROMEOS_SYSTEM_FAKE_STATISTICS_PROVIDER_H_ |
| 6 #define CHROMEOS_SYSTEM_FAKE_STATISTICS_PROVIDER_H_ | 6 #define CHROMEOS_SYSTEM_FAKE_STATISTICS_PROVIDER_H_ |
| 7 | 7 |
| 8 #include <map> | 8 #include <map> |
| 9 #include <string> | 9 #include <string> |
| 10 | 10 |
| 11 #include "chromeos/system/statistics_provider.h" | 11 #include "chromeos/system/statistics_provider.h" |
| 12 | 12 |
| 13 namespace chromeos { | 13 namespace chromeos { |
| 14 namespace system { | 14 namespace system { |
| 15 | 15 |
| 16 // A fake StatisticsProvider implementation that is useful in tests. | 16 // A fake StatisticsProvider implementation that is useful in tests. |
| 17 class FakeStatisticsProvider : public StatisticsProvider { | 17 class FakeStatisticsProvider : public StatisticsProvider { |
| 18 public: | 18 public: |
| 19 FakeStatisticsProvider(); | 19 FakeStatisticsProvider(); |
| 20 virtual ~FakeStatisticsProvider(); | 20 ~FakeStatisticsProvider() override; |
| 21 | 21 |
| 22 // StatisticsProvider implementation: | 22 // StatisticsProvider implementation: |
| 23 virtual void StartLoadingMachineStatistics( | 23 void StartLoadingMachineStatistics( |
| 24 const scoped_refptr<base::TaskRunner>& file_task_runner, | 24 const scoped_refptr<base::TaskRunner>& file_task_runner, |
| 25 bool load_oem_manifest) override; | 25 bool load_oem_manifest) override; |
| 26 virtual bool GetMachineStatistic(const std::string& name, | 26 bool GetMachineStatistic(const std::string& name, |
| 27 std::string* result) override; | 27 std::string* result) override; |
| 28 virtual bool HasMachineStatistic(const std::string& name) override; | 28 bool HasMachineStatistic(const std::string& name) override; |
| 29 virtual bool GetMachineFlag(const std::string& name, bool* result) override; | 29 bool GetMachineFlag(const std::string& name, bool* result) override; |
| 30 virtual bool HasMachineFlag(const std::string& name) override; | 30 bool HasMachineFlag(const std::string& name) override; |
| 31 virtual void Shutdown() override; | 31 void Shutdown() override; |
| 32 | 32 |
| 33 void SetMachineStatistic(const std::string& key, const std::string& value); | 33 void SetMachineStatistic(const std::string& key, const std::string& value); |
| 34 void ClearMachineStatistic(const std::string& key); | 34 void ClearMachineStatistic(const std::string& key); |
| 35 void SetMachineFlag(const std::string& key, bool value); | 35 void SetMachineFlag(const std::string& key, bool value); |
| 36 void ClearMachineFlag(const std::string& key); | 36 void ClearMachineFlag(const std::string& key); |
| 37 | 37 |
| 38 private: | 38 private: |
| 39 std::map<std::string, std::string> machine_statistics_; | 39 std::map<std::string, std::string> machine_statistics_; |
| 40 std::map<std::string, bool> machine_flags_; | 40 std::map<std::string, bool> machine_flags_; |
| 41 | 41 |
| 42 DISALLOW_COPY_AND_ASSIGN(FakeStatisticsProvider); | 42 DISALLOW_COPY_AND_ASSIGN(FakeStatisticsProvider); |
| 43 }; | 43 }; |
| 44 | 44 |
| 45 // A convenience subclass that automatically registers itself as the test | 45 // A convenience subclass that automatically registers itself as the test |
| 46 // StatisticsProvider during construction and cleans up at destruction. | 46 // StatisticsProvider during construction and cleans up at destruction. |
| 47 class ScopedFakeStatisticsProvider : public FakeStatisticsProvider { | 47 class ScopedFakeStatisticsProvider : public FakeStatisticsProvider { |
| 48 public: | 48 public: |
| 49 ScopedFakeStatisticsProvider(); | 49 ScopedFakeStatisticsProvider(); |
| 50 virtual ~ScopedFakeStatisticsProvider(); | 50 ~ScopedFakeStatisticsProvider() override; |
| 51 | 51 |
| 52 private: | 52 private: |
| 53 DISALLOW_COPY_AND_ASSIGN(ScopedFakeStatisticsProvider); | 53 DISALLOW_COPY_AND_ASSIGN(ScopedFakeStatisticsProvider); |
| 54 }; | 54 }; |
| 55 | 55 |
| 56 } // namespace system | 56 } // namespace system |
| 57 } // namespace chromeos | 57 } // namespace chromeos |
| 58 | 58 |
| 59 #endif // CHROMEOS_SYSTEM_FAKE_STATISTICS_PROVIDER_H_ | 59 #endif // CHROMEOS_SYSTEM_FAKE_STATISTICS_PROVIDER_H_ |
| OLD | NEW |