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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <optional>
#include "base/memory/raw_ptr.h"
#include "base/run_loop.h"
#include "base/time/time.h"
#include "chrome/browser/metrics/ukm_background_recorder_service.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "components/history/core/browser/history_service.h"
#include "content/public/test/browser_test.h"
#include "services/metrics/public/cpp/ukm_builders.h"
#include "services/metrics/public/cpp/ukm_source_id.h"
#include "url/gurl.h"
#include "url/origin.h"
namespace {
constexpr char kVisitedUrl[] = "https://foobar.com/baz";
void DidGetRecordResult(base::OnceClosure quit_closure,
std::optional<ukm::SourceId>* out_result,
std::optional<ukm::SourceId> result) {
*out_result = std::move(result);
std::move(quit_closure).Run();
}
} // namespace
class UkmBackgroundRecorderBrowserTest : public InProcessBrowserTest {
public:
UkmBackgroundRecorderBrowserTest() = default;
UkmBackgroundRecorderBrowserTest(const UkmBackgroundRecorderBrowserTest&) =
delete;
UkmBackgroundRecorderBrowserTest& operator=(
const UkmBackgroundRecorderBrowserTest&) = delete;
~UkmBackgroundRecorderBrowserTest() override = default;
void SetUpOnMainThread() override {
background_recorder_service_ =
ukm::UkmBackgroundRecorderFactory::GetForProfile(browser()->profile());
DCHECK(background_recorder_service_);
// Adds the URL to the history so that UKM events for this origin are
// recorded.
background_recorder_service_->history_service_->AddPage(
GURL(kVisitedUrl), base::Time::Now(), history::SOURCE_BROWSED);
}
void TearDownOnMainThread() override {
background_recorder_service_ = nullptr;
}
protected:
std::optional<ukm::SourceId> GetSourceId(const url::Origin& origin) {
std::optional<ukm::SourceId> result;
base::RunLoop run_loop;
background_recorder_service_->GetBackgroundSourceIdIfAllowed(
origin,
base::BindOnce(&DidGetRecordResult, run_loop.QuitClosure(), &result));
run_loop.Run();
return result;
}
private:
raw_ptr<ukm::UkmBackgroundRecorderService> background_recorder_service_ =
nullptr;
};
IN_PROC_BROWSER_TEST_F(UkmBackgroundRecorderBrowserTest,
SourceIdReturnedWhenOriginInHistory) {
// Check visited origin.
auto source_id = GetSourceId(url::Origin::Create(GURL(kVisitedUrl)));
ASSERT_TRUE(source_id);
EXPECT_NE(*source_id, ukm::kInvalidSourceId);
EXPECT_EQ(ukm::GetSourceIdType(*source_id), ukm::SourceIdType::HISTORY_ID);
// Check unvisited origin.
EXPECT_FALSE(
GetSourceId(url::Origin::Create(GURL("https://notvisited.com"))));
}
|
