131 lines
4.9 KiB
C++
131 lines
4.9 KiB
C++
// 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.
|
|
|
|
#include "net/base/network_delegate.h"
|
|
|
|
#include "net/cookies/canonical_cookie.h"
|
|
#include "net/cookies/canonical_cookie_test_helpers.h"
|
|
#include "testing/gmock/include/gmock/gmock.h"
|
|
#include "testing/gtest/include/gtest/gtest.h"
|
|
|
|
namespace net {
|
|
|
|
using testing::_;
|
|
using testing::ElementsAre;
|
|
using testing::IsEmpty;
|
|
using testing::UnorderedElementsAre;
|
|
|
|
namespace {
|
|
|
|
constexpr char kURL[] = "example.test";
|
|
|
|
CanonicalCookie MakeCookie(const std::string& name) {
|
|
return *CanonicalCookie::CreateUnsafeCookieForTesting(
|
|
name, "value", kURL, /*path=*/"/", /*creation=*/base::Time(),
|
|
/*expiration=*/base::Time(), /*last_access=*/base::Time(),
|
|
/*last_update=*/base::Time(),
|
|
/*secure=*/true, /*httponly=*/false, CookieSameSite::UNSPECIFIED,
|
|
CookiePriority::COOKIE_PRIORITY_DEFAULT, /*same_party=*/false);
|
|
}
|
|
|
|
CookieAccessResult Include() {
|
|
return {};
|
|
}
|
|
|
|
CookieAccessResult Exclude(CookieInclusionStatus::ExclusionReason reason) {
|
|
return CookieAccessResult(CookieInclusionStatus(reason));
|
|
}
|
|
|
|
} // namespace
|
|
|
|
TEST(NetworkDelegateTest, ExcludeAllCookies) {
|
|
CookieAccessResultList maybe_included_cookies = {
|
|
{MakeCookie("1"), Include()}, {MakeCookie("2"), Include()}};
|
|
|
|
CookieAccessResultList excluded_cookies = {
|
|
{MakeCookie("3"),
|
|
Exclude(CookieInclusionStatus::ExclusionReason::EXCLUDE_SECURE_ONLY)}};
|
|
|
|
NetworkDelegate::ExcludeAllCookies(
|
|
CookieInclusionStatus::ExclusionReason::EXCLUDE_USER_PREFERENCES,
|
|
maybe_included_cookies, excluded_cookies);
|
|
|
|
EXPECT_THAT(maybe_included_cookies, IsEmpty());
|
|
EXPECT_THAT(
|
|
excluded_cookies,
|
|
UnorderedElementsAre(
|
|
MatchesCookieWithAccessResult(
|
|
MatchesCookieWithName("1"),
|
|
MatchesCookieAccessResult(
|
|
HasExactlyExclusionReasonsForTesting(
|
|
std::vector<CookieInclusionStatus::ExclusionReason>{
|
|
CookieInclusionStatus::ExclusionReason::
|
|
EXCLUDE_USER_PREFERENCES}),
|
|
_, _, _)),
|
|
MatchesCookieWithAccessResult(
|
|
MatchesCookieWithName("2"),
|
|
MatchesCookieAccessResult(
|
|
HasExactlyExclusionReasonsForTesting(
|
|
std::vector<CookieInclusionStatus::ExclusionReason>{
|
|
CookieInclusionStatus::ExclusionReason::
|
|
EXCLUDE_USER_PREFERENCES}),
|
|
_, _, _)),
|
|
MatchesCookieWithAccessResult(
|
|
MatchesCookieWithName("3"),
|
|
MatchesCookieAccessResult(
|
|
HasExactlyExclusionReasonsForTesting(
|
|
std::vector<CookieInclusionStatus::ExclusionReason>{
|
|
CookieInclusionStatus::ExclusionReason::
|
|
EXCLUDE_SECURE_ONLY,
|
|
CookieInclusionStatus::ExclusionReason::
|
|
EXCLUDE_USER_PREFERENCES}),
|
|
_, _, _))));
|
|
}
|
|
|
|
TEST(NetworkDelegateTest, MoveExcludedCookies) {
|
|
CookieAccessResultList maybe_included_cookies = {
|
|
{MakeCookie("1"), Include()},
|
|
{MakeCookie("2"),
|
|
Exclude(CookieInclusionStatus::ExclusionReason::EXCLUDE_SECURE_ONLY)},
|
|
{MakeCookie("3"), Include()}};
|
|
|
|
CookieAccessResultList excluded_cookies = {{
|
|
MakeCookie("4"),
|
|
Exclude(CookieInclusionStatus::ExclusionReason::EXCLUDE_SECURE_ONLY),
|
|
}};
|
|
|
|
NetworkDelegate::MoveExcludedCookies(maybe_included_cookies,
|
|
excluded_cookies);
|
|
|
|
EXPECT_THAT(
|
|
maybe_included_cookies,
|
|
ElementsAre(MatchesCookieWithAccessResult(
|
|
MatchesCookieWithName("1"),
|
|
MatchesCookieAccessResult(IsInclude(), _, _, _)),
|
|
MatchesCookieWithAccessResult(
|
|
MatchesCookieWithName("3"),
|
|
MatchesCookieAccessResult(IsInclude(), _, _, _))));
|
|
EXPECT_THAT(
|
|
excluded_cookies,
|
|
UnorderedElementsAre(
|
|
MatchesCookieWithAccessResult(
|
|
MatchesCookieWithName("2"),
|
|
MatchesCookieAccessResult(
|
|
HasExactlyExclusionReasonsForTesting(
|
|
std::vector<CookieInclusionStatus::ExclusionReason>{
|
|
CookieInclusionStatus::ExclusionReason::
|
|
EXCLUDE_SECURE_ONLY}),
|
|
_, _, _)),
|
|
MatchesCookieWithAccessResult(
|
|
MatchesCookieWithName("4"),
|
|
MatchesCookieAccessResult(
|
|
HasExactlyExclusionReasonsForTesting(
|
|
std::vector<CookieInclusionStatus::ExclusionReason>{
|
|
CookieInclusionStatus::ExclusionReason::
|
|
EXCLUDE_SECURE_ONLY}),
|
|
_, _, _))));
|
|
}
|
|
|
|
} // namespace net
|