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 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125
|
// 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 "net/cookies/cookie_inclusion_status.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);
}
CookieAccessResult Include() {
return {};
}
CookieAccessResult Exclude(CookieInclusionStatus::ExclusionReason reason) {
return CookieAccessResult(CookieInclusionStatus::MakeFromReasonsForTesting(
/*exclusions=*/{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(
{CookieInclusionStatus::ExclusionReason::
EXCLUDE_USER_PREFERENCES}),
_, _, _)),
MatchesCookieWithAccessResult(
MatchesCookieWithName("2"),
MatchesCookieAccessResult(
HasExactlyExclusionReasonsForTesting(
{CookieInclusionStatus::ExclusionReason::
EXCLUDE_USER_PREFERENCES}),
_, _, _)),
MatchesCookieWithAccessResult(
MatchesCookieWithName("3"),
MatchesCookieAccessResult(
HasExactlyExclusionReasonsForTesting(
{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(
{CookieInclusionStatus::ExclusionReason::
EXCLUDE_SECURE_ONLY}),
_, _, _)),
MatchesCookieWithAccessResult(
MatchesCookieWithName("4"),
MatchesCookieAccessResult(
HasExactlyExclusionReasonsForTesting(
{CookieInclusionStatus::ExclusionReason::
EXCLUDE_SECURE_ONLY}),
_, _, _))));
}
} // namespace net
|