chromium/components/history_clusters/core/history_clusters_service_test_api.h

// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef COMPONENTS_HISTORY_CLUSTERS_CORE_HISTORY_CLUSTERS_SERVICE_TEST_API_H_
#define COMPONENTS_HISTORY_CLUSTERS_CORE_HISTORY_CLUSTERS_SERVICE_TEST_API_H_

#include <vector>

#include "base/run_loop.h"
#include "base/test/bind.h"
#include "base/time/time.h"
#include "components/history/core/browser/history_service.h"
#include "components/history/core/browser/history_types.h"
#include "components/history/core/test/history_service_test_util.h"
#include "components/history_clusters/core/clustering_backend.h"
#include "components/history_clusters/core/history_clusters_service.h"

namespace history_clusters {

class HistoryClustersServiceTestApi { … };

// Fetches two hardcoded test visits.
std::vector<history::AnnotatedVisit> GetHardcodedTestVisits();

// Fetches the hardcoded `ClusterVisit` with ID `visit_id`.
history::ClusterVisit GetHardcodedClusterVisit(history::VisitID visit_id,
                                               float score = 0.5,
                                               int engagement_score = 0);

history::ClusterVisit AnnotatedVisitToClusterVisit(
    const history::AnnotatedVisit& visit,
    float score = .5,
    int engagement_score = 0);

}  // namespace history_clusters

#endif  // COMPONENTS_HISTORY_CLUSTERS_CORE_HISTORY_CLUSTERS_SERVICE_TEST_API_H_