1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70
|
// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_NEW_TAB_PAGE_MODULES_TEST_SUPPORT_H_
#define CHROME_BROWSER_NEW_TAB_PAGE_MODULES_TEST_SUPPORT_H_
#include <string>
#include <vector>
#include "base/test/scoped_feature_list.h"
#include "components/history/core/browser/history_service.h"
#include "components/search/ntp_features.h"
#include "testing/gmock/include/gmock/gmock.h"
namespace ntp {
class MockHistoryService : public history::HistoryService {
public:
MockHistoryService();
MockHistoryService(const MockHistoryService&) = delete;
MockHistoryService& operator=(const MockHistoryService&) = delete;
~MockHistoryService() override;
MOCK_METHOD1(ClearCachedDataForContextID,
void(history::ContextID context_id));
MOCK_CONST_METHOD5(
GetAnnotatedVisits,
base::CancelableTaskTracker::TaskId(
const history::QueryOptions& options,
bool compute_redirect_chain_start_properties,
bool get_unclustered_visits_only,
history::HistoryService::GetAnnotatedVisitsCallback callback,
base::CancelableTaskTracker* tracker));
MOCK_METHOD3(HideVisits,
base::CancelableTaskTracker::TaskId(
const std::vector<history::VisitID>& visit_ids,
base::OnceClosure callback,
base::CancelableTaskTracker* tracker));
MOCK_METHOD4(
UpdateVisitsInteractionState,
base::CancelableTaskTracker::TaskId(
const std::vector<history::VisitID>& visit_ids,
const history::ClusterVisit::InteractionState interaction_state,
base::OnceClosure callback,
base::CancelableTaskTracker* tracker));
MOCK_CONST_METHOD4(ToAnnotatedVisits,
base::CancelableTaskTracker::TaskId(
const history::VisitVector& visit_rows,
bool compute_redirect_chain_start_properties,
ToAnnotatedVisitsCallback callback,
base::CancelableTaskTracker* tracker));
};
inline constexpr size_t kNumModuleFeatures = 5;
extern const std::vector<base::test::FeatureRef>& kAllModuleFeatures;
std::vector<base::test::FeatureRef> ComputeDisabledFeaturesList(
const std::vector<base::test::FeatureRef>& features,
const std::vector<base::test::FeatureRef>& enabled_features);
} // namespace ntp
#endif // CHROME_BROWSER_NEW_TAB_PAGE_MODULES_TEST_SUPPORT_H_
|