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
89
90
91
92
93
94
95
96
97
98
99
100
101
|
// Copyright 2012 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_EXTENSIONS_TEST_BLOCKLIST_H_
#define CHROME_BROWSER_EXTENSIONS_TEST_BLOCKLIST_H_
#include <map>
#include <string>
#include "base/memory/raw_ptr.h"
#include "chrome/browser/extensions/blocklist.h"
#include "chrome/browser/extensions/blocklist_state_fetcher.h"
#include "chrome/browser/extensions/scoped_database_manager_for_test.h"
#include "extensions/buildflags/buildflags.h"
static_assert(BUILDFLAG(ENABLE_EXTENSIONS_CORE));
namespace extensions {
class FakeSafeBrowsingDatabaseManager;
// Replace BlocklistStateFetcher for testing of the Boacklist class.
class BlocklistStateFetcherMock : public BlocklistStateFetcher {
public:
BlocklistStateFetcherMock();
~BlocklistStateFetcherMock() override;
void Request(const std::string& id, RequestCallback callback) override;
void SetState(const std::string& id, BlocklistState state);
void Clear();
int request_count() const { return request_count_; }
private:
std::map<std::string, BlocklistState> states_;
int request_count_;
};
// A wrapper for an extensions::Blocklist that provides functionality for
// testing. It sets up mocks for SafeBrowsing database and BlocklistFetcher,
// that are used by blocklist to retrieve respectively the set of blocklisted
// extensions and their blocklist states.
class TestBlocklist {
public:
// Use this if the SafeBrowsing and/or StateFetcher mocks should be created
// before initializing the Blocklist.
TestBlocklist();
explicit TestBlocklist(Blocklist* blocklist);
TestBlocklist(const TestBlocklist&) = delete;
TestBlocklist& operator=(const TestBlocklist&) = delete;
~TestBlocklist();
void Attach(Blocklist* blocklist);
// Only call this if Blocklist is destroyed before TestBlocklist, otherwise
// it will be performed from the destructor.
void Detach();
Blocklist* blocklist() { return blocklist_; }
// Set the extension state in SafeBrowsingDatabaseManager and
// BlocklistFetcher.
void SetBlocklistState(const std::string& extension_id,
BlocklistState state,
bool notify);
BlocklistState GetBlocklistState(const std::string& extension_id);
void Clear(bool notify);
void DisableSafeBrowsing();
void EnableSafeBrowsing();
void NotifyUpdate();
const BlocklistStateFetcherMock* fetcher() { return &state_fetcher_mock_; }
private:
raw_ptr<Blocklist> blocklist_;
// The BlocklistStateFetcher object is normally managed by Blocklist. Because
// of this, we need to prevent this object from being deleted with Blocklist.
// For this, Detach() should be called before blocklist_ is deleted.
BlocklistStateFetcherMock state_fetcher_mock_;
scoped_refptr<FakeSafeBrowsingDatabaseManager> blocklist_db_;
ScopedDatabaseManagerForTest scoped_blocklist_db_;
};
} // namespace extensions
#endif // CHROME_BROWSER_EXTENSIONS_TEST_BLOCKLIST_H_
|
