mock_tracker.h 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  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
  3. // found in the LICENSE file.
  4. #ifndef COMPONENTS_FEATURE_ENGAGEMENT_TEST_MOCK_TRACKER_H_
  5. #define COMPONENTS_FEATURE_ENGAGEMENT_TEST_MOCK_TRACKER_H_
  6. #include <memory>
  7. #include <string>
  8. #include "components/feature_engagement/public/tracker.h"
  9. #include "testing/gmock/include/gmock/gmock.h"
  10. #include "third_party/abseil-cpp/absl/types/optional.h"
  11. namespace feature_engagement {
  12. namespace test {
  13. class MockTracker : public Tracker {
  14. public:
  15. MockTracker();
  16. MockTracker(const MockTracker&) = delete;
  17. MockTracker& operator=(const MockTracker&) = delete;
  18. ~MockTracker() override;
  19. // Tracker implememtation.
  20. MOCK_METHOD1(NotifyEvent, void(const std::string& event));
  21. MOCK_METHOD1(ShouldTriggerHelpUI, bool(const base::Feature& feature));
  22. MOCK_METHOD1(ShouldTriggerHelpUIWithSnooze,
  23. TriggerDetails(const base::Feature& feature));
  24. MOCK_CONST_METHOD1(WouldTriggerHelpUI, bool(const base::Feature& feature));
  25. MOCK_CONST_METHOD2(HasEverTriggered,
  26. bool(const base::Feature& feature, bool from_window));
  27. MOCK_CONST_METHOD1(GetTriggerState,
  28. Tracker::TriggerState(const base::Feature& feature));
  29. MOCK_CONST_METHOD0(IsInitialized, bool());
  30. MOCK_METHOD1(Dismissed, void(const base::Feature& feature));
  31. MOCK_METHOD2(DismissedWithSnooze,
  32. void(const base::Feature& feature,
  33. absl::optional<SnoozeAction> snooze_action));
  34. MOCK_METHOD0(AcquireDisplayLock, std::unique_ptr<DisplayLockHandle>());
  35. MOCK_METHOD1(SetPriorityNotification, void(const base::Feature&));
  36. MOCK_METHOD0(GetPendingPriorityNotification, absl::optional<std::string>());
  37. MOCK_METHOD2(RegisterPriorityNotificationHandler,
  38. void(const base::Feature&, base::OnceClosure));
  39. MOCK_METHOD1(UnregisterPriorityNotificationHandler,
  40. void(const base::Feature&));
  41. MOCK_METHOD1(AddOnInitializedCallback, void(OnInitializedCallback callback));
  42. };
  43. } // namespace test
  44. } // namespace feature_engagement
  45. #endif // COMPONENTS_FEATURE_ENGAGEMENT_TEST_MOCK_TRACKER_H_