-
Notifications
You must be signed in to change notification settings - Fork 54
/
TestLookupCacheV4.cpp
45 lines (35 loc) · 1.49 KB
/
TestLookupCacheV4.cpp
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
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "Common.h"
void TestHasPrefix(const _Fragment& aFragment, bool aExpectedHas,
bool aExpectedComplete) {
_PrefixArray array = {GeneratePrefix(_Fragment("bravo.com/"), 32),
GeneratePrefix(_Fragment("browsing.com/"), 8),
GeneratePrefix(_Fragment("gound.com/"), 5),
GeneratePrefix(_Fragment("small.com/"), 4)};
RunTestInNewThread([&]() -> void {
RefPtr<LookupCache> cache = SetupLookupCache<LookupCacheV4>(array);
Completion lookupHash;
lookupHash.FromPlaintext(aFragment);
bool has, confirmed;
uint32_t matchLength;
// Freshness is not used in V4 so we just put dummy values here.
TableFreshnessMap dummy;
nsresult rv = cache->Has(lookupHash, &has, &matchLength, &confirmed);
EXPECT_EQ(rv, NS_OK);
EXPECT_EQ(has, aExpectedHas);
EXPECT_EQ(matchLength == COMPLETE_SIZE, aExpectedComplete);
EXPECT_EQ(confirmed, false);
cache->ClearAll();
});
}
TEST(UrlClassifierLookupCacheV4, HasComplete) {
TestHasPrefix(_Fragment("bravo.com/"), true, true);
}
TEST(UrlClassifierLookupCacheV4, HasPrefix) {
TestHasPrefix(_Fragment("browsing.com/"), true, false);
}
TEST(UrlClassifierLookupCacheV4, Nomatch) {
TestHasPrefix(_Fragment("nomatch.com/"), false, false);
}