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 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/webid/federated_identity_identity_provider_registration_context.h"
#include <memory>
#include "chrome/test/base/testing_profile.h"
#include "content/public/test/browser_task_environment.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "url/gurl.h"
#include "url/origin.h"
class FederatedIdentityIdentityProviderRegistrationContextTest
: public testing::Test {
public:
FederatedIdentityIdentityProviderRegistrationContextTest() {
context_ =
std::make_unique<FederatedIdentityIdentityProviderRegistrationContext>(
&profile_);
}
void TearDown() override { context_.reset(); }
~FederatedIdentityIdentityProviderRegistrationContextTest() override =
default;
FederatedIdentityIdentityProviderRegistrationContextTest(
FederatedIdentityIdentityProviderRegistrationContextTest&) = delete;
FederatedIdentityIdentityProviderRegistrationContextTest& operator=(
FederatedIdentityIdentityProviderRegistrationContextTest&) = delete;
FederatedIdentityIdentityProviderRegistrationContext* context() {
return context_.get();
}
TestingProfile* profile() { return &profile_; }
private:
content::BrowserTaskEnvironment task_environment_;
std::unique_ptr<FederatedIdentityIdentityProviderRegistrationContext>
context_;
TestingProfile profile_;
};
// Test the initial state of the registry.
TEST_F(FederatedIdentityIdentityProviderRegistrationContextTest,
InitiallyEmptyRegistry) {
EXPECT_TRUE(context()->GetRegisteredIdPs().empty());
}
// Test the state of the registry after registration.
TEST_F(FederatedIdentityIdentityProviderRegistrationContextTest, RegisterIdP) {
const GURL configURL = GURL("https://idp.example/fedcm.json");
context()->RegisterIdP(configURL);
EXPECT_EQ(std::vector<GURL>{configURL}, context()->GetRegisteredIdPs());
}
// Test that registrating twice is idempotent.
TEST_F(FederatedIdentityIdentityProviderRegistrationContextTest,
RegistrationIsIdempotent) {
const GURL configURL = GURL("https://idp.example/fedcm.json");
context()->RegisterIdP(configURL);
context()->RegisterIdP(configURL);
// Removes duplicates.
EXPECT_EQ(std::vector<GURL>{configURL}, context()->GetRegisteredIdPs());
}
// Test that registering two IdPs preserves the order.
TEST_F(FederatedIdentityIdentityProviderRegistrationContextTest,
RegisteringTwoIdPs) {
const GURL configURL1 = GURL("https://idp1.example/fedcm.json");
const GURL configURL2 = GURL("https://idp2.example/fedcm.json");
context()->RegisterIdP(configURL1);
context()->RegisterIdP(configURL2);
std::vector<GURL> expected = {configURL1, configURL2};
EXPECT_EQ(expected, context()->GetRegisteredIdPs());
}
// Registers and unregisters an IdP.
TEST_F(FederatedIdentityIdentityProviderRegistrationContextTest,
Unregistering) {
const GURL configURL = GURL("https://idp.example/fedcm.json");
context()->RegisterIdP(configURL);
EXPECT_EQ(std::vector<GURL>{configURL}, context()->GetRegisteredIdPs());
context()->UnregisterIdP(configURL);
EXPECT_TRUE(context()->GetRegisteredIdPs().empty());
}
// Unregistering an IdP that wasn't registered is a no-op.
TEST_F(FederatedIdentityIdentityProviderRegistrationContextTest,
UnregisteringWithoutRegistering) {
const GURL configURL = GURL("https://idp.example/fedcm.json");
EXPECT_TRUE(context()->GetRegisteredIdPs().empty());
context()->UnregisterIdP(configURL);
EXPECT_TRUE(context()->GetRegisteredIdPs().empty());
}
// RegistersTwoIdPsUnregistersOne
TEST_F(FederatedIdentityIdentityProviderRegistrationContextTest,
RegisteringTwoIdPsUnregistersOne) {
const GURL configURL1 = GURL("https://idp1.example/fedcm.json");
const GURL configURL2 = GURL("https://idp2.example/fedcm.json");
context()->RegisterIdP(configURL1);
context()->RegisterIdP(configURL2);
context()->UnregisterIdP(configURL1);
std::vector<GURL> expected = {configURL2};
EXPECT_EQ(expected, context()->GetRegisteredIdPs());
}
|