3657c84a89
With a common prefix, all of the URL Classifier gtests can be run like this: ./mach gtest UrlClassifier* MozReview-Commit-ID: IqQznsldFOD
53 lines
1.5 KiB
C++
53 lines
1.5 KiB
C++
/* 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 {
|
|
UniquePtr<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);
|
|
}
|