mock_ukm_data_manager.h 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. // Copyright 2022 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #ifndef COMPONENTS_SEGMENTATION_PLATFORM_INTERNAL_MOCK_UKM_DATA_MANAGER_H_
  5. #define COMPONENTS_SEGMENTATION_PLATFORM_INTERNAL_MOCK_UKM_DATA_MANAGER_H_
  6. #include "base/files/file_path.h"
  7. #include "components/segmentation_platform/internal/signals/ukm_config.h"
  8. #include "components/segmentation_platform/internal/ukm_data_manager.h"
  9. #include "testing/gmock/include/gmock/gmock.h"
  10. namespace segmentation_platform {
  11. class MockUkmDataManager : public UkmDataManager {
  12. public:
  13. MockUkmDataManager();
  14. ~MockUkmDataManager() override;
  15. MOCK_METHOD(void,
  16. Initialize,
  17. (const base::FilePath& database_path, UkmObserver*),
  18. (override));
  19. MOCK_METHOD(bool, IsUkmEngineEnabled, (), (override));
  20. MOCK_METHOD(void, StartObservingUkm, (const UkmConfig& config), (override));
  21. MOCK_METHOD(void, PauseOrResumeObservation, (bool pause), (override));
  22. MOCK_METHOD(UrlSignalHandler*, GetOrCreateUrlHandler, (), (override));
  23. MOCK_METHOD(UkmDatabase*, GetUkmDatabase, (), (override));
  24. MOCK_METHOD(void, OnEntryAdded, (ukm::mojom::UkmEntryPtr), (override));
  25. MOCK_METHOD(void,
  26. OnUkmSourceUpdated,
  27. (ukm::SourceId, const std::vector<GURL>&),
  28. (override));
  29. MOCK_METHOD(void, AddRef, (), (override));
  30. MOCK_METHOD(void, RemoveRef, (), (override));
  31. };
  32. } // namespace segmentation_platform
  33. #endif // COMPONENTS_SEGMENTATION_PLATFORM_INTERNAL_MOCK_UKM_DATA_MANAGER_H_